def test_write_file(self, mock_inst, mock_disconn, mock_cmds, mock_date):
"""Function: test_write_file
Description: Test option to write to file.
Arguments:
"""
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_disconn.return_value = True
mock_date.return_value = self.setdate
mongo_perf.run_program(self.args_array3,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list)
self.assertTrue(filecmp.cmp(self.outfile, self.ofile))
def test_flatten_json(self, mock_inst, mock_disconn, mock_cmds, mock_date):
"""Function: test_flatten_json
Description: Test option to flatten JSON data structure.
Arguments:
"""
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_disconn.return_value = True
mock_date.return_value = self.setdate
mongo_perf.run_program(self.args_array5,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list)
self.assertTrue(filecmp.cmp(self.outfile3, self.ofile))
def test_run_program(self, mock_inst, mock_disconn, mock_cfg):
"""Function: test_run_program
Description: Test run_program function.
Arguments:
"""
mock_inst.return_value = self.server
mock_disconn.return_value = True
mock_cfg.return_value = self.cfg
self.assertFalse(
mongo_perf.run_program(self.args_array3, self.func_dict))
def test_mongo(self, mock_inst, mock_cfg, mock_disconn):
"""Function: test_mongo
Description: Test with mongo option.
Arguments:
"""
mock_inst.return_value = self.server
mock_cfg.side_effect = [self.cfg, True]
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array, self.func_dict))
def test_replica_set(self, mock_inst, mock_cfg, mock_disconn):
"""Function: test_replica_set
Description: Test connecting to Mongo replica set.
Arguments:
"""
self.cfg.repset = "replicasetname"
self.cfg.repset_hosts = self.repset_list
mock_inst.return_value = self.server
mock_cfg.side_effect = [self.cfg, True]
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array, self.func_dict))
def test_no_auth_mech(self, mock_inst, mock_cfg, mock_disconn):
"""Function: test_no_auth_mech
Description: Test with no authorization mechanism setting.
Arguments:
"""
self.cfg.repset = "replicasetname"
self.cfg.repset_hosts = self.repset_list
mock_inst.return_value = self.server
mock_cfg.side_effect = [self.cfg, True]
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array, self.func_dict))
def test_conn_fail_suppress(self, mock_inst, mock_cfg, mock_disconn):
"""Function: test_conn_fail_suppress
Description: Test with failed connection with suppression.
Arguments:
"""
self.server.status = False
self.server.err_msg = "Error Connection Message"
mock_inst.return_value = self.server
mock_cfg.side_effect = [self.cfg, True]
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array4, self.func_dict))
def test_no_rep_arg(self, mock_inst, mock_cfg, mock_disconn):
"""Function: test_no_rep_arg
Description: Test with not passing rep_arg argument.
Arguments:
"""
self.cfg2.repset = "replicasetname"
self.cfg2.repset_hosts = self.repset_list
mock_inst.return_value = self.server
mock_cfg.side_effect = [self.cfg2, self.cfg2]
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array, self.func_dict))
def test_default_args_array(self, mock_inst, mock_disconn, mock_popen):
"""Function: test_default_args_array
Description: Test with default options.
Arguments:
"""
mock_inst.return_value = self.server
mock_disconn.return_value = True
mock_popen.return_value = self.subproc
self.assertFalse(
mongo_perf.run_program(self.args_array,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_json(self, mock_inst, mock_disconn, mock_cmds):
"""Function: test_json
Description: Test with JSON option.
Arguments:
"""
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array2,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_replica_set(self, mock_inst, mock_disconn, mock_popen):
"""Function: test_replica_set
Description: Test connecting to Mongo replica set.
Arguments:
"""
mock_popen.return_value = self.subproc
mock_inst.return_value = self.server
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array7,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_no_suppress(self, mock_inst, mock_disconn, mock_cmds):
"""Function: test_no_suppress
Description: Test with no suppression.
Arguments:
"""
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_disconn.return_value = True
with gen_libs.no_std_out():
self.assertFalse(
mongo_perf.run_program(self.args_array8,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_mongo(self, mock_inst, mock_disconn, mock_popen, mock_cmd,
mock_mongo):
"""Function: test_mongo
Description: Test with mongo option.
Arguments:
"""
mock_mongo.return_value = (True, None)
mock_cmd.return_value = self.results
mock_popen.return_value = self.subproc
mock_inst.return_value = self.server
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array6,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_conn_fail_suppress(self, mock_inst, mock_disconn, mock_cmds):
"""Function: test_conn_fail_suppress
Description: Test with failed conn with suppression.
Arguments:
"""
self.server.status = False
self.server.err_msg = "Error connection message"
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_disconn.return_value = True
self.assertFalse(
mongo_perf.run_program(self.args_array2a,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_connection_fail(self, mock_inst, mock_disconn, mock_cmds):
"""Function: test_connection_fail
Description: Test with failed connection to mongo.
Arguments:
"""
self.server.status = False
self.server.err_msg = "Error connection message"
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_disconn.return_value = True
with gen_libs.no_std_out():
self.assertFalse(
mongo_perf.run_program(self.args_array2,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
def test_insert_fail(self, mock_inst, mock_disconn, mock_popen, mock_cmd,
mock_mongo):
"""Function: test_insert_fail
Description: Test with failed insert into Mongo.
Arguments:
"""
mock_mongo.return_value = (False, "Connection error")
mock_cmd.return_value = self.results
mock_popen.return_value = self.subproc
mock_inst.return_value = self.server
mock_disconn.return_value = True
with gen_libs.no_std_out():
self.assertFalse(
mongo_perf.run_program(self.args_array6,
self.func_dict,
req_arg=self.req_arg_list,
opt_arg=self.opt_arg_list))
