def test_delete(self):
user = factory.make_admin()
staticroute = factory.make_StaticRoute()
handler = StaticRouteHandler(user, {}, None)
handler.delete({"id": staticroute.id})
self.assertIsNone(
get_one(StaticRoute.objects.filter(id=staticroute.id)))
def test_get(self):
user = factory.make_User()
handler = StaticRouteHandler(user, {}, None)
staticroute = factory.make_StaticRoute()
self.assertEqual(
self.dehydrate_staticroute(staticroute),
handler.get({"id": staticroute.id}),
)
def test_list(self):
user = factory.make_User()
handler = StaticRouteHandler(user, {}, None)
for _ in range(3):
factory.make_StaticRoute()
expected_staticroutes = [
self.dehydrate_staticroute(staticroute, for_list=True)
for staticroute in StaticRoute.objects.all()
]
self.assertItemsEqual(expected_staticroutes, handler.list({}))
def test_update(self):
user = factory.make_admin()
staticroute = factory.make_StaticRoute()
handler = StaticRouteHandler(user, {}, None)
data = self.dehydrate_staticroute(staticroute)
data["metric"] = random.randint(0, 500)
handler.update(data)
self.assertThat(
StaticRoute.objects.get(id=data['id']),
MatchesStructure.byEquality(source=staticroute.source,
destination=staticroute.destination,
gateway_ip=staticroute.gateway_ip,
metric=data["metric"]))
def test_update_admin_only(self):
user = factory.make_User()
staticroute = factory.make_StaticRoute()
handler = StaticRouteHandler(user, {}, None)
data = self.dehydrate_staticroute(staticroute)
data["metric"] = random.randint(0, 500)
self.assertRaises(PermissionDenied, handler.update, data)
def test_create(self):
user = factory.make_admin()
source = factory.make_Subnet()
destination = factory.make_Subnet(
version=source.get_ipnetwork().version)
gateway_ip = factory.pick_ip_in_Subnet(source)
metric = random.randint(0, 500)
handler = StaticRouteHandler(user, {}, None)
staticroute = handler.create({
"source": source.id,
"destination": destination.id,
"gateway_ip": gateway_ip,
"metric": metric,
})
self.assertThat(
StaticRoute.objects.get(id=staticroute['id']),
MatchesStructure.byEquality(source=source,
destination=destination,
gateway_ip=gateway_ip,
metric=metric))
def test_create_admin_only(self):
user = factory.make_User()
source = factory.make_Subnet()
destination = factory.make_Subnet(
version=source.get_ipnetwork().version)
gateway_ip = factory.pick_ip_in_Subnet(source)
metric = random.randint(0, 500)
handler = StaticRouteHandler(user, {}, None)
self.assertRaises(
PermissionDenied, handler.create, {
"source": source.id,
"destination": destination.id,
"gateway_ip": gateway_ip,
"metric": metric,
})
def test_delete_admin_only(self):
user = factory.make_User()
staticroute = factory.make_StaticRoute()
handler = StaticRouteHandler(user, {}, None)
self.assertRaises(PermissionDenied, handler.delete,
{"id": staticroute.id})