def test_list(self): user = factory.make_User() handler = ServiceHandler(user, {}, None) node = factory.make_RackController() factory.make_Service(node) expected_services = [ self.dehydrate_service(service, for_list=True) for service in Service.objects.all() ] self.assertGreater(len(expected_services), 0) self.assertItemsEqual(expected_services, handler.list({}))
def test_get(self): user = factory.make_User() handler = ServiceHandler(user, {}, None) node = factory.make_RackController() service = factory.make_Service(node) self.assertEqual(self.dehydrate_service(service), handler.get({"id": service.id}))