def test_list(self): owner = factory.make_User() handler = DeviceHandler(owner, {}) device = self.make_device_with_ip_address(owner=owner) self.assertItemsEqual( [self.dehydrate_device(device, owner, for_list=True)], handler.list({}))
def test_list_ignores_devices_with_parents(self): owner = factory.make_User() handler = DeviceHandler(owner, {}) device = self.make_device_with_ip_address(owner=owner) # Create a node. factory.make_Node(owner=owner) self.assertItemsEqual( [self.dehydrate_device(device, owner, for_list=True)], handler.list({}))
def test_list_returns_devices_only_viewable_by_user(self): user = factory.make_User() # Create another user. factory.make_User() device = self.make_device_with_ip_address(owner=user) # Create another device not ownered by user. self.make_device_with_ip_address() handler = DeviceHandler(user, {}) self.assertItemsEqual([ self.dehydrate_device(device, user, for_list=True), ], handler.list({}))
def test_list_ignores_nodes(self): owner = factory.make_User() handler = DeviceHandler(owner, {}, None) device = self.make_device_with_ip_address(owner=owner) # Create a device with parent. node = factory.make_Node(owner=owner) device_with_parent = self.make_device_with_ip_address(owner=owner) device_with_parent.parent = node device_with_parent.save() self.assertItemsEqual( [self.dehydrate_device(device, owner, for_list=True)], handler.list({}))