def test_get(self): user = factory.make_User() handler = SubnetHandler(user, {}) subnet = factory.make_Subnet() expected_data = self.dehydrate_subnet(subnet) result = handler.get({"id": subnet.id}) self.assertThat(result, Equals(expected_data))
def test_get_handles_null_dns_servers(self): user = factory.make_User() handler = SubnetHandler(user, {}, None) subnet = factory.make_Subnet() subnet.dns_servers = None subnet.save() expected_data = self.dehydrate_subnet(subnet) result = handler.get({"id": subnet.id}) self.assertThat(result, Equals(expected_data))