rubydns 0.6.0 → 2.0.0

data/test/test_resolver.rb

@@ -1,105 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'helper'
-require 'rubydns'
-
-class ResolverTest < Test::Unit::TestCase
-	def test_basic_resolver
-		resolver = RubyDNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
-		
-		EventMachine::run do
-			resolver.query('google.com') do |response|
-				assert_equal RubyDNS::Message, response.class
-				EventMachine::stop
-			end
-		end
-		
-		EventMachine::run do
-			resolver.query('nonexistant.private') do |response|
-				assert_equal response.rcode, Resolv::DNS::RCode::NXDomain
-				EventMachine::stop
-			end
-		end
-	end
-	
-	def test_broken_resolver
-		resolver = RubyDNS::Resolver.new([])
-		
-		EventMachine::run do
-			resolver.query('google.com') do |response|
-				assert_equal RubyDNS::ResolutionFailure, response.class
-				
-				EventMachine::stop
-			end
-		end
-	end
-	
-	class MockRequest
-		attr :response
-		
-		def process_response!(response)
-			@response = response
-		end
-	end
-	
-	def test_dirty_packets_udp
-		mock_request = MockRequest.new
-		
-		handler_class = Class.new{ include RubyDNS::Resolver::Request::UDPRequestHandler }
-		handler = handler_class.new(mock_request, nil, nil)
-		
-		handler.receive_data("This is not a real message!")
-		
-		assert_equal Resolv::DNS::DecodeError, mock_request.response.class
-	end
-	
-	def test_dirty_packets_tcp
-		mock_request = MockRequest.new
-		
-		handler_class = Class.new{ include RubyDNS::Resolver::Request::TCPRequestHandler }
-		handler = handler_class.new(mock_request)
-		
-		data = "This is not a real message!"
-		handler.receive_data([data.length].pack('n') + data)
-		
-		assert_equal Resolv::DNS::DecodeError, mock_request.response.class
-	end
-	
-	def test_addresses_for
-		resolver = RubyDNS::Resolver.new([[:udp, "8.8.8.8", 53], [:tcp, "8.8.8.8", 53]])
-		resolved_addresses = nil
-		
-		EventMachine::run do
-			resolver.addresses_for("www.google.com.") do |addresses|
-				resolved_addresses = addresses
-				
-				EventMachine::stop
-			end
-		end
-		
-		assert resolved_addresses.count > 0
-		
-		address = resolved_addresses[0]
-		assert address.kind_of?(Resolv::IPv4) || address.kind_of?(Resolv::IPv6)
-	end
-end

data/test/test_rules.rb

@@ -1,74 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'helper'
-require 'pathname'
-
-require 'rubydns'
-require 'rubydns/resolver'
-require 'rubydns/extensions/string'
-
-class RulesTest < Test::Unit::TestCase
-	IN = Resolv::DNS::Resource::IN
-	
-	def setup
-		@server = RubyDNS::Server.new
-		@true_callback = Proc.new { true }
-	end
-	
-	def teardown
-	end
-	
-	def test_string_pattern
-		rule = RubyDNS::Server::Rule.new(["foobar", IN::A], @true_callback)
-		
-		assert rule.call(@server, "foobar", IN::A)
-		assert !rule.call(@server, "barfoo", IN::A)
-		assert !rule.call(@server, "foobar", IN::TXT)
-	end
-	
-	def test_regexp_pattern
-		rule = RubyDNS::Server::Rule.new([/foo/, IN::A], @true_callback)
-		
-		assert rule.call(@server, "foobar", IN::A)
-		assert !rule.call(@server, "barbaz", IN::A)
-		assert !rule.call(@server, "foobar", IN::TXT)
-	end
-	
-	def test_callback_pattern
-		calls = 0
-		
-		callback = Proc.new do |name, resource_class|
-			# A counter used to check the number of times this block was invoked.
-			calls += 1
-			
-			name.size == 6
-		end
-		
-		rule = RubyDNS::Server::Rule.new([callback], @true_callback)
-		
-		assert rule.call(@server, "foobar", IN::A)
-		assert !rule.call(@server, "foobarbaz", IN::A)
-		
-		assert_equal 2, calls
-	end
-end

