def test_coll_passed2(self, mock_mongo):
"""Function: test_coll_passed2
Description: Test with coll passed.
Arguments:
"""
mock_mongo.return_value = {
"db_name": RepSetColl(),
"test": {
"coll_name": True
}
}
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset,
coll=self.coll,
db_auth=self.db_auth,
db=self.dbs,
auth=True,
use_uri=self.use_uri)
mongo.connect()
self.assertEqual(mongo.coll, self.coll)
def test_no_auth_mech(self, mock_mongo):
"""Function: test_no_auth_mech
Description: Test with no authenticate mechanism passed.
Arguments:
"""
mock_mongo.return_value = {
"db_name": RepSetColl(),
"test": {
"coll_name": True
}
}
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset,
coll=self.coll,
db_auth=self.db_auth,
db=self.dbs,
auth=True,
use_uri=self.use_uri)
self.assertEqual(mongo.connect(), (True, None))
def setUp(self):
"""Function: setUp
Description: Initialization for unit testing.
Arguments:
"""
self.base_dir = "test/integration"
self.config_dir = os.path.join(self.base_dir, "config")
self.config_name = "mongo"
self.cfg = gen_libs.load_module(self.config_name, self.config_dir)
self.database = "admin"
self.coll = "system.users"
self.mongo = mongo_class.RepSetColl(
self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
use_uri=self.cfg.use_uri,
auth=self.cfg.auth,
use_arg=self.cfg.use_arg,
auth_db=self.cfg.auth_db,
conf_file=self.cfg.conf_file,
db=self.database,
coll=self.coll,
ssl_client_ca=self.cfg.ssl_client_ca,
ssl_client_key=self.cfg.ssl_client_key,
ssl_client_cert=self.cfg.ssl_client_cert,
ssl_client_phrase=self.cfg.ssl_client_phrase)
self.mongo.connect()
def test_db_passed(self, mock_mongo):
"""Function: test_db_passed
Description: Test with db passed.
Arguments:
"""
mock_mongo.return_value = {
"db_name": RepSetColl(),
"MyDatabase": {
"coll_name": True
}
}
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset,
coll=self.coll,
db_auth=self.db_auth,
auth=True,
use_uri=self.use_uri,
db=self.dbn)
self.assertEqual(mongo.connect(), (True, None))
def test_auth_true(self, mock_mongo):
"""Function: test_auth_true
Description: Test with auth set to true.
Arguments:
"""
mock_mongo.return_value = {
"db_name": RepSetColl(),
"test": {
"coll_name": True
}
}
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset,
coll=self.coll,
db_auth=self.db_auth,
db=self.dbs,
auth=True)
self.assertEqual(mongo.connect(), (True, None))
def test_ssl_all_phrase(self):
"""Function: test_ssl_all_phrase
Description: Test with all ssl arguments and phrase present.
Arguments:
"""
mongo = mongo_class.RepSetColl(
self.name,
self.user,
self.japd,
self.host,
self.port,
conf_file=self.conf_file,
ssl_client_key=self.ssl_client_key,
ssl_client_cert=self.ssl_client_cert,
ssl_client_ca=self.ssl_client_ca,
ssl_client_phrase=self.ssl_client_phrase,
coll=self.coll,
repset=self.repset)
self.assertEqual((mongo.ssl_client_key, mongo.ssl_client_cert,
mongo.ssl_client_ca, mongo.ssl_client_phrase),
(self.ssl_client_key, self.ssl_client_cert,
self.ssl_client_ca, self.ssl_client_phrase))
def test_no_repset_hosts_attr(self):
"""Function: test_no_repset_hosts_attr
Description: Test no repset_hosts attribute passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name, self.user, self.japd,
self.host, self.port)
self.assertFalse(mongo.repset_hosts)
def test_db_coll_attr(self):
"""Function: test_db_coll_attr
Description: Test db_coll attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port)
self.assertFalse(mongo.db_coll)
def test_default_uri(self):
"""Function: test_default_uri
Description: Test using the default uri setting.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port)
self.assertFalse(mongo.use_uri)
def test_default_auth_db(self):
"""Function: test_default_auth_db
Description: Test using the default auth_db setting.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port)
self.assertEqual(mongo.auth_db, self.cfg.auth_db)
def test_config_attr(self):
"""Function: test_config_attr
Description: Test setting the config attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port)
self.assertEqual(mongo.config, self.config2)
def test_query(self):
"""Function: test_query
Description: Test method.
Arguments:
"""
mongo = mongo_class.RepSetColl(
self.name, self.user, self.japd, self.host, self.port,
repset=self.repset)
mongo.db_coll = CollOpts()
self.assertTrue(mongo.coll_options())
def test_no_query(self):
"""Function: test_no_query
Description: Test with no query command.
Arguments:
"""
mongo = mongo_class.RepSetColl(
self.name, self.user, self.japd, self.host, self.port,
repset=self.repset)
mongo.db_coll = CollFind()
self.assertTrue(mongo.coll_find1())
def test_db_attr2(self):
"""Function: test_db_attr2
Description: Test db attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port)
self.assertEqual(mongo.db, "test")
def test_no_db_attr(self):
"""Function: test_no_db_attr
Description: Test no db attribute passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
self.assertEqual(mongo.db, "test")
def test_repset_hosts_attr(self):
"""Function: test_repset_hosts_attr
Description: Test repset_hosts attribute passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset_hosts=self.repset_hosts)
self.assertEqual(mongo.repset_hosts, self.repset_hosts)
def test_using_conf_file(self):
"""Function: test_using_conf_file
Description: Test using the conf_file connection.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
conf_file="myfile")
self.assertEqual(mongo.conf_file, "myfile")
def test_using_auth(self):
"""Function: test_using_auth
Description: Test using the auth connection.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
auth=False)
self.assertFalse(mongo.auth)
def test_using_auth_db(self):
"""Function: test_using_auth_db
Description: Test using the auth_db attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
auth_db=self.cfg.auth_db)
self.assertEqual(mongo.auth_db, self.cfg.auth_db)
def test_using_uri(self):
"""Function: test_using_uri
Description: Test using the uri connection.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
use_uri=True)
self.assertTrue(mongo.use_uri)
def test_config_attr3(self):
"""Function: test_config_attr3
Description: Test with SCRAM-SHA-1 setting.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
auth_mech=self.auth_mech2)
self.assertEqual(mongo.config, self.config2)
def test_config_attr2(self):
"""Function: test_config_attr2
Description: Test with MONGODB-CR setting.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.cfg.name,
self.cfg.user,
self.cfg.japd,
host=self.cfg.host,
port=self.cfg.port,
auth_mech=self.auth_mech)
self.assertEqual(mongo.config, self.config)
def test_db_auth_conn_attr(self):
"""Function: test_db_auth_conn_attr
Description: Test db_auth attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset_hosts=self.repset_hosts)
self.assertFalse(mongo.db_auth)
def test_no_conf_file_attr(self):
"""Function: test_no_conf_file_attr
Description: Test no conf_file attribute passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
self.assertFalse(mongo.conf_file)
def test_no_auth_uri(self):
"""Function: test_no_auth_uri
Description: Test with auth and no uri present.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
self.assertFalse(mongo.use_uri)
def test_using_no_auth_db(self):
"""Function: test_using_no_auth_db
Description: Test using no auth_db attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
self.assertEqual(mongo.auth_db, "admin")
def test_db_not_passed(self):
"""Function: test_db_not_passed
Description: Test with db not passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(
self.cfg.name, self.cfg.user, self.cfg.japd, host=self.cfg.host,
port=self.cfg.port, auth=True, use_arg=self.cfg.use_arg,
auth_db=self.cfg.auth_db, conf_file=self.cfg.conf_file,
repset=self.cfg.repset, coll=self.coll, db_auth=self.cfg.auth_db)
self.assertEqual(mongo.connect(), (True, None))
def test_no_auth_attr(self):
"""Function: test_no_auth_attr
Description: Test no auth attribute passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
self.assertTrue(mongo.auth)
def test_conn_list_attr(self):
"""Function: test_conn_list_attr
Description: Test setting the conn_list attribute.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
self.assertEqual(mongo.conn_list, self.conn_list)
def test_no_qry(self):
"""Function: test_no_qry
Description: Test with no query passed.
Arguments:
"""
mongo = mongo_class.RepSetColl(self.name,
self.user,
self.japd,
self.host,
self.port,
repset=self.repset)
mongo.db_coll = CollCnt()
self.assertEqual(mongo.coll_cnt(), 10)