use std::env; use std::path::PathBuf; use std::sync::Arc; use std::thread; use mongo_driver::client::{ClientPool,SslOptions,Uri}; #[test] fn test_new_pool_pop_client_and_borrow_collection() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let pool = ClientPool::new(uri.clone(), None); assert_eq!(pool.get_uri(), &uri); // Pop a client and get a database and collection let client = pool.pop(); pool.pop(); let database = client.get_database("rust_test"); assert_eq!("rust_test", database.get_name().to_mut()); let collection = client.get_collection("rust_test", "items"); assert_eq!("items", collection.get_name().to_mut()); } #[test] fn test_new_pool_pop_client_and_take_collection() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let pool = ClientPool::new(uri.clone(), None); assert_eq!(pool.get_uri(), &uri); // Pop a client and take collection let client = pool.pop(); let collection = client.take_collection("rust_test", "items"); assert_eq!("items", collection.get_name().to_mut()); } #[test] fn test_new_pool_pop_client_and_take_database_and_collection() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let pool = ClientPool::new(uri.clone(), None); assert_eq!(pool.get_uri(), &uri); // Pop a client and take database and collection let client = pool.pop(); let database = client.take_database("rust_test"); assert_eq!("rust_test", database.get_name().to_mut()); let collection = database.take_collection("items"); assert_eq!("items", collection.get_name().to_mut()); } #[test] fn test_new_pool_and_pop_client_in_threads() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let pool = Arc::new(ClientPool::new(uri, None)); let pool1 = pool.clone(); let guard1 = thread::spawn(move || { let client = pool1.pop(); client.get_collection("test", "items"); }); let pool2 = pool.clone(); let guard2 = thread::spawn(move || { let client = pool2.pop(); client.get_collection("test", "items"); }); guard1.join().unwrap(); guard2.join().unwrap(); } #[test] fn test_get_server_status() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let pool = ClientPool::new(uri, None); let client = pool.pop(); let status = client.get_server_status(None).unwrap(); assert!(status.contains_key("host")); assert!(status.contains_key("version")); } #[test] fn test_read_command_with_opts() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let pool = ClientPool::new(uri, None); let client = pool.pop(); let db_name = "rust_driver_test"; let coll_name = "test_read_command_with_opts"; let collection = client.get_collection(db_name, coll_name); let document = doc! { "key_1" => "Value 1", "key_2" => "kācaṃ śaknomyattum; nopahinasti mām. \u{0}" }; collection.insert(&document, None).expect("Could not insert document"); let status = client.read_command_with_opts(db_name, &doc!{"collStats": coll_name}, None, None).unwrap(); assert!(status.contains_key("totalIndexSize")); assert!(status.contains_key("storageSize")); } #[test] fn test_new_pool_with_ssl_options() { let uri = Uri::new("mongodb://localhost:27017/").unwrap(); let ssl_options = SslOptions::new( Some(PathBuf::from("./README.md")), Some("password".to_string()), Some(PathBuf::from("./README.md")), Some(PathBuf::from("./README.md")), Some(PathBuf::from("./README.md")), false ); assert!(ssl_options.is_ok()); ClientPool::new(uri, Some(ssl_options.unwrap())); } #[test] fn test_ssl_options_nonexistent_file() { assert!(SslOptions::new( Some(PathBuf::from("/tmp/aaaaa.aa")), Some("password".to_string()), Some(PathBuf::from("/tmp/aaaaa.aa")), Some(PathBuf::from("/tmp/aaaaa.aa")), Some(PathBuf::from("/tmp/aaaaa.aa")), false ).is_err()); } // SSL tests below are currently tested on a private replica set, will be skipped if you set // the env var SKIP_SSL_CONNECTION_TESTS to true. #[test] fn test_ssl_connection_success() { if env::var("SKIP_SSL_CONNECTION_TESTS") == Ok("true".to_string()) { return } let uri = Uri::new(env::var("MONGO_RUST_DRIVER_SSL_URI").unwrap()).unwrap(); let pem_file = PathBuf::from(env::var("MONGO_RUST_DRIVER_SSL_PEM_FILE").unwrap()); let ca_file = PathBuf::from(env::var("MONGO_RUST_DRIVER_SSL_CA_FILE").unwrap()); let ssl_options = SslOptions::new( Some(pem_file), None, Some(ca_file), None, None, false ).unwrap(); let pool = ClientPool::new(uri, Some(ssl_options)); let client = pool.pop(); let database = client.get_database("admin"); let result = database.command_simple(doc!{"ping" => 1}, None).unwrap(); assert!(result.contains_key("ok")); }