def test_get_deleted(self):
     record = zdb.get_ipv4_record(0x01020304, 443, 80, 5)
     hq = zdb.host_query_from_record(record)
     delta = self.service.PutHostIPv4Record(record, self.TIMEOUT)
     delta = self.service.DelHostIPv4Record(hq, self.TIMEOUT)
     res = self.service.GetHostIPv4Record(hq, self.TIMEOUT)
     self.assertEqual(HostQueryResponse.NO_RECORD, res.status)
 def test_get_exists(self):
     record = zdb.get_ipv4_record(0x01020304, 443, 80, 5)
     delta = self.service.PutHostIPv4Record(record, self.TIMEOUT)
     self.assertIsNotNone(delta)
     hq = zdb.host_query_from_record(record)
     res = self.service.GetHostIPv4Record(hq, self.TIMEOUT)
     self.assertEqual(HostQueryResponse.SUCCESS, res.status)
     self.assertHostQueryResponseMatchesHostQuery(res, hq)
     self.assertRecordEqual(record, res.record)
 def test_insert_delete_one_ipv4(self):
     record = zdb.get_ipv4_record(0x01020304, 443, 80, 5)
     delta = self.service.PutHostIPv4Record(record, self.TIMEOUT)
     self.assertIsNotNone(delta)
     hq = zdb.host_query_from_record(record)
     delta = self.service.DelHostIPv4Record(hq, self.TIMEOUT)
     self.assertIsNotNone(delta)
     self.assertEqual(delta.delta_type, common_pb2.DT_DELETE)
     self.assertDeltaMatchesHostQuery(delta, hq)
     self.assertDeltaMatchesRecords(delta, [])