@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());
}