data/test/test_slow_server.rb

@@ -1,98 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'helper'
-require 'pathname'
-
-require 'rubydns'
-require 'rubydns/resolver'
-require 'rubydns/extensions/string'
-
-class SlowServer < RExec::Daemon::Base
-	SERVER_PORTS = [[:udp, '127.0.0.1', 5330], [:tcp, '127.0.0.1', 5330]]
-	
-	@@base_directory = File.dirname(__FILE__)
-
-	Name = Resolv::DNS::Name
-	IN = Resolv::DNS::Resource::IN
-
-	def self.run
-		RubyDNS::run_server(:listen => SERVER_PORTS) do
-			match(/\.*.com/, IN::A) do |transaction|
-				transaction.defer!
-			
-				# No domain exists, after 5 seconds:
-				EventMachine::Timer.new(2) do
-					transaction.failure!(:NXDomain)
-				end
-			end
-		
-			otherwise do |transaction|
-				transaction.failure!(:NXDomain)
-			end
-		end
-	end
-end
-
-class SlowServerTest < Test::Unit::TestCase
-	def setup
-		SlowServer.start
-	end
-	
-	def teardown
-		SlowServer.stop
-	end
-	
-	def test_slow_request
-		start_time = Time.now
-		end_time = nil
-		
-		resolver = RubyDNS::Resolver.new(SlowServer::SERVER_PORTS, :timeout => 10)
-		
-		EventMachine::run do
-			resolver.query("apple.com", IN::A) do |response|
-				end_time = Time.now
-				
-				EventMachine::stop
-			end
-		end
-		
-		assert (end_time - start_time) > 2.0
-	end
-	
-	def test_normal_request
-		start_time = Time.now
-		end_time = nil
-		
-		resolver = RubyDNS::Resolver.new(SlowServer::SERVER_PORTS, :timeout => 10)
-		
-		EventMachine::run do
-			resolver.query("oriontransfer.org", IN::A) do |response|
-				end_time = Time.now
-				
-				EventMachine::stop
-			end
-		end
-		
-		assert (end_time - start_time) < 2.0
-	end
-end

data/test/test_truncation.rb

@@ -1,78 +0,0 @@
-#!/usr/bin/env ruby
-
-# Copyright, 2012, by Samuel G. D. Williams. <http://www.codeotaku.com>
-# 
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-# 
-# The above copyright notice and this permission notice shall be included in
-# all copies or substantial portions of the Software.
-# 
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
-# THE SOFTWARE.
-
-require 'helper'
-require 'pathname'
-
-require 'rubydns'
-require 'rubydns/resolver'
-require 'rubydns/extensions/string'
-
-class TruncatedServer < RExec::Daemon::Base
-	SERVER_PORTS = [[:udp, '127.0.0.1', 5320], [:tcp, '127.0.0.1', 5320]]
-	
-	@@base_directory = File.dirname(__FILE__)
-
-	Name = Resolv::DNS::Name
-	IN = Resolv::DNS::Resource::IN
-
-	def self.run
-		# Start the RubyDNS server
-		RubyDNS::run_server(:listen => SERVER_PORTS) do
-			match("truncation", IN::TXT) do |transaction|
-				text = "Hello World! " * 100
-				transaction.respond!(*text.chunked)
-			end
-			
-			# Default DNS handler
-			otherwise do |transaction|
-				transaction.failure!(:NXDomain)
-			end
-		end
-	end
-end
-
-class TruncationTest < Test::Unit::TestCase
-	def setup
-		TruncatedServer.start
-	end
-	
-	def teardown
-		TruncatedServer.stop
-	end
-	
-	def test_tcp_failover
-		resolver = RubyDNS::Resolver.new(TruncatedServer::SERVER_PORTS)
-		
-		EventMachine::run do
-			resolver.query("truncation", IN::TXT) do |response|
-				
-				
-				text = response.answer.first
-				
-				assert_equal "Hello World! " * 100, text[2].strings.join
-				
-				EventMachine::stop
-			end
-		end
-	end
-end