dnsruby 1.1

data/test/tc_packet_unique_push.rb

@@ -0,0 +1,102 @@
+#--
+#Copyright 2007 Nominet UK
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License. 
+#You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0 
+#
+#Unless required by applicable law or agreed to in writing, software 
+#distributed under the License is distributed on an "AS IS" BASIS, 
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
+#See the License for the specific language governing permissions and 
+#limitations under the License.
+#++
+require 'rubygems'
+require 'test/unit'
+require 'dnsruby'
+include Dnsruby
+class TestPacketUniquePush < Test::Unit::TestCase
+  #  def test_packUniquePush
+  #    
+  #    
+  #    testProc('unique_push');
+  #  end
+  #  
+  ##  def test_packetSafePush
+  ##    begin
+  ##      testProc('safe_push');
+  ##      flunk("Shouldn't work!")
+  ##    rescue Exception
+  ##    end
+  ##  end
+  
+  #  def testProc (method)
+  def test_proc
+    domain = 'example.com';
+    
+    tests = [
+    [ 
+    1,
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    ],
+    [
+    2,
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('bar.example.com 60 IN A 10.0.0.1'),
+    ],
+    [ 
+    1, # RFC 2136 section 1.1
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 90 IN A 10.0.0.1'),
+    ],
+    [ 
+    3,
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 60 IN A 10.0.0.2'),
+    RR.create('foo.example.com 60 IN A 10.0.0.3'),
+    ],
+    [ 
+    3,
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 60 IN A 10.0.0.2'),
+    RR.create('foo.example.com 60 IN A 10.0.0.3'),
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    ],
+    [ 
+    3,
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 60 IN A 10.0.0.2'),
+    RR.create('foo.example.com 60 IN A 10.0.0.1'),
+    RR.create('foo.example.com 60 IN A 10.0.0.4'),
+    ],
+    ]
+    
+    methods = {
+		'add_answer'     => 'ancount',
+		'add_authority'  => 'nscount',
+		'add_additional' => 'arcount',
+    }
+    
+    tests.each do | try |  
+      count = try.shift;
+      rrs = try;
+      
+      methods.each do |method, count_meth|
+        
+        packet = Message.new(domain)
+        
+        rrs.each do |rr|
+          packet.send(method,rr)
+        end
+        
+        assert_equal(count, packet.header.send(count_meth), "#{method} right for #{rrs.inspect}");
+        assert_equal(count, packet.header.send(count_meth), "#{method} right for #{rrs.inspect}");
+        
+      end
+    end
+  end
+end

data/test/tc_question.rb

@@ -0,0 +1,51 @@
+#--
+#Copyright 2007 Nominet UK
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License. 
+#You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0 
+#
+#Unless required by applicable law or agreed to in writing, software 
+#distributed under the License is distributed on an "AS IS" BASIS, 
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
+#See the License for the specific language governing permissions and 
+#limitations under the License.
+#++
+require 'rubygems'
+require 'test/unit'
+require 'dnsruby'
+include Dnsruby
+class TestQuestion < Test::Unit::TestCase
+  def test_question
+    domain = "example.com"
+    type = Types.MX
+    klass = Classes.IN
+    
+    q = Question.new(domain, type, klass)
+    assert(q, "new() returned something")
+    assert_equal(domain, q.qname.to_s, "qName()")
+    assert_equal(type, q.qtype, "qType()")
+    assert_equal(klass, q.qclass, "qClass()")
+    
+    #
+    # Check the aliases
+    #
+    assert_equal(q.zname.to_s,  domain, 'zName()'  );
+    assert_equal(q.ztype,  type,   'zType()'  );
+    assert_equal(q.zclass, klass,  'zClass()' );
+    
+    #
+    # Check that we can change stuff
+    #
+    q.qname=('example.net');
+    q.qtype=('A');
+    q.qclass=('CH');
+    
+    assert_equal('example.net', q.qname.to_s,  'qName()'  );
+    assert_equal(q.qtype,  Types.A,           'qType()'  );
+    assert_equal(q.qclass, Classes.CH,          'qClass()' );
+    
+  end
+end

data/test/tc_res_config.rb

