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())