81 lines
2.8 KiB
Rust
81 lines
2.8 KiB
Rust
// Copyright 2016 Mozilla
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License"); you may not use
|
|
// this file except in compliance with the License. You may obtain a copy of the
|
|
// License at http://www.apache.org/licenses/LICENSE-2.0
|
|
// Unless required by applicable law or agreed to in writing, software distributed
|
|
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
|
|
// CONDITIONS OF ANY KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations under the License.
|
|
|
|
extern crate core_traits;
|
|
extern crate edn;
|
|
extern crate mentat_core;
|
|
extern crate mentat_query_algebrizer;
|
|
extern crate query_algebrizer_traits;
|
|
|
|
mod utils;
|
|
|
|
use crate::utils::{alg, bails, SchemaBuilder};
|
|
|
|
use core_traits::ValueType;
|
|
|
|
use mentat_core::Schema;
|
|
|
|
use mentat_query_algebrizer::Known;
|
|
|
|
fn prepopulated_schema() -> Schema {
|
|
SchemaBuilder::new()
|
|
.define_simple_attr("test", "boolean", ValueType::Boolean, false)
|
|
.define_simple_attr("test", "long", ValueType::Long, false)
|
|
.define_simple_attr("test", "double", ValueType::Double, false)
|
|
.define_simple_attr("test", "string", ValueType::String, false)
|
|
.define_simple_attr("test", "keyword", ValueType::Keyword, false)
|
|
.define_simple_attr("test", "uuid", ValueType::Uuid, false)
|
|
.define_simple_attr("test", "instant", ValueType::Instant, false)
|
|
.define_simple_attr("test", "ref", ValueType::Ref, false)
|
|
.define_simple_attr("test", "bytes", ValueType::Bytes, false)
|
|
.schema
|
|
}
|
|
|
|
#[test]
|
|
fn test_empty_known() {
|
|
let schema = prepopulated_schema();
|
|
let known = Known::for_schema(&schema);
|
|
for known_type in ValueType::all_enums().iter() {
|
|
for required in ValueType::all_enums().iter() {
|
|
let q = format!(
|
|
"[:find ?e :where [?e :test/{} ?v] [(type ?v {})]]",
|
|
known_type.into_keyword().name(),
|
|
required
|
|
);
|
|
println!("Query: {}", q);
|
|
let cc = alg(known, &q);
|
|
// It should only be empty if the known type and our requirement differ.
|
|
assert_eq!(
|
|
cc.empty_because.is_some(),
|
|
known_type != required,
|
|
"known_type = {}; required = {}",
|
|
known_type,
|
|
required
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_multiple() {
|
|
let schema = prepopulated_schema();
|
|
let known = Known::for_schema(&schema);
|
|
let q = "[:find ?e :where [?e _ ?v] [(type ?v :db.type/long)] [(type ?v :db.type/double)]]";
|
|
let cc = alg(known, &q);
|
|
assert!(cc.empty_because.is_some());
|
|
}
|
|
|
|
#[test]
|
|
fn test_unbound() {
|
|
let schema = prepopulated_schema();
|
|
let known = Known::for_schema(&schema);
|
|
bails(known, "[:find ?e :where [(type ?e :db.type/string)]]");
|
|
}
|