@@ -0,0 +1,92 @@
+#--
+#Copyright 2007 Nominet UK
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License. 
+#You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0 
+#
+#Unless required by applicable law or agreed to in writing, software 
+#distributed under the License is distributed on an "AS IS" BASIS, 
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
+#See the License for the specific language governing permissions and 
+#limitations under the License.
+#++
+require 'rubygems'
+require 'test/unit'
+require 'dnsruby'
+class TestResolverConfig < Test::Unit::TestCase
+  def setup
+    Dnsruby::Config.reset
+  end
+  
+  GoodInput = {
+	"port"		  => 54,
+	"src_address"        => '10.1.0.1',
+	"src_port"        => 53,
+	"use_tcp"		   => true,
+    #	"stayopen"       => 1,
+	"ignore_truncation"          => true,
+	"recurse"        => false,
+	"packet_timeout"    => 5,
+	"persistent_tcp" => true,
+	"persistent_udp" => true,
+    #	"dnssec"         => 1,
+    #	"force_v4"       => 1,
+  };
+  
+  ExtendedInput={
+    "query_timeout"        => 30,
+	"retry_delay"	       => 6,
+	"retry_times"		   => 5,
+  }
+  
+  LookupInput={
+	"domain"	       => 'dnsruby.rubyforge.org',
+	"apply_search_list"         => false,
+	"ndots"       => 4	,
+	"apply_domain" => false	
+  }
+  
+  def test_multiple_resolver
+    res = Dnsruby::Resolver.new();
+    assert(res, "new returned something");
+    assert_instance_of(Dnsruby::Resolver, res, "new() returns an object of the correct class.");
+    
+    assert(res.config.nameserver,       'nameserver() works');
+    
+    searchlist = ["t.dnsruby.validation-test-servers.nominet.org.uk", "t2.dnsruby.validation-test-servers.nominet.org.uk"];
+    assert_equal(res.config.search=searchlist, searchlist, 'setting searchlist returns correctly.');
+    assert_equal(res.config.search,               searchlist, 'setting searchlist stickts.');
+    
+    
+    #~ #diag "\n\nIf you do not have Net::DNS::SEC installed you will see a warning.\n";
+    #~ #diag "It is safe to ignore this\n";
+    
+     (GoodInput.merge(ExtendedInput)).each do | param, value |
+      #      puts("Setting " + param);
+      res.send(param+"=", value)
+      assert_equal(res.send(param), value,       "setting #param sticks");
+    end;
+    
+  end
+  
+  def test_single_resolver
+    res = Dnsruby::SingleResolver.new
+    GoodInput.each do | param, value |
+      #      puts("Setting " + param);
+      res.send(param+"=", value)
+      assert_equal(res.send(param), value,       "setting #param sticks");
+    end;
+  end
+  
+  def test_dns
+    res = Dnsruby::DNS.new
+    LookupInput.each do | param, value |
+      res.config.send(param+"=", value)
+      assert_equal(res.config.send(param), value,       "setting #param sticks");
+    end;    
+  end
+  
+end

data/test/tc_res_env.rb

@@ -0,0 +1,51 @@
+#--
+#Copyright 2007 Nominet UK
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License. 
+#You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0 
+#
+#Unless required by applicable law or agreed to in writing, software 
+#distributed under the License is distributed on an "AS IS" BASIS, 
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
+#See the License for the specific language governing permissions and 
+#limitations under the License.
+#++
+require 'rubygems'
+require 'test/unit'
+require 'dnsruby'
+include Dnsruby
+class TestResolverEnv < Test::Unit::TestCase
+# @todo@ Dnsruby does not provide this functionality
+  def test_res_env
+    ENV['RES_NAMESERVERS'] = '10.0.1.128 10.0.2.128';
+    ENV['RES_SEARCHLIST']  = 'dnsruby.validation-test-servers.nominet.org.uk lib.dnsruby.validation-test-servers.nominet.org.uk';
+    ENV['LOCALDOMAIN']     = 't.dnsruby.validation-test-servers.nominet.org.uk';
+    ENV['RES_OPTIONS']     = 'retrans:3 retry:2 debug';
+    
+    
+    res = DNS.new;
+    
+    assert(res,                       "new() returned something");
+    assert(res.config.nameserver,   "nameservers() works");
+    
+    servers = res.config.nameserver;
+    
+    assert_equal(servers[0], '10.0.1.128',  'Nameserver set correctly');
+    assert_equal(servers[1], '10.0.2.128',  'Nameserver set correctly');
+    
+    
+    search = res.searchlist;
+    assert_equal(search[0], 'dnsruby.validation-test-servers.nominet.org.uk',     'Search set correctly' );
+    assert_equal(search[1], 'lib.dnsruby.validation-test-servers.nominet.org.uk', 'Search set correctly' );
+    
+    assert_equal(res.domain,  't.dnsruby.validation-test-servers.nominet.org.uk', 'Local domain works'  );
+    assert_equal(3, res.retrans,               'Retransmit works'    );
+    assert_equal(2, res.retry,                 'Retry works'         );
+    assert(res.debug,                    'Debug works'         );
+    
+    
+  end
+end

