// -*- Java -*-
//
// Copyright (c) 2005, Matthew J. Rutherford <rutherfo@cs.colorado.edu>
// Copyright (c) 2005, University of Colorado at Boulder
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
//
// * Redistributions in binary form must reproduce the above copyright
// notice, this list of conditions and the following disclaimer in the
// documentation and/or other materials provided with the distribution.
//
// * Neither the name of the University of Colorado at Boulder nor the
// names of its contributors may be used to endorse or promote
// products derived from this software without specific prior written
// permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//
package org.xbill.DNS;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.Date;
import java.util.Iterator;
import junit.framework.TestCase;
import org.xbill.DNS.ARecord;
import org.xbill.DNS.CNAMERecord;
import org.xbill.DNS.DClass;
import org.xbill.DNS.Name;
import org.xbill.DNS.RRset;
import org.xbill.DNS.Record;
import org.xbill.DNS.RRSIGRecord;
import org.xbill.DNS.TextParseException;
import org.xbill.DNS.Type;
public class RRsetTest extends TestCase
{
private RRset m_rs;
Name m_name, m_name2;
long m_ttl;
ARecord m_a1, m_a2;
RRSIGRecord m_s1, m_s2;
public void setUp() throws TextParseException, UnknownHostException
{
m_rs = new RRset();
m_name = Name.fromString("this.is.a.test.");
m_name2 = Name.fromString("this.is.another.test.");
m_ttl = 0xABCDL;
m_a1 = new ARecord(m_name, DClass.IN, m_ttl,
InetAddress.getByName("192.169.232.11"));
m_a2 = new ARecord(m_name, DClass.IN, m_ttl+1,
InetAddress.getByName("192.169.232.12"));
m_s1 = new RRSIGRecord(m_name, DClass.IN, m_ttl, Type.A, 0xF, 0xABCDEL,
new Date(), new Date(), 0xA, m_name,
new byte[ 0 ]);
m_s2 = new RRSIGRecord(m_name, DClass.IN, m_ttl, Type.A, 0xF, 0xABCDEL,
new Date(), new Date(), 0xA, m_name2,
new byte[ 0 ]);
}
public void test_ctor_0arg()
{
assertEquals(0, m_rs.size());
try {m_rs.getDClass(); fail("IllegalStateException not thrown");}
catch( IllegalStateException e ){}
try {m_rs.getType();fail("IllegalStateException not thrown");}
catch( IllegalStateException e ){}
try {m_rs.getTTL();fail("IllegalStateException not thrown");}
catch( IllegalStateException e ){}
try {m_rs.getName();fail("IllegalStateException not thrown");}
catch( IllegalStateException e ){}
try {m_rs.first();fail("IllegalStateException not thrown");}
catch( IllegalStateException e ){}
try {m_rs.toString();fail("IllegalStateException not thrown");}
catch( IllegalStateException e ){}
Iterator itr = m_rs.rrs();
assertNotNull(itr);
assertFalse(itr.hasNext());
itr = m_rs.sigs();
assertNotNull(itr);
assertFalse(itr.hasNext());
}
public void test_basics() throws TextParseException,
UnknownHostException
{
m_rs.addRR(m_a1);
assertEquals(1, m_rs.size());
assertEquals(DClass.IN, m_rs.getDClass());
assertEquals(m_a1, m_rs.first());
assertEquals(m_name, m_rs.getName());
assertEquals(m_ttl, m_rs.getTTL());
assertEquals(Type.A, m_rs.getType());
// add it again, and make sure nothing changed
m_rs.addRR(m_a1);
assertEquals(1, m_rs.size());
assertEquals(DClass.IN, m_rs.getDClass());
assertEquals(m_a1, m_rs.first());
assertEquals(m_name, m_rs.getName());
assertEquals(m_ttl, m_rs.getTTL());
assertEquals(Type.A, m_rs.getType());
m_rs.addRR(m_a2);
assertEquals(2, m_rs.size());
assertEquals(DClass.IN, m_rs.getDClass());
Record r = m_rs.first();
assertEquals(m_a1, r);
assertEquals(m_name, m_rs.getName());
assertEquals(m_ttl, m_rs.getTTL());
assertEquals(Type.A, m_rs.getType());
Iterator itr = m_rs.rrs();
assertEquals(m_a1, itr.next());
assertEquals(m_a2, itr.next());
// make sure that it rotates
itr = m_rs.rrs();
assertEquals(m_a2, itr.next());
assertEquals(m_a1, itr.next());
itr = m_rs.rrs();
assertEquals(m_a1, itr.next());
assertEquals(m_a2, itr.next());
m_rs.deleteRR(m_a1);
assertEquals(1, m_rs.size());
assertEquals(DClass.IN, m_rs.getDClass());
assertEquals(m_a2, m_rs.first());
assertEquals(m_name, m_rs.getName());
assertEquals(m_ttl, m_rs.getTTL());
assertEquals(Type.A, m_rs.getType());
// the signature records
m_rs.addRR(m_s1);
assertEquals(1, m_rs.size());
itr = m_rs.sigs();
assertEquals(m_s1, itr.next());
assertFalse(itr.hasNext());
m_rs.addRR(m_s1);
itr = m_rs.sigs();
assertEquals(m_s1, itr.next());
assertFalse(itr.hasNext());
m_rs.addRR(m_s2);
itr = m_rs.sigs();
assertEquals(m_s1, itr.next());
assertEquals(m_s2, itr.next());
assertFalse(itr.hasNext());
m_rs.deleteRR(m_s1);
itr = m_rs.sigs();
assertEquals(m_s2, itr.next());
assertFalse(itr.hasNext());
// clear it all
m_rs.clear();
assertEquals(0, m_rs.size());
assertFalse(m_rs.rrs().hasNext());
assertFalse(m_rs.sigs().hasNext());
}
public void test_ctor_1arg()
{
m_rs.addRR(m_a1);
m_rs.addRR(m_a2);
m_rs.addRR(m_s1);
m_rs.addRR(m_s2);
RRset rs2 = new RRset( m_rs );
assertEquals(2, rs2.size());
assertEquals(m_a1, rs2.first());
Iterator itr = rs2.rrs();
assertEquals(m_a1, itr.next());
assertEquals(m_a2, itr.next());
assertFalse(itr.hasNext());
itr = rs2.sigs();
assertTrue(itr.hasNext());
assertEquals(m_s1, itr.next());
assertTrue(itr.hasNext());
assertEquals(m_s2, itr.next());
assertFalse(itr.hasNext());
}
public void test_toString()
{
m_rs.addRR(m_a1);
m_rs.addRR(m_a2);
m_rs.addRR(m_s1);
m_rs.addRR(m_s2);
String out = m_rs.toString();
assertTrue(out.indexOf(m_name.toString()) != -1);
assertTrue(out.indexOf(" IN A ") != -1);
assertTrue(out.indexOf("[192.169.232.11]") != -1);
assertTrue(out.indexOf("[192.169.232.12]") != -1);
}
public void test_addRR_invalidType() throws TextParseException
{
m_rs.addRR(m_a1);
CNAMERecord c = new CNAMERecord(m_name, DClass.IN, m_ttl, Name.fromString("an.alias."));
try {
m_rs.addRR(c);
fail("IllegalArgumentException not thrown");
}
catch( IllegalArgumentException e ){
}
}
public void test_addRR_invalidName() throws TextParseException, UnknownHostException
{
m_rs.addRR(m_a1);
m_a2 = new ARecord(m_name2, DClass.IN, m_ttl,
InetAddress.getByName("192.169.232.11"));
try {
m_rs.addRR(m_a2);
fail("IllegalArgumentException not thrown");
}
catch( IllegalArgumentException e ){
}
}
public void test_addRR_invalidDClass() throws TextParseException, UnknownHostException
{
m_rs.addRR(m_a1);
m_a2 = new ARecord(m_name, DClass.CHAOS, m_ttl,
InetAddress.getByName("192.169.232.11"));
try {
m_rs.addRR(m_a2);
fail("IllegalArgumentException not thrown");
}
catch( IllegalArgumentException e ){
}
}
public void test_TTLcalculation()
{
m_rs.addRR(m_a2);
assertEquals(m_a2.getTTL(), m_rs.getTTL());
m_rs.addRR(m_a1);
assertEquals(m_a1.getTTL(), m_rs.getTTL());
Iterator itr = m_rs.rrs();
while( itr.hasNext() ){
Record r = (Record)itr.next();
assertEquals( m_a1.getTTL(), r.getTTL());
}
}
public void test_Record_placement()
{
m_rs.addRR(m_a1);
m_rs.addRR(m_s1);
m_rs.addRR(m_a2);
Iterator itr = m_rs.rrs();
assertTrue(itr.hasNext());
assertEquals(m_a1, itr.next());
assertTrue(itr.hasNext());
assertEquals(m_a2, itr.next());
assertFalse(itr.hasNext());
itr = m_rs.sigs();
assertTrue(itr.hasNext());
assertEquals(m_s1, itr.next());
assertFalse(itr.hasNext());
}
public void test_noncycling_iterator()
{
m_rs.addRR(m_a1);
m_rs.addRR(m_a2);
Iterator itr = m_rs.rrs(false);
assertTrue(itr.hasNext());
assertEquals(m_a1, itr.next());
assertTrue(itr.hasNext());
assertEquals(m_a2, itr.next());
itr = m_rs.rrs(false);
assertTrue(itr.hasNext());
assertEquals(m_a1, itr.next());
assertTrue(itr.hasNext());
assertEquals(m_a2, itr.next());
}
}