def test_list_ignores_nodes_that_arent_switches(self): owner = factory.make_User() handler = SwitchHandler(owner, {}) factory.make_Device(owner=owner) factory.make_Machine(owner=owner) self.assertItemsEqual([], handler.list({}))
def test_get_no_switch(self): owner = factory.make_User() handler = SwitchHandler(owner, {}) device = factory.make_Device(owner=owner) self.assertRaises(HandlerDoesNotExistError, handler.get, {"system_id": device.system_id})