def test_returns_many_networks(self):
subnets = [factory.make_Subnet() for _ in range(random.randint(1, 5))]
expected = [str(subnet.cidr) for subnet in subnets]
# Note: This test was seen randomly failing because the networks were
# in an unexpected order...
self.assertItemsEqual(expected, get_trusted_networks())
def test_returns_single_network(self):
subnet = factory.make_Subnet()
expected = [str(subnet.cidr)]
self.assertEqual(expected, get_trusted_networks())
def test_returns_no_networks_if_not_allow_dns(self):
factory.make_Subnet(allow_dns=False)
subnet_allowed = factory.make_Subnet(allow_dns=True)
expected = [str(subnet_allowed.cidr)]
self.assertEqual(expected, get_trusted_networks())
def test_returns_empty_string_if_no_networks(self):
self.assertEqual([], get_trusted_networks())
