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:])