Updated a couple of test helpers.

1 files changed, 7 insertions, 1 deletions

diff --git a/test/th_mauve_resolv.rb b/test/th_mauve_resolv.rb
index 98b597b..10bdacc 100644
--- a/test/th_mauve_resolv.rb
+++ b/test/th_mauve_resolv.rb
@@ -14,13 +14,19 @@ module Mauve
       alias_method :get_ips_for_without_testing, :get_ips_for
 
       def get_ips_for_with_testing(host)
+
         lookup = {
          "test-1.example.com" => %w(1.2.3.4 2001:1:2:3::4),
          "test-2.example.com" => %w(1.2.3.5 2001:1:2:3::5),
          "www.example.com"    => %w(1.2.3.4),
          "www2.example.com"   => %w(1.2.3.5 2001:2::2)
         }
-        lookup[host] || get_ips_for_without_testing(host)
+        if lookup.has_key?(host)
+          self.count += lookup[host].length if $debug
+          lookup[host]
+        else
+          get_ips_for_without_testing(host)
+        end
       end
 
       alias_method :get_ips_for, :get_ips_for_with_testing