def test_raises_if_missing_mac(self):
user = factory.make_User()
handler = DiscoveryHandler(user, {}, None)
disco = factory.make_Discovery()
num_discoveries = Discovery.objects.count()
self.assertThat(num_discoveries, Equals(1))
with ExpectedException(HandlerPermissionError):
handler.delete_by_mac_and_ip(dict(ip=disco.ip))
def test_deletes_discovery_and_returns_number_deleted(self):
user = factory.make_admin()
handler = DiscoveryHandler(user, {}, None)
disco = factory.make_Discovery()
num_discoveries = Discovery.objects.count()
self.assertThat(num_discoveries, Equals(1))
result = handler.delete_by_mac_and_ip(
dict(ip=disco.ip, mac=disco.mac_address))
num_discoveries = Discovery.objects.count()
self.assertThat(num_discoveries, Equals(0))
self.assertThat(result, Equals(1))