data/test/tc_res_file.rb

@@ -0,0 +1,42 @@
+#--
+#Copyright 2007 Nominet UK
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License. 
+#You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0 
+#
+#Unless required by applicable law or agreed to in writing, software 
+#distributed under the License is distributed on an "AS IS" BASIS, 
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
+#See the License for the specific language governing permissions and 
+#limitations under the License.
+#++
+require 'rubygems'
+require 'test/unit'
+require 'dnsruby'
+class TestAResolverFile < Test::Unit::TestCase
+  def setup
+    Dnsruby::Config.reset
+  end
+  
+  def test_resFile
+    res = Dnsruby::DNS.new("test/resolv.conf")
+    
+    assert(res,                "new() returned something")
+    assert(res.config.nameserver,   "nameservers() works")
+    
+    servers = res.config.nameserver
+    
+    assert_equal(servers[0], '10.0.1.128',  'Nameserver set correctly')
+    assert_equal(servers[1], '10.0.2.128',  'Nameserver set correctly')
+    
+    
+    search = res.config.search
+    assert(search.include?('dnsruby.validation-test-servers.nominet.org.uk'),     'Search set correctly' )
+    assert(search.include?('lib.dnsruby.validation-test-servers.nominet.org.uk'), 'Search set correctly' )
+    
+    assert(res.config.domain=='t.dnsruby.validation-test-servers.nominet.org.uk', 'Local domain works'  )
+  end
+end

data/test/tc_res_opt.rb

@@ -0,0 +1,187 @@
+#--
+#Copyright 2007 Nominet UK
+#
+#Licensed under the Apache License, Version 2.0 (the "License");
+#you may not use this file except in compliance with the License. 
+#You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0 
+#
+#Unless required by applicable law or agreed to in writing, software 
+#distributed under the License is distributed on an "AS IS" BASIS, 
+#WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 
+#See the License for the specific language governing permissions and 
+#limitations under the License.
+#++
+require 'rubygems'
+require 'test/unit'
+require 'dnsruby'
+include Dnsruby
+class TestResOpt < Test::Unit::TestCase
+  def test_dns_file
+    
+    # .txt because this test will run under windows, unlike the other file
+    # configuration tests.
+    res = Dnsruby::DNS.new('test/custom.txt')
+    
+    assert(res,                           'new() returned something')
+    assert_instance_of(DNS, res, 'new() returns an object of the correct class.')
+    assert(res.config.nameserver,       'nameservers() works')
+    
+    servers = res.config.nameserver
+    
+    assert_equal('10.0.1.42', servers[0],  'Nameserver set correctly')
+    assert_equal('10.0.2.42',  servers[1], 'Nameserver set correctly')
+    
+    
+    search = res.config.search
+    assert(search.include?('alt.dnsruby.validation-test-servers.nominet.org.uk'), 'Search set correctly' )
+    assert(search.include?('ext.dnsruby.validation-test-servers.nominet.org.uk'), 'Search set correctly' )
+    
+    assert(res.config.domain == 't2.dnsruby.validation-test-servers.nominet.org.uk',  'Local domain works'  )
+  end
+  
+  def test_resolver_file
+    res = Dnsruby::Resolver.new({:config_info => 'test/custom.txt'})
+    assert(res.config.nameserver==['10.0.1.42', '10.0.2.42'], res.config.nameserver.to_s)
+  end
+  
+  def test_no_file
+    Dnsruby::TheLog.level=Logger::FATAL
+    res=nil
+    begin
+      res = DNS.new('nosuch.txt')
+      assert_equal(["0.0.0.0"], res.nameserver,"No nameservers should be set for #{test} = #{input}")
+    rescue Exception
+    end
+    begin
+      res = Resolver.new('nosuch.txt')
+      assert_equal(["0.0.0.0"], res.nameserver,"No nameservers should be set for #{test} = #{input}")
+    rescue Exception
+    end
+#    Dnsruby::TheLog.level=Logger::ERROR
+  end
+  
+  def test_config_hash_singleresolver
+    # Resolver interface gives us : port, TCP, IgnoreTruncation, TSIGkey, timeout
+    # SR : server, local_address, persistent_tcp, persistent_udp, udp_size
+    test_config = {
+      :server	   => '10.0.0.1',
+      :port		   => 54, # SingleResolver and Multi-Resolver
+      :src_address        => '10.1.0.1', # SingleResolver and Multi-Resolver
+      :src_port        => 53, # SingleResolver and Multi-Resolver
+      :use_tcp		   => true, # SingleResolver and Multi-Resolver
+      :ignore_truncation          => true, # SingleResolver and Multi-Resolver
+      :recurse        => false,
+      :packet_timeout    => 60, # SingleResolver and Multi-Resolver # Only have one timeout for both UDP and TCP
+      :persistent_tcp => true, # SingleResolver and Multi-Resolver
+      :persistent_udp => true, # SingleResolver and Multi-Resolver
+      :dnssec         => true,
+    }
+    
+    res = SingleResolver.new(test_config)
+    test_config.keys.each do |item|
+      assert_equal(test_config[item], res.send(item), "#{item} is correct")
+    end	
+  end
+  
+  def test_config_hash_multiresolver
+    # Resolver interface gives us : port, TCP, IgnoreTruncation, TSIGkey, timeout
+    # ER : retries, load_balance. Also loads servers from Config and configures SRs to point to them
+    #   Also implements Resolver interface - but iterates this through *all* SRs
+    test_config = {
+      :nameserver	   => ['10.0.0.1', '10.0.0.2'], # for Multi-Resolver & DNS
+      :port		   => 54, # SingleResolver and Multi-Resolver
+      :src_address        => '10.1.0.1', # SingleResolver and Multi-Resolver
+      :src_port        => 53, # SingleResolver and Multi-Resolver
+      :retry_delay	       => 6, # DNS and Multi-Resolver
+      :retry_times		   => 5, # DNSand Multi-Resolver
+      :use_tcp		   => true, # SingleResolver and Multi-Resolver
+      :ignore_truncation          => true, # SingleResolver and Multi-Resolver
+      :recurse        => false, 
+      :packet_timeout    => 60, # SingleResolver and Multi-Resolver # Only have one timeout for both UDP and TCP
+      :query_timeout    => 60, # Multi-Resolver only
+      :persistent_tcp => true, # SingleResolver and Multi-Resolver
+      :persistent_udp => true, # SingleResolver and Multi-Resolver
+      :dnssec         => true,
+    }
+    
+    res = Resolver.new(test_config)
+    test_config.keys.each do |item|
+      if (item==:nameserver)
+        assert_equal(res.config.nameserver, test_config[item], "#{item} is correct")
+      else
+        assert_equal(res.send(item), test_config[item], "#{item} is correct")
+      end
+    end	
+  end
+  
+  def test_config_hash_lookup
+    # Lookup : can specify resolver, searchpath
+    #
+    # Check that we can set things in new()
+    #
+    res=nil
+    
+    test_config = {
+      :nameserver	   => ['10.0.0.1', '10.0.0.2'], # for Multi-Resolver & DNS
+      :domain	       => 'dnsruby.validation-test-servers.nominet.org.uk', # one for DNS only?
+      :search	   => ['dnsruby.validation-test-servers.nominet.org.uk', 't.dnsruby.validation-test-servers.nominet.org.uk'], # one for DNS
+      :ndots          => 2, # DNS only
+      :apply_search_list         => false, # DNS only 
+      :apply_domain => false, # DNS only 
+    }
+    
+    res = DNS.new(test_config)
+    test_config.keys.each do |item|
+      assert_equal(res.config.send(item), test_config[item], "#{item} is correct")
+    end	
+  end
+  
+  
+  def test_bad_config
+    res=nil
+    Dnsruby::TheLog.level=Logger::FATAL
+    
+    bad_input = {
+      :tsig_rr        => 'set',
+      :errorstring    => 'set',
+      :answerfrom     => 'set',
+      :answersize     => 'set',
+      :querytime      => 'set',
+      :axfr_sel       => 'set',
+      :axfr_rr        => 'set',
+      :axfr_soa_count => 'set',
+      :udppacketsize  => 'set',
+      :cdflag         => 'set',
+    }
+    res=nil
+    begin    
+      res = Resolver.new(bad_input)
+    rescue Exception 
+    end
+    if (res)    
+      bad_input.keys.each do |key|
+        begin
+          assert_not_equal(res.send(key), 'set', "#{key} is not set")
+        rescue Exception
+        end
+      end
+    end
+    
+    res=nil
+    begin    
+      res = DNS.new(bad_input)
+    rescue Exception 
+    end
+    if (res)        
+      bad_input.keys.each do |key|
+        begin
+          assert_not_equal(res.send(key), 'set', "#{key} is not set")
+        rescue Exception
+        end
+      end
+#      Dnsruby::TheLog.level=Logger::ERROR
+    end
+  end
+end