def test_insert_failed(self, mock_cmdline, mock_inst, mock_cmds, mock_date,
mock_mongo):
"""Function: test_insert_failed
Description: Test with failed insert into Mongo.
Arguments:
"""
self.cmdline.argv.append("-j")
self.cmdline.argv.append("-f")
self.cmdline.argv.append("-o")
self.cmdline.argv.append(self.ofile)
self.cmdline.argv.append("-m")
self.cmdline.argv.append(self.config)
self.cmdline.argv.append("-i")
self.cmdline.argv.append(self.db_tbl)
mock_mongo.return_value = (False, "Connection error")
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
mock_date.return_value = self.setdate
with gen_libs.no_std_out():
mongo_perf.main()
self.assertTrue(filecmp.cmp(self.outfile3, self.ofile))
def test_mongo(self, mock_cmdline, mock_inst, mock_cmds, mock_date,
mock_mongo):
"""Function: test_mongo
Description: Test with mongo option.
Arguments:
"""
self.cmdline.argv.append("-j")
self.cmdline.argv.append("-f")
self.cmdline.argv.append("-o")
self.cmdline.argv.append(self.ofile)
self.cmdline.argv.append("-m")
self.cmdline.argv.append(self.config)
self.cmdline.argv.append("-i")
self.cmdline.argv.append(self.db_tbl)
mock_mongo.return_value = (True, None)
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
mock_date.return_value = self.setdate
mongo_perf.main()
self.assertTrue(filecmp.cmp(self.outfile3, self.ofile))
def test_write_file(self, mock_cmdline, mock_inst, mock_cmds, mock_date):
"""Function: test_write_file
Description: Test option to write to file.
Arguments:
"""
self.cmdline.argv.append("-j")
self.cmdline.argv.append("-o")
self.cmdline.argv.append(self.ofile)
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
mock_date.return_value = self.setdate
mongo_perf.main()
self.assertTrue(filecmp.cmp(self.outfile, self.ofile))
def test_flatten_json(self, mock_cmdline, mock_inst, mock_cmds, mock_date):
"""Function: test_flatten_json
Description: Test option to flatten JSON data structure.
Arguments:
"""
self.cmdline.argv.append("-j")
self.cmdline.argv.append("-f")
self.cmdline.argv.append("-o")
self.cmdline.argv.append(self.ofile)
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
mock_date.return_value = self.setdate
mongo_perf.main()
self.assertTrue(filecmp.cmp(self.outfile3, self.ofile))
def test_help_false(self, mock_cmdline, mock_req):
"""Function: test_help_false
Description: Test help if returns false.
Arguments:
"""
mock_cmdline.return_value = self.cmdline
mock_req.return_value = True
self.assertFalse(mongo_perf.main())
def test_arg_dir_true(self, mock_cmdline, mock_dir):
"""Function: test_arg_dir_true
Description: Test arg_dir_chk_crt if returns true.
Arguments:
"""
mock_cmdline.return_value = self.cmdline
mock_dir.return_value = True
self.assertFalse(mongo_perf.main())
def test_arg_file_false(self, mock_cmdline, mock_run):
"""Function: test_arg_file_false
Description: Test arg_file_chk if returns false.
Arguments:
"""
mock_cmdline.return_value = self.cmdline
mock_run.return_value = True
self.assertFalse(mongo_perf.main())
def test_help_true(self, mock_arg, mock_help):
"""Function: test_help_true
Description: Test help if returns true.
Arguments:
"""
mock_arg.return_value = self.args_array
mock_help.return_value = True
self.assertFalse(mongo_perf.main())
def test_arg_cond_false(self, mock_cmdline, mock_cond):
"""Function: test_arg_cond_false
Description: Test arg_cond_req if returns false.
Arguments:
"""
mock_cmdline.return_value = self.cmdline
mock_cond.return_value = False
self.assertFalse(mongo_perf.main())
def test_default_args_array(self, mock_cmdline, mock_inst, mock_popen):
"""Function: test_default_args_array
Description: Test with default options.
Arguments:
"""
mock_inst.return_value = self.server
mock_cmdline.side_effect = [self.cmdline, self.subproc]
mock_popen.return_value = self.subproc
self.assertFalse(mongo_perf.main())
def test_set_default_args(self, mock_cmdline, mock_run):
"""Function: test_set_default_args
Description: Test setting default arguments.
Arguments:
"""
self.cmdline.argv.append("-j")
mock_cmdline.return_value = self.cmdline
mock_run.return_value = True
self.assertFalse(mongo_perf.main())
def test_arg_req_true(self, mock_arg, mock_help, mock_req):
"""Function: test_arg_req_true
Description: Test arg_require if returns true.
Arguments:
"""
mock_arg.return_value = self.args_array
mock_help.return_value = False
mock_req.return_value = True
self.assertFalse(mongo_perf.main())
def test_help_true(self, mock_cmdline):
"""Function: test_help_true
Description: Test help if returns true.
Arguments:
"""
self.cmdline.argv.append("-h")
mock_cmdline.return_value = self.cmdline
with gen_libs.no_std_out():
self.assertFalse(mongo_perf.main())
def test_replica_set(self, mock_cmdline, mock_inst, mock_cmds, mock_date):
"""Function: test_replica_set
Description: Test connecting to Mongo replica set.
Arguments:
"""
self.cmdline.argv.append("-j")
self.cmdline.argv.append("-f")
self.cmdline.argv.append("-o")
self.cmdline.argv.append(self.ofile)
self.cmdline.argv.append("-m")
self.cmdline.argv.append(self.config2)
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
mock_date.return_value = self.setdate
mongo_perf.main()
self.assertTrue(filecmp.cmp(self.outfile3, self.ofile))
def test_arg_cond_false(self, mock_arg, mock_help, mock_req, mock_cond):
"""Function: test_arg_cond_false
Description: Test arg_cond_req if returns false.
Arguments:
"""
mock_arg.return_value = self.args_array
mock_help.return_value = False
mock_req.return_value = False
mock_cond.return_value = False
self.assertFalse(mongo_perf.main())
def test_suppress(self, mock_cmdline, mock_inst, mock_cmds):
"""Function: test_suppress
Description: Test with suppression.
Arguments:
"""
self.cmdline.argv.append("-j")
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
self.assertFalse(mongo_perf.main())
def test_arg_dir_true(self, mock_arg, mock_help, mock_req, mock_cond,
mock_dir):
"""Function: test_arg_dir_true
Description: Test arg_dir_chk_crt if returns true.
Arguments:
"""
mock_arg.return_value = self.args_array
mock_help.return_value = False
mock_req.return_value = False
mock_cond.return_value = True
mock_dir.return_value = True
self.assertFalse(mongo_perf.main())
def test_arg_file_true(self, mock_arg, mock_help):
"""Function: test_arg_file_true
Description: Test arg_file_chk if returns true.
Arguments:
"""
mock_arg.arg_parse2.return_value = self.args_array
mock_help.return_value = False
mock_arg.arg_require.return_value = False
mock_arg.arg_cond_req.return_value = True
mock_arg.arg_dir_chk_crt.return_value = False
mock_arg.arg_file_chk.return_value = True
self.assertFalse(mongo_perf.main())
def test_no_suppress(self, mock_cmdline, mock_inst, mock_cmds):
"""Function: test_no_suppress
Description: Test with no suppression.
Arguments:
"""
self.cmdline.argv.append("-j")
self.cmdline.argv.remove("-z")
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
with gen_libs.no_std_out():
self.assertFalse(mongo_perf.main())
def test_programlock_id(self, mock_arg, mock_help, mock_lock):
"""Function: test_programlock_id
Description: Test ProgramLock with flavor ID.
Arguments:
"""
mock_arg.arg_parse2.return_value = self.args_array6
mock_help.return_value = False
mock_arg.arg_require.return_value = False
mock_arg.arg_cond_req.return_value = True
mock_arg.arg_dir_chk_crt.return_value = False
mock_arg.arg_file_chk.return_value = False
mock_lock.return_value = self.proglock
self.assertFalse(mongo_perf.main())
def test_set_default_args(self, mock_arg, mock_help, mock_run):
"""Function: test_set_default_args
Description: Test setting default arguments.
Arguments:
"""
mock_arg.arg_parse2.return_value = self.args_array2
mock_arg.arg_add_def.side_effect = [self.args_array3, self.args_array3]
mock_help.return_value = False
mock_arg.arg_require.return_value = False
mock_arg.arg_cond_req.return_value = True
mock_arg.arg_dir_chk_crt.return_value = False
mock_arg.arg_file_chk.return_value = False
mock_run.return_value = True
self.assertFalse(mongo_perf.main())
def test_conn_fail_suppress(self, mock_cmdline, mock_inst, 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"
self.cmdline.argv.append("-j")
self.cmdline.argv.append("-w")
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
self.assertFalse(mongo_perf.main())
def test_connection_fail(self, mock_cmdline, mock_inst, mock_cmds):
"""Function: test_connection_fail
Description: Test with failed mongo connection.
Arguments:
"""
self.server.status = False
self.server.err_msg = "Error connection message"
self.cmdline.argv.append("-j")
mock_cmds.return_value = self.results
mock_inst.return_value = self.server
mock_cmdline.return_value = self.cmdline
with gen_libs.no_std_out():
self.assertFalse(mongo_perf.main())
def test_programlock_false(self, mock_arg, mock_help, mock_lock):
"""Function: test_programlock_false
Description: Test with ProgramLock returns False.
Arguments:
"""
mock_arg.arg_parse2.return_value = self.args_array
mock_help.return_value = False
mock_arg.arg_require.return_value = False
mock_arg.arg_cond_req.return_value = True
mock_arg.arg_dir_chk_crt.return_value = False
mock_arg.arg_file_chk.return_value = False
mock_lock.side_effect = \
mongo_perf.gen_class.SingleInstanceException
with gen_libs.no_std_out():
self.assertFalse(mongo_perf.main())
