Skip to content

Commit

Permalink
Remove unit tests for Couchbase ticket
Browse files Browse the repository at this point in the history
  • Loading branch information
@Nao-ris
Nao-ris committed Sep 30, 2024
1 parent decd5dc commit fe1c865
Showing 1 changed file with 0 additions and 284 deletions.
284 changes: 0 additions & 284 deletions tests/query_tests.rs
View file
Original file line number Diff line number Diff line change
Expand Up @@ -212,287 +212,3 @@ fn parameters() {
assert_eq!(query.execute().unwrap().count(), 1);
});
}

// Use case 1: ARRAY_CONTAINS => bad index (because of partial match)
#[test]
fn use_case_1() {
utils::with_db(|db| {
assert!(db
.create_index(
"good_index",
&ValueIndexConfiguration::new(
QueryLanguage::JSON,
r#"[[".type"],[".session_id"]]"#
),
)
.unwrap());

assert!(db
.create_index(
"bad_index",
&ValueIndexConfiguration::new(QueryLanguage::JSON, r#"[[".type"]]"#),
)
.unwrap());

let mut doc = Document::new_with_id("id1");
let mut props = doc.mutable_properties();
props.at("type").put_string("HardcodedType");
props.at("session_id").put_string("session1");
db.save_document_with_concurency_control(&mut doc, ConcurrencyControl::FailOnConflict)
.expect("save");

let query = Query::new(
db,
QueryLanguage::N1QL,
"SELECT _.* FROM _ \
WHERE _.type = 'HardcodedType' \
AND ARRAY_CONTAINS($sessionIds, _.session_id)",
)
.expect("create query");

let mut params = MutableDict::new();
let mut session_ids = MutableArray::new();
session_ids.append().put_string("session1");
session_ids.append().put_string("session2");
params.at("sessionIds").put_value(&session_ids);
query.set_parameters(&params);

let params = query.parameters();
assert_eq!(
params.get("sessionIds").as_array().get(0).as_string(),
Some("session1")
);
assert_eq!(
params.get("sessionIds").as_array().get(1).as_string(),
Some("session2")
);

println!("Explain: {:?}", query.explain());
assert_eq!(query.execute().unwrap().count(), 1);
});
}

// Use case 2: IN ($truc) + set_parameters => no result
#[test]
fn use_case_2() {
utils::with_db(|db| {
assert!(db
.create_index(
"good_index",
&ValueIndexConfiguration::new(
QueryLanguage::JSON,
r#"[[".type"],[".session_id"]]"#
),
)
.unwrap());

assert!(db
.create_index(
"bad_index",
&ValueIndexConfiguration::new(QueryLanguage::JSON, r#"[[".type"]]"#),
)
.unwrap());

let mut doc = Document::new_with_id("id1");
let mut props = doc.mutable_properties();
props.at("type").put_string("HardcodedType");
props.at("session_id").put_string("session1");
db.save_document_with_concurency_control(&mut doc, ConcurrencyControl::FailOnConflict)
.expect("save");

let query = Query::new(
db,
QueryLanguage::N1QL,
"SELECT _.* FROM _ \
WHERE _.type = 'HardcodedType' \
AND _.session_id IN ($sessionIds)",
)
.expect("create query");

let mut params = MutableDict::new();
let mut session_ids = MutableArray::new();
session_ids.append().put_string("session1");
session_ids.append().put_string("session2");
params.at("sessionIds").put_value(&session_ids);
query.set_parameters(&params);

let params = query.parameters();
assert_eq!(
params.get("sessionIds").as_array().get(0).as_string(),
Some("session1")
);
assert_eq!(
params.get("sessionIds").as_array().get(1).as_string(),
Some("session2")
);

println!("Explain: {:?}", query.explain());
assert_eq!(query.execute().unwrap().count(), 1);
});
}

// Use case 3: IN + hardcoded values => result + good index, but sql injection
#[test]
fn use_case_3() {
utils::with_db(|db| {
assert!(db
.create_index(
"good_index",
&ValueIndexConfiguration::new(
QueryLanguage::JSON,
r#"[[".type"],[".session_id"]]"#
),
)
.unwrap());

assert!(db
.create_index(
"bad_index",
&ValueIndexConfiguration::new(QueryLanguage::JSON, r#"[[".type"]]"#),
)
.unwrap());

let mut doc = Document::new_with_id("id1");
let mut props = doc.mutable_properties();
props.at("type").put_string("HardcodedType");
props.at("session_id").put_string("session1");
db.save_document_with_concurency_control(&mut doc, ConcurrencyControl::FailOnConflict)
.expect("save");

let query = Query::new(
db,
QueryLanguage::N1QL,
"SELECT _.* FROM _ \
WHERE _.type = 'HardcodedType' \
AND _.session_id in ('session1', 'session2')",
)
.expect("create query");

println!("Explain: {:?}", query.explain());
assert_eq!(query.execute().unwrap().count(), 1);
});
}

// Use case 4: IN [$truc] + set_parameters => n1ql error
#[test]
fn use_case_4() {
utils::with_db(|db| {
assert!(db
.create_index(
"good_index",
&ValueIndexConfiguration::new(
QueryLanguage::JSON,
r#"[[".type"],[".session_id"]]"#
),
)
.unwrap());

assert!(db
.create_index(
"bad_index",
&ValueIndexConfiguration::new(QueryLanguage::JSON, r#"[[".type"]]"#),
)
.unwrap());

let mut doc = Document::new_with_id("id1");
let mut props = doc.mutable_properties();
props.at("type").put_string("HardcodedType");
props.at("session_id").put_string("session1");
db.save_document_with_concurency_control(&mut doc, ConcurrencyControl::FailOnConflict)
.expect("save");

let query = Query::new(
db,
QueryLanguage::N1QL,
"SELECT _.* FROM _ \
WHERE _.type = 'HardcodedType' \
AND _.session_id IN [$sessionIds]",
)
.expect("create query");

let mut params = MutableDict::new();
let mut session_ids = MutableArray::new();
session_ids.append().put_string("session1");
session_ids.append().put_string("session2");
params.at("sessionIds").put_value(&session_ids);
query.set_parameters(&params);

let params = query.parameters();
assert_eq!(
params.get("sessionIds").as_array().get(0).as_string(),
Some("session1")
);
assert_eq!(
params.get("sessionIds").as_array().get(1).as_string(),
Some("session2")
);

println!("Explain: {:?}", query.explain());
assert_eq!(query.execute().unwrap().count(), 1);
});
}

// Use case 5: IN [$truc] + set_parameters => n1ql error
#[test]
fn use_case_5() {
utils::with_db(|db| {
assert!(db
.create_index(
"good_index",
&ValueIndexConfiguration::new(
QueryLanguage::JSON,
r#"[[".type"],[".session_id"]]"#
),
)
.unwrap());

assert!(db
.create_index(
"bad_index",
&ValueIndexConfiguration::new(QueryLanguage::JSON, r#"[[".type"]]"#),
)
.unwrap());

let mut doc = Document::new_with_id("id1");
let mut props = doc.mutable_properties();
props.at("type").put_string("HardcodedType");
props.at("session_id").put_string("session1");
db.save_document_with_concurency_control(&mut doc, ConcurrencyControl::FailOnConflict)
.expect("save");

let query = Query::new(
db,
QueryLanguage::N1QL,
"SELECT _.* FROM _ \
WHERE _.type = 'HardcodedType' \
AND _.session_id IN $sessionIds",
)
.expect("create query");

let mut params = MutableDict::new();
let mut session_ids = MutableArray::new();
session_ids.append().put_string("session1");
session_ids.append().put_string("session2");
params.at("sessionIds").put_value(&session_ids);
query.set_parameters(&params);

let params = query.parameters();
assert_eq!(
params.get("sessionIds").as_array().get(0).as_string(),
Some("session1")
);
assert_eq!(
params.get("sessionIds").as_array().get(1).as_string(),
Some("session2")
);

println!("Explain: {:?}", query.explain());
assert_eq!(query.execute().unwrap().count(), 1);
});
}

// Use case 1: ARRAY_CONTAINS => bad index (because of partial match)
// Use case 2: IN ($truc) + set_parameters => no result
// Use case 3: IN + hardcoded values => result + good index, but sql injection
// Use case 4: IN [$truc] + set_parameters => n1ql error
// Use case 5: IN $truc + set_parameters => n1ql error

0 comments on commit fe1c865

Please sign in to comment.