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({}))
