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__
