@Test
public void testRemoveByRecords() throws Exception {
testCleanDatabase();
assertEquals(0, dnsDao.count());
// Add 5 record
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
DNSRecord record2 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.2");
DNSRecord record3 = DNSRecordUtils.createARecord("example2.domain.com", 86400L, "127.0.0.3");
DNSRecord record4 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example.domain.com", 86400L, 3506, 1, 1);
DNSRecord record5 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example2.domain.com", 86400L, 3506, 1, 1);
dnsDao.add(Arrays.asList(record1, record2, record3, record4, record5));
assertEquals(5, dnsDao.count());
// remove the first three records
dnsDao.remove(Arrays.asList(record1, record2, record3));
Collection<DNSRecord> records = dnsDao.get(Type.ANY);
assertEquals(2, records.size());
assertTrue(records.contains(record4));
assertTrue(records.contains(record5));
// remove the last two records
dnsDao.remove(Arrays.asList(record4, record5));
records = dnsDao.get(Type.ANY);
assertEquals(0, records.size());
}
@Test
public void testCleanDatabase() throws Exception {
Collection<DNSRecord> records = dnsDao.get(Type.ANY);
if (records != null && records.size() > 0) {
for (DNSRecord record : records) dnsDao.remove(record.getId());
}
records = dnsDao.get(Type.ANY);
assertEquals(0, records.size());
}
@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 testUpdateRecord_illegalAnyType() throws Exception {
testCleanDatabase();
assertEquals(0, dnsDao.count());
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
dnsDao.add(Arrays.asList(record1));
Collection<DNSRecord> records = dnsDao.get(Type.ANY);
assertEquals(1, records.size());
DNSRecord checkRecord = records.iterator().next();
assertEquals(record1, checkRecord);
checkRecord.setType(Type.ANY);
boolean exceptionOccured = false;
try {
dnsDao.update(checkRecord.getId(), checkRecord);
} catch (ConfigurationStoreException e) {
exceptionOccured = true;
}
assertTrue(exceptionOccured);
}
@Test
public void testRemoveByIds() throws Exception {
testCleanDatabase();
assertEquals(0, dnsDao.count());
// Add 5 record
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
DNSRecord record2 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.2");
DNSRecord record3 = DNSRecordUtils.createARecord("example2.domain.com", 86400L, "127.0.0.3");
DNSRecord record4 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example.domain.com", 86400L, 3506, 1, 1);
DNSRecord record5 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example2.domain.com", 86400L, 3506, 1, 1);
dnsDao.add(Arrays.asList(record1, record2, record3, record4, record5));
assertEquals(5, dnsDao.count());
// remove record 3
Collection<DNSRecord> records = dnsDao.get(record3.getName());
assertEquals(1, records.size());
dnsDao.remove(records.iterator().next().getId());
records = dnsDao.get(Type.ANY);
assertEquals(4, dnsDao.count());
assertTrue(records.contains(record1));
assertTrue(records.contains(record2));
assertTrue(records.contains(record4));
assertTrue(records.contains(record5));
// remove the rest
long[] ids = new long[records.size()];
int cnt = 0;
for (DNSRecord record : records) ids[cnt++] = record.getId();
dnsDao.remove(ids);
records = dnsDao.get(Type.ANY);
assertEquals(0, records.size());
}
@Test
public void testGetByType() throws Exception {
testCleanDatabase();
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
DNSRecord record2 = DNSRecordUtils.createARecord("example2.domain.com", 86400L, "127.0.0.1");
DNSRecord record3 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example.domain.com", 86400L, 3506, 1, 1);
dnsDao.add(Arrays.asList(record1, record2, record3));
/*
* By A
*/
Collection<DNSRecord> records = dnsDao.get(Type.A);
assertEquals(2, records.size());
assertTrue(records.contains(record1));
assertTrue(records.contains(record2));
/*
* By SRV
*/
records = dnsDao.get(Type.SRV);
assertEquals(1, records.size());
assertTrue(records.contains(record3));
/*
* By ANY
*/
records = dnsDao.get(Type.ANY);
assertEquals(3, records.size());
assertTrue(records.contains(record1));
assertTrue(records.contains(record2));
assertTrue(records.contains(record3));
}
@Test
public void testUpdateRecord() throws Exception {
testCleanDatabase();
assertEquals(0, dnsDao.count());
DNSRecord record1 = DNSRecordUtils.createMXRecord("example.domain.com", "127.0.0.1", 86400L, 1);
dnsDao.add(Arrays.asList(record1));
Collection<DNSRecord> records = dnsDao.get(Type.ANY);
assertEquals(1, records.size());
DNSRecord checkRecord = records.iterator().next();
assertEquals(record1, checkRecord);
checkRecord.setName("example2.domain.com.");
dnsDao.update(checkRecord.getId(), checkRecord);
records = dnsDao.get(Type.ANY);
assertEquals(1, records.size());
DNSRecord modRecord = records.iterator().next();
assertEquals(checkRecord, modRecord);
}
@Test
public void testGetByRecord() throws Exception {
testCleanDatabase();
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
DNSRecord record2 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.2");
DNSRecord record3 = DNSRecordUtils.createARecord("example2.domain.com", 86400L, "127.0.0.3");
DNSRecord record4 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example.domain.com", 86400L, 3506, 1, 1);
DNSRecord record5 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example2.domain.com", 86400L, 3506, 1, 1);
dnsDao.add(Arrays.asList(record1, record2, record3, record4, record5));
Collection<DNSRecord> records = dnsDao.get(record3.getName());
assertEquals(1, records.size());
DNSRecord checkRec = dnsDao.get(records.iterator().next().getId());
assertNotNull(checkRec);
assertEquals(checkRec, record3);
records = dnsDao.get(Type.ANY);
assertEquals(5, records.size());
long[] ids = new long[records.size()];
int cnt = 0;
for (DNSRecord record : records) ids[cnt++] = record.getId();
records = dnsDao.get(ids);
assertEquals(5, records.size());
assertTrue(records.contains(record1));
assertTrue(records.contains(record2));
assertTrue(records.contains(record3));
assertTrue(records.contains(record4));
assertTrue(records.contains(record5));
}
@Test
public void testGetByName() throws Exception {
testCleanDatabase();
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
DNSRecord record2 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.2");
DNSRecord record3 = DNSRecordUtils.createARecord("example2.domain.com", 86400L, "127.0.0.3");
DNSRecord record4 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example.domain.com", 86400L, 3506, 1, 1);
DNSRecord record5 =
DNSRecordUtils.createSRVRecord(
"_ldap_cerner._tcp.cerner.com", "example2.domain.com", 86400L, 3506, 1, 1);
dnsDao.add(Arrays.asList(record1, record2, record3, record4, record5));
Collection<DNSRecord> records = dnsDao.get(record1.getName());
assertEquals(2, records.size());
assertTrue(records.contains(record1));
assertTrue(records.contains(record2));
records = dnsDao.get(record3.getName());
assertEquals(1, records.size());
assertTrue(records.contains(record3));
records = dnsDao.get(record4.getName());
assertEquals(2, records.size());
assertTrue(records.contains(record4));
assertTrue(records.contains(record5));
records = dnsDao.get("bogus.com.");
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());
}
@Test
public void testAddCertRecord() throws Exception {
testCleanDatabase();
byte[] certData = loadCertificateData("gm2552.der");
assertNotNull(certData);
ByteArrayInputStream bais = new ByteArrayInputStream(certData);
X509Certificate cert =
(X509Certificate) CertificateFactory.getInstance("X.509").generateCertificate(bais);
DNSRecord record1 =
DNSRecordUtils.createX509CERTRecord("*****@*****.**", 86400L, cert);
dnsDao.add(Arrays.asList(record1));
Collection<DNSRecord> records = dnsDao.get(Type.CERT);
assertEquals(1, records.size());
assertEquals(record1, records.iterator().next());
}
@Test
public void testMultipleARecords_differentRdata() throws Exception {
testCleanDatabase();
DNSRecord record1 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.1");
dnsDao.add(Arrays.asList(record1));
DNSRecord record2 = DNSRecordUtils.createARecord("example.domain.com", 86400L, "127.0.0.2");
dnsDao.add(Arrays.asList(record2));
/*
* Get by name
*/
Collection<DNSRecord> records = dnsDao.get(record1.getName());
assertEquals(2, records.size());
records.contains(record1);
records.contains(record2);
}
/** {@inheritDoc} */
@Override
public Collection<DNSRecord> getDNSByType(int type) throws ConfigurationServiceException {
return dao.get(type);
}
/** {@inheritDoc} */
@Override
public Collection<DNSRecord> getDNSByRecordIds(long[] recordIds)
throws ConfigurationServiceException {
return dao.get(recordIds);
}
/** {@inheritDoc} */
@Override
public DNSRecord getDNSByRecordId(long recordId) throws ConfigurationServiceException {
return dao.get(recordId);
}
/** {@inheritDoc} */
@Override
public Collection<DNSRecord> getDNSByNameAndType(String name, int type)
throws ConfigurationServiceException {
return dao.get(name, type);
}