def setUp(self): mock.reset_mocks() self.query_device = mock.MagicMock() mock.wrapper.session.query.side_effect = { Device: self.query_device }.__getitem__
def setUp(self): mock.reset_mocks() self.query_service = mock.MagicMock() self.query_workload = mock.MagicMock() mock.wrapper.session.query.side_effect = { Service: self.query_service, Workload: self.query_workload, }.__getitem__
def setUp(self): mock.reset_mocks() self.query_miner = mock.MagicMock() self.query_service = mock.MagicMock() mock.wrapper.session.query.side_effect = { Miner: self.query_miner, Service: self.query_service }.__getitem__
def setUp(self): mock.reset_mocks() self.query_device = mock.MagicMock() self.query_file = mock.MagicMock() file.func = self.sqlalchemy_func = mock.MagicMock() self.query_func_count = mock.MagicMock() mock.wrapper.session.query.side_effect = { Device: self.query_device, File: self.query_file, self.sqlalchemy_func.count(): self.query_func_count, }.__getitem__
def setUp(self): mock.reset_mocks() self.query_service = mock.MagicMock() service.func = self.sqlalchemy_func = mock.MagicMock() self.query_func_count = mock.MagicMock() self.query_bruteforce = mock.MagicMock() self.query_miner = mock.MagicMock() mock.wrapper.session.query.side_effect = { Service: self.query_service, Bruteforce: self.query_bruteforce, Miner: self.query_miner, self.sqlalchemy_func.count(): self.query_func_count, }.__getitem__
def setUp(self): mock.reset_mocks() self.query_device = mock.MagicMock() self.query_hardware = mock.MagicMock() self.query_file = mock.MagicMock() self.query_workload = mock.MagicMock() self.query_service = mock.MagicMock() device.func = self.sqlalchemy_func = mock.MagicMock() self.query_func_count = mock.MagicMock() mock.wrapper.session.query.side_effect = { Device: self.query_device, Hardware: self.query_hardware, File: self.query_file, Workload: self.query_workload, Service: self.query_service, self.sqlalchemy_func.count(): self.query_func_count, }.__getitem__
def test__check_exists(self): elements = [ "cpu1", "cpu2", "gpu1", "gpu2", "mainboard", "ram1", "ram2", "disk1", "disk2", "powerPack", "case", "cooler1", "cooler2", None, ] expected_results = [ { "error": "cpu_not_in_inventory" }, { "error": "cpu_not_in_inventory" }, { "error": "gpu_not_in_inventory" }, { "error": "gpu_not_in_inventory" }, { "error": "mainboard_not_in_inventory" }, { "error": "ram_not_in_inventory" }, { "error": "ram_not_in_inventory" }, { "error": "disk_not_in_inventory" }, { "error": "disk_not_in_inventory" }, { "error": "powerPack_not_in_inventory" }, { "error": "case_not_in_inventory" }, { "error": "processorCooler_not_in_inventory" }, { "error": "processorCooler_not_in_inventory" }, {}, ] def build_summary_response(missing): out = {} for element in elements: if element == missing: continue out[element] = out.get(element, 0) + 1 return {"elements": out} for missing_element, expected_result in zip(elements, expected_results): mock.m.contact_microservice.return_value = build_summary_response( missing_element) actual_result = game_content.check_exists( "super", { "cpu": ["cpu1", "cpu2"], "ram": ["ram1", "ram2"], "gpu": ["gpu1", "gpu2"], "disk": ["disk1", "disk2"], "mainboard": "mainboard", "powerPack": "powerPack", "case": "case", "processorCooler": ["cooler1", "cooler2"], }, ) self.assertEqual((not expected_result, expected_result), actual_result) mock.m.contact_microservice.assert_called_with( "inventory", ["inventory", "summary"], {"owner": "super"}) mock.reset_mocks()
def setUp(self): mock.reset_mocks()
def setUp(self): mock.reset_mocks() self.query_inventory = mock.MagicMock() mock.wrapper.session.query.side_effect = {inventory.Inventory: self.query_inventory}.__getitem__