@Test public void testMultipleARecords() throws Exception { testCleanDatabase(); DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1"); DNSRecord record2 = DNSRecordUtils.createARecord("example2.domain.com", 86400L, "74.22.43.123"); DNSRecord record3 = DNSRecordUtils.createARecord("sample.domain.com", 86400L, "81.142.48.20"); dnsDao.add(Arrays.asList(record1, record2, record3)); /* * Get by name */ Collection<DNSRecord> records = dnsDao.get(record1.getName()); assertEquals(1, records.size()); DNSRecord compareRec = records.iterator().next(); assertEquals(record1.getName(), compareRec.getName()); assertEquals(Type.A, compareRec.getType()); /* * Get all types */ records = dnsDao.get(Type.ANY); assertEquals(3, records.size()); assertTrue(records.contains(record1)); assertTrue(records.contains(record2)); assertTrue(records.contains(record3)); /* * Get A only */ records = dnsDao.get(Type.A); assertEquals(3, records.size()); assertTrue(records.contains(record1)); assertTrue(records.contains(record2)); assertTrue(records.contains(record3)); /* * Get SRV only */ records = dnsDao.get(Type.SRV); assertEquals(0, records.size()); }
@Test public void testAddSingleARecords() throws Exception { testCleanDatabase(); // Add 1 record DNSRecord record = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1"); dnsDao.add(Arrays.asList(record)); Collection<DNSRecord> records = dnsDao.get(record.getName()); assertEquals(1, records.size()); DNSRecord compareRec = records.iterator().next(); assertEquals(record.getName(), compareRec.getName()); assertEquals(Type.A, compareRec.getType()); }