def runTest(self, export_password_env_mock, batch_process_mock, filename_mock, backup_message_mock, server_mock): class TestMockServer(): def __init__(self, name, host, port, id, username, maintenance_db): self.name = name self.host = host self.port = port self.id = id self.username = username self.maintenance_db = maintenance_db self.server_id = parent_node_dict["server"][-1]["server_id"] mock_obj = TestMockServer(self.class_params['name'], self.class_params['host'], self.class_params['port'], self.server_id, self.class_params['username'], self.class_params['database']) mock_result = server_mock.query.filter_by.return_value mock_result.first.return_value = mock_obj filename_mock.return_value = self.params['file'] batch_process_mock.return_value.id = 140391 batch_process_mock.return_value.set_env_variables = MagicMock( return_value=True) batch_process_mock.return_value.start = MagicMock(return_value=True) backup_message_mock.message = 'test' batch_process_mock.return_value.desc = backup_message_mock export_password_env_mock.return_value = True server_response = server_utils.connect_server(self, self.server_id) if server_response["info"] == "Server connected.": db_owner = server_response['data']['user']['name'] self.data = database_utils.get_db_data(db_owner) if hasattr(self, 'server_min_version') and \ server_response["data"]["version"] < \ self.server_min_version: self.skipTest(self.message) url = self.url.format(self.server_id) # Create the backup job response = self.tester.post(url, data=json.dumps(self.params), content_type='html/json') self.assertEqual(response.status_code, 200) self.assertTrue(backup_message_mock.called) self.assertTrue(batch_process_mock.called) if self.expected_cmd_opts: for opt in self.expected_cmd_opts: self.assertIn(opt, batch_process_mock.call_args_list[0][1]['args']) if self.not_expected_cmd_opts: for opt in self.not_expected_cmd_opts: self.assertNotIn( opt, batch_process_mock.call_args_list[0][1]['args'])
def runTest(self, export_password_env_mock, batch_process_mock, message_mock, server_mock): self.server_id = parent_node_dict["database"][-1]["server_id"] self.db_id = parent_node_dict["database"][-1]["db_id"] url = self.url.format(self.server_id, self.db_id) class TestMockServer(): def __init__(self, host, port, id, username): self.host = host self.port = port self.id = id self.username = username mock_obj = TestMockServer(self.class_params['host'], self.class_params['port'], self.server_id, self.class_params['username'] ) mock_result = server_mock.query.filter_by.return_value mock_result.first.return_value = mock_obj batch_process_mock.set_env_variables = MagicMock( return_value=True ) batch_process_mock.start = MagicMock( return_value=True ) export_password_env_mock.return_value = True server_response = server_utils.connect_server(self, self.server_id) if server_response["info"] == "Server connected.": db_owner = server_response['data']['user']['name'] self.data = database_utils.get_db_data(db_owner) self.db_name = self.data['name'] # Create the backup job response = self.tester.post(url, data=json.dumps(self.params), content_type='html/json') self.assertEqual(response.status_code, 200) self.assertTrue(message_mock.called) self.assertTrue(batch_process_mock.called) if self.expected_cmd_opts: for opt in self.expected_cmd_opts: self.assertIn(opt, batch_process_mock.call_args_list[0][1]['args'])