def test_get(self): user = factory.make_User() handler = ZoneHandler(user, {}) zone = factory.make_Zone() self.assertEqual( self.dehydrate_zone(zone), handler.get({"id": zone.id}))
def test_delete_as_admin_success(self): user = factory.make_admin() handler = ZoneHandler(user, {}, None) zone = factory.make_Zone() handler.delete({"id": zone.id}) zone = reload_object(zone) self.assertThat(zone, Equals(None))
def test_list(self): user = factory.make_User() handler = ZoneHandler(user, {}) factory.make_Zone() expected_zones = [ self.dehydrate_zone(zone) for zone in Zone.objects.all() ] self.assertItemsEqual(expected_zones, handler.list({}))
def test__delete_default_zone_fails(self): zone = Zone.objects.get_default_zone() user = factory.make_admin() handler = ZoneHandler(user, {}) with ExpectedException(ValidationError): handler.delete({ "id": zone.id, })
def test__delete_as_non_admin_asserts(self): user = factory.make_User() handler = ZoneHandler(user, {}) zone = factory.make_Zone() with ExpectedException(AssertionError, "Permission denied."): handler.delete({ "id": zone.id, })
def test_get(self): user = factory.make_User() handler = ZoneHandler(user, {}) zone = factory.make_Zone() for _ in range(3): factory.make_Node(zone=zone) for _ in range(3): factory.make_Device(zone=zone) for _ in range(3): factory.make_RackController(zone=zone) for _ in range(3): factory.make_RegionController(zone=zone) result = handler.get({"id": zone.id}) self.assertEqual(self.dehydrate_zone(zone), result) self.assertEquals(3, result['machines_count']) self.assertEquals(3, result['devices_count']) self.assertEquals(6, result['controllers_count'])
def test_get_query_count(self): user = factory.make_User() handler = ZoneHandler(user, {}) zone = factory.make_Zone() for _ in range(3): factory.make_Node(zone=zone) for _ in range(3): factory.make_Device(zone=zone) for _ in range(3): factory.make_RackController(zone=zone) for _ in range(3): factory.make_RegionController(zone=zone) count, _ = count_queries(handler.get, {"id": zone.id}) self.assertEqual(count, 3)