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, [])
def test_insert_delete_one_domain(self): domain = "davidadrian.org" ip = 0x8dd47859 record = zdb.get_domain_record(domain, 80, 20, 5) delta = self.service.PutHostDomainRecord(record, self.TIMEOUT) self.assertIsNotNone(delta) self.assertDeltaMatchesRecords(delta, [record]) self.assertEqual(delta.delta_type, common_pb2.DT_UPDATE) host_query = zdb.host_query_from_record(record) delta = self.service.DelHostDomainRecord(host_query, self.TIMEOUT) self.assertIsNotNone(delta) self.assertEqual(delta.delta_type, common_pb2.DT_DELETE) self.assertDeltaMatchesHostQuery(delta, host_query) self.assertDeltaMatchesRecords(delta, [])
def test_insert_many_delete_many(self): domain = "a.com" records = list() for port in range(1,10): record = zdb.get_domain_record(domain, port, 20, 5) records.append(record) delta = self.service.PutHostDomainRecord(record, self.TIMEOUT) self.assertIsNotNone(delta) self.assertDeltaMatchesRecords(delta, records) self.assertEqual(delta.delta_type, common_pb2.DT_UPDATE) for idx, record in enumerate(records): hq = zdb.host_query_from_record(record) delta = self.service.DelHostDomainRecord(hq, self.TIMEOUT) self.assertIsNotNone(delta) self.assertEqual(delta.delta_type, common_pb2.DT_DELETE) self.assertDeltaMatchesRecords(delta, records[idx + 1:])