fixed tests

2 files changed, 4 insertions, 2 deletions

diff --git a/test/unit/config_object_list_test.rb b/test/unit/config_object_list_test.rb
index a0ca9d5..a3b76f2 100644
--- a/test/unit/config_object_list_test.rb
+++ b/test/unit/config_object_list_test.rb
@@ -9,8 +9,9 @@ class TestMeme < MiniTest::Unit::TestCase
   end
 
   def test_complex_node_search
+    domain = manager.provider.domain
     nodes = manager.nodes['dns.public' => true]
-    expected = [{"domain_full"=>"ns1.rewire.co"}, {"domain_full"=>"ns2.rewire.co"}, {"domain_full"=>"vpn1.rewire.co"}, {"domain_full"=>"web1.rewire.co"}]
+    expected = [{"domain_full"=>"ns1.#{domain}"}, {"domain_full"=>"ns2.#{domain}"}, {"domain_full"=>"vpn1.#{domain}"}, {"domain_full"=>"web1.#{domain}"}]
     assert_equal expected.size, nodes.size
     assert_equal expected, nodes.fields('domain.full')
   end
diff --git a/test/unit/config_object_test.rb b/test/unit/config_object_test.rb
index 2cd6dff..0c293c2 100644
--- a/test/unit/config_object_test.rb
+++ b/test/unit/config_object_test.rb
@@ -3,8 +3,9 @@ require File.expand_path('test_helper', File.dirname(__FILE__))
 class TestMeme < MiniTest::Unit::TestCase
 
   def test_bracket_lookup
+    domain = manager.provider.domain
     vpn1 = manager.nodes['vpn1']
-    assert_equal 'vpn1.rewire.co', vpn1['domain.full']
+    assert_equal "vpn1.#{domain}", vpn1['domain.full']
   end
 
 end