use bson::Bson;
use mongodb::{CommandType, Client, ThreadedClient};
use mongodb::db::ThreadedDatabase;
use mongodb::error::Error::OperationError;
fn doc_vec_find(vec: &[Bson], key: &str, val: &str) -> Option<Bson> {
vec.iter()
.cloned()
.find(|bdoc| match *bdoc {
Bson::Document(ref doc) => {
match doc.get(key) {
Some(&Bson::String(ref s)) => s == val,
_ => false,
}
}
_ => false,
})
.map(|ref bson| bson.to_owned())
}
#[test]
fn invalid_user() {
let client = Client::connect("localhost", 27017).unwrap();
let db = client.db("test-auth-mod-invalid_user");
let _ = db.drop_user("test-auth-mod-invalid_user-saghm", None);
let doc = doc! { "connectionStatus": 1};
let before = db.command(doc.clone(), CommandType::Suppressed, None)
.unwrap();
let info = match before.get("authInfo") {
Some(&Bson::Document(ref doc)) => doc.clone(),
_ => panic!("Invalid response for initial connectionStatus command"),
};
match info.get("authenticatedUsers") {
Some(&Bson::Array(ref vec)) => {
assert!(doc_vec_find(&vec, "user", "test-auth-mod-invalid_user-saghm").is_none())
}
_ => panic!("Invalid array of authenticatedUsers for initial connectionStatus command"),
};
match db.auth("test-auth-mod-invalid_user-saghm", "some_password") {
Err(OperationError(_)) => (),
Err(_) => {
panic!(
"Expected OperationError for invalid authentication, but got some other error instead"
)
}
_ => panic!("Authentication succeeded despite invalid credentials"),
};
let after = db.command(doc, CommandType::Suppressed, None).unwrap();
let info = match after.get("authInfo") {
Some(&Bson::Document(ref doc)) => doc.clone(),
_ => panic!("Invalid response for subsequent connectionStatus command"),
};
match info.get("authenticatedUsers") {
Some(&Bson::Array(ref vec)) => {
assert!(doc_vec_find(&vec, "user", "test-auth-mod-invalid_user-saghm").is_none())
}
_ => panic!("Invalid array of authenticatedUsers for initial connectionStatus command"),
};
}
#[test]
fn invalid_password() {
let client = Client::connect("localhost", 27017).unwrap();
let db = client.db("test-auth-mod-invalid_password");
let _ = db.drop_user("test-auth-mod-invalid_password-saghm", None);
let doc = doc! { "connectionStatus": 1};
let before = db.command(doc.clone(), CommandType::Suppressed, None)
.unwrap();
let info = match before.get("authInfo") {
Some(&Bson::Document(ref doc)) => doc.clone(),
_ => panic!("Invalid response for initial connectionStatus command"),
};
match info.get("authenticatedUsers") {
Some(&Bson::Array(ref vec)) => {
assert!(doc_vec_find(&vec, "user", "test-auth-mod-invalid_password-saghm").is_none())
}
_ => panic!("Invalid array of authenticatedUsers for initial connectionStatus command"),
};
db.create_user(
"test-auth-mod-invalid_password-saghm",
"such_secure_password",
None,
).unwrap();
match db.auth("test-auth-mod-invalid_password-saghm", "wrong_password") {
Err(OperationError(_)) => (),
Err(_) => {
panic!(
"Expected OperationError for invalid authentication, but got some other error instead"
)
}
_ => panic!("Authentication succeeded despite invalid credentials"),
};
let after = db.command(doc, CommandType::Suppressed, None).unwrap();
let info = match after.get("authInfo") {
Some(&Bson::Document(ref doc)) => doc.clone(),
_ => panic!("Invalid response for subsequent connectionStatus command"),
};
match info.get("authenticatedUsers") {
Some(&Bson::Array(ref vec)) => {
assert!(doc_vec_find(&vec, "user", "test-auth-mod-invalid_password-saghm").is_none())
}
_ => panic!("Invalid array of authenticatedUsers for subsequent connectionStatus command"),
};
}
#[test]
fn successful_login() {
let client = Client::connect("localhost", 27017).unwrap();
let db = client.db("test-auth-mod-successful_login");
let _ = db.drop_user("test-auth-mod-successful_login-saghm", None);
let doc = doc! { "connectionStatus": 1};
let before = db.command(doc.clone(), CommandType::Suppressed, None)
.unwrap();
let info = match before.get("authInfo") {
Some(&Bson::Document(ref doc)) => doc.clone(),
_ => panic!("Invalid response for initial connectionStatus command"),
};
match info.get("authenticatedUsers") {
Some(&Bson::Array(ref vec)) => {
assert!(doc_vec_find(&vec, "user", "test-auth-mod-successful_login-saghm").is_none())
}
_ => panic!("Invalid array of authenticatedUsers for initial connectionStatus command"),
};
db.create_user(
"test-auth-mod-successful_login-saghm",
"such_secure_password",
None,
).unwrap();
db.auth(
"test-auth-mod-successful_login-saghm",
"such_secure_password",
).unwrap();
let after = db.command(doc, CommandType::Suppressed, None).unwrap();
let info = match after.get("authInfo") {
Some(&Bson::Document(ref doc)) => doc.clone(),
_ => panic!("Invalid response for subsequent connectionStatus command"),
};
let authed_users = match info.get("authenticatedUsers") {
Some(&Bson::Array(ref vec)) => vec.clone(),
_ => panic!("Invalid array of authenticatedUsers for subsequent connectionStatus command"),
};
let bson_user = doc_vec_find(
&authed_users,
"user",
"test-auth-mod-successful_login-saghm",
).unwrap();
let user = match bson_user {
Bson::Document(ref doc) => doc.clone(),
_ => panic!("Invalid auth'd user in subsequent connectionStatus response"),
};
match user.get("user") {
Some(&Bson::String(ref s)) => assert_eq!(s, "test-auth-mod-successful_login-saghm"),
_ => panic!("Invalid `user` field of auth'd user"),
};
match user.get("db") {
Some(&Bson::String(ref s)) => assert_eq!(s, "test-auth-mod-successful_login"),
_ => panic!("Invalid `db` field of auth'd user"),
};
}