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)
