rubber 2.0.5 → 2.0.6

data/test/dns/aws_test.rb

@@ -0,0 +1,192 @@
+require File.expand_path(File.join(__FILE__, '../..', 'test_helper'))
+require 'rubber/dns/aws'
+
+if ENV['NO_FOG_MOCK']
+
+class AwsTest < Test::Unit::TestCase
+
+   context "fog aws dns" do
+
+      setup do
+        env = {'access_key' => get_secret('cloud_providers.aws.access_key') || 'xxx',
+               'access_secret' => get_secret('cloud_providers.aws.secret_access_key') || 'yyy'}
+        @env = Rubber::Configuration::Environment::BoundEnv.new(env, nil, nil)
+
+        @dns = Rubber::Dns::Aws.new(@env)
+        destroy_test_domains(@dns)
+      end
+      
+      context "compatibility" do
+        should "create using old credential style" do
+          env = {'dns_provider' => 'fog',
+                 'dns_providers' => {
+                     'fog' => {
+                         'credentials' => {
+                            'provider' => 'aws', 'aws_access_key_id' => 'xxx', 'aws_secret_access_key' => 'yyy'
+                         }
+                     }
+                 }
+          }
+          @env = Rubber::Configuration::Environment::BoundEnv.new(env, nil, nil)
+          
+          provider = Rubber::Dns::get_provider(@env.dns_provider, @env)
+          assert provider
+          assert provider.instance_of?(Rubber::Dns::Aws)
+        end
+      end
+      
+      context "find_or_create" do
+      
+        should "create domain if it doesn't exist" do
+          assert_equal 0, all_test_zones(@dns).size
+                
+          zone0 = @dns.find_or_create_zone("#{TEST_DOMAIN}1.com")
+      
+          assert_equal 1, all_test_zones(@dns).size
+          zone1 = all_test_zones(@dns).find {|z| z.domain =~ /^#{TEST_DOMAIN}1.com/ }
+          assert zone1
+          assert_equal zone0.id, zone1.id
+          assert_equal zone0.domain, zone1.domain
+        end
+      
+        should "match the same domain that was passed" do
+          assert_equal 0, all_test_zones(@dns).size
+      
+          zone0 = @dns.find_or_create_zone("abc#{TEST_DOMAIN}2.com")
+          zone1 = @dns.find_or_create_zone("#{TEST_DOMAIN}2.com")
+      
+          assert_equal 2, all_test_zones(@dns).size
+      
+          zone2 = all_test_zones(@dns).find {|z| z.domain =~ /^#{TEST_DOMAIN}2.com/ }
+          assert zone2
+          assert_equal zone1.id, zone2.id
+          assert_equal zone1.domain, zone2.domain
+        end
+      
+        should "do nothing if domain already exists" do
+          @dns.client.zones.create(:domain => "#{TEST_DOMAIN}3.com")
+          assert_equal 1, all_test_zones(@dns).size
+      
+          zone0 = @dns.find_or_create_zone("#{TEST_DOMAIN}3.com")
+      
+          assert_equal 1, all_test_zones(@dns).size
+          zone1 = all_test_zones(@dns).find {|z| z.domain =~ /^#{TEST_DOMAIN}3.com/ }
+          assert_equal zone0.id, zone1.id
+          assert_equal zone0.domain, zone1.domain
+        end
+      
+      end
+
+      context "records" do
+      
+        setup do
+          @domain = "#{TEST_DOMAIN}#{rand(90) + 10}.com"
+          @zone = @dns.find_or_create_zone(@domain)
+        end
+      
+        should "create_record" do
+          @dns.create_host_record({:host => 'newhost', :domain => @domain, :data => ['1.1.1.1'], :type => 'A', :ttl => '333'})
+        
+          assert_equal @zone.records.all.size, 1
+          record = @zone.records.first
+          attributes = @dns.host_to_opts(record)
+        
+          assert_equal 'newhost',             attributes[:host]
+          assert_equal @domain,               attributes[:domain]
+          assert_equal ['1.1.1.1'],             attributes[:data]
+          assert_equal 'A',                   attributes[:type]
+          assert_equal 333,                   attributes[:ttl]
+        end
+        
+        should "destroy_record" do
+          # Create the record we want to test destroying.
+          @dns.create_host_record({:host => 'newhost', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'})
+          assert_equal 1, @zone.records.all.size
+        
+          @dns.destroy_host_record({:host => 'newhost', :domain => @domain})
+        
+          assert_equal 0, @zone.records.all.size
+        end
+        
+        should "update_record" do
+          params = {:host => 'host1', :domain => @domain, :data => ["1.1.1.1"]}
+          new = {:host => 'host1', :domain => @domain, :data => ["1.1.1.2"]}
+        
+          @dns.create_host_record({:host => 'host1', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'})
+          assert_equal 1, @zone.records.all.size
+        
+          @dns.update_host_record(params, new)
+          assert_equal 1, @zone.records.all.size
+        
+          record = @zone.records.all.first
+          attributes = @dns.host_to_opts(record)
+          assert_equal ['1.1.1.2'], attributes[:data]
+        end
+        
+        should "find_records" do
+          # Set up some sample records.
+          created = []
+          created << {:host => 'host1', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'}
+          created << {:host => '', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'}
+          created.each {|r| @dns.create_host_record(r) }
+        
+          # Search for records through the rubber DNS interface and make sure whe get what we expected.
+        
+          # Wildcard search.
+          records = @dns.find_host_records(:host => '*', :domain => @domain)
+          assert_equal 2, records.size
+        
+          # Blank hostname search.
+          records = @dns.find_host_records(:host => '', :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal '', records.first[:host]
+        
+          # Specific hostname search.
+          records = @dns.find_host_records(:host => 'host1', :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal 'host1', records.first[:host]
+        end
+      
+        should "find_many_records" do
+          # Set up many sample records - more than the max returned
+          # in a single fetch so we can test pagination
+          max = 102
+          created = []
+          max.times do |i|
+            created << {:host => "host#{i}", :domain => @domain, :data => ['1.1.1.1'], :type => 'A'}
+          end
+          created.each_slice(20) do |group|
+            group = group.collect do |o|
+              r = @dns.opts_to_host(o)
+              r[:action] = 'CREATE'
+              r[:resource_records] = r.delete(:value)
+              r[:ttl] = 300
+              r
+            end
+            @zone.connection.change_resource_record_sets(@zone.id, group)
+          end
+          
+          # Search for records through the rubber DNS interface and make sure whe get what we expected.
+        
+          # Wildcard search.
+          records = @dns.find_host_records(:host => '*', :domain => @domain)
+          assert_equal max, records.size
+        
+          # Specific hostname search.
+          records = @dns.find_host_records(:host => 'host1', :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal 'host1', records.first[:host]
+
+          # Specific hostname search.
+          records = @dns.find_host_records(:host => "host#{max - 1}", :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal "host#{max - 1}", records.first[:host]
+        end
+      
+      end
+
+   end
+
+end
+
+end

data/test/dns/zerigo_test.rb

@@ -0,0 +1,180 @@
+require File.expand_path(File.join(__FILE__, '../..', 'test_helper'))
+require 'rubber/dns/zerigo'
+
+class ZerigoTest < Test::Unit::TestCase
+
+   context "fog zerigo dns" do
+
+      setup do
+        env = {'email' => get_secret('dns_providers.zerigo.email') || 'xxx',
+               'token' => get_secret('dns_providers.zerigo.token') || 'yyy'}
+        @env = Rubber::Configuration::Environment::BoundEnv.new(env, nil, nil)
+
+        @dns = Rubber::Dns::Zerigo.new(@env)
+        destroy_test_domains(@dns)
+      end
+
+      context "compatibility" do
+        should "create using old credential style" do
+          env = {'dns_provider' => 'fog',
+                 'dns_providers' => {
+                     'fog' => {
+                         'credentials' => {
+                            'provider' => 'zerigo', 'zerigo_email' => 'xxx', 'zerigo_token' => 'yyy'
+                         }
+                     }
+                 }
+          }
+          @env = Rubber::Configuration::Environment::BoundEnv.new(env, nil, nil)
+          
+          provider = Rubber::Dns::get_provider(@env.dns_provider, @env)
+          assert provider
+          assert provider.instance_of?(Rubber::Dns::Zerigo)
+        end
+      end
+      
+      context "find_or_create" do
+      
+        should "create domain if it doesn't exist" do
+          assert_equal 0, all_test_zones(@dns).size
+                
+          zone0 = @dns.find_or_create_zone("#{TEST_DOMAIN}1.com")
+      
+          assert_equal 1, all_test_zones(@dns).size
+          zone1 = all_test_zones(@dns).find {|z| z.domain =~ /^#{TEST_DOMAIN}1.com/ }
+          assert zone1
+          assert_equal zone0.id, zone1.id
+          assert_equal zone0.domain, zone1.domain
+        end
+      
+        should "match the same domain that was passed" do
+          assert_equal 0, all_test_zones(@dns).size
+      
+          zone0 = @dns.find_or_create_zone("abc#{TEST_DOMAIN}2.com")
+          zone1 = @dns.find_or_create_zone("#{TEST_DOMAIN}2.com")
+      
+          assert_equal 2, all_test_zones(@dns).size
+      
+          zone2 = all_test_zones(@dns).find {|z| z.domain =~ /^#{TEST_DOMAIN}2.com/ }
+          assert zone2
+          assert_equal zone1.id, zone2.id
+          assert_equal zone1.domain, zone2.domain
+        end
+      
+        should "do nothing if domain already exists" do
+          @dns.client.zones.create(:domain => "#{TEST_DOMAIN}3.com")
+          assert_equal 1, all_test_zones(@dns).size
+      
+          zone0 = @dns.find_or_create_zone("#{TEST_DOMAIN}3.com")
+      
+          assert_equal 1, all_test_zones(@dns).size
+          zone1 = all_test_zones(@dns).find {|z| z.domain =~ /^#{TEST_DOMAIN}3.com/ }
+          assert_equal zone0.id, zone1.id
+          assert_equal zone0.domain, zone1.domain
+        end
+      
+      end
+
+      context "records" do
+      
+        setup do
+          @domain = "#{TEST_DOMAIN}#{rand(90) + 10}.com"
+          @zone = @dns.find_or_create_zone(@domain)
+        end
+      
+        should "create_record" do
+          @dns.create_host_record({:host => 'newhost', :domain => @domain, :data => ['1.1.1.1'], :type => 'A', :ttl => '333'})
+        
+          assert_equal @zone.records.all.size, 1
+          record = @zone.records.first
+          attributes = @dns.hosts_to_opts([record])
+        
+          assert_equal 'newhost',             attributes[:host]
+          assert_equal @domain,               attributes[:domain]
+          assert_equal ['1.1.1.1'],             attributes[:data]
+          assert_equal 'A',                   attributes[:type]
+          assert_equal 333,                   attributes[:ttl]
+        end
+        
+        should "destroy_record" do
+          # Create the record we want to test destroying.
+          @dns.create_host_record({:host => 'newhost', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'})
+          assert_equal 1, @zone.records.all.size
+        
+          @dns.destroy_host_record({:host => 'newhost', :domain => @domain})
+        
+          assert_equal 0, @zone.records.all.size
+        end
+        
+        should "update_record" do
+          params = {:host => 'host1', :domain => @domain, :data => ["1.1.1.1"]}
+          new = {:host => 'host1', :domain => @domain, :data => ["1.1.1.2"]}
+        
+          @dns.create_host_record({:host => 'host1', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'})
+          assert_equal 1, @zone.records.all.size
+        
+          @dns.update_host_record(params, new)
+          assert_equal 1, @zone.records.all.size
+        
+          record = @zone.records.all.first
+          attributes = @dns.hosts_to_opts([record])
+          assert_equal ['1.1.1.2'], attributes[:data]
+        end
+        
+        should "find_records" do
+          # Set up some sample records.
+          created = []
+          created << {:host => 'host1', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'}
+          created << {:host => '', :domain => @domain, :data => ['1.1.1.1'], :type => 'A'}
+          created.each {|r| @dns.create_host_record(r) }
+          # Search for records through the rubber DNS interface and make sure whe get what we expected.
+        
+          # Wildcard search.
+          records = @dns.find_host_records(:host => '*', :domain => @domain)
+          assert_equal 2, records.size
+        
+          # Blank hostname search.
+          records = @dns.find_host_records(:host => '', :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal '', records.first[:host]
+        
+          # Specific hostname search.
+          records = @dns.find_host_records(:host => 'host1', :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal 'host1', records.first[:host]
+        end
+      
+        should "find_many_records" do
+          omit "fog-zerigo implementation doesn't support pagination" if ENV['NO_FOG_MOCK']
+          
+          # Set up many sample records - more than the max returned
+          # in a single fetch so we can test pagination
+          max = 302
+          created = []
+          max.times do |i|
+            created << {:host => "host#{i}", :domain => @domain, :data => ['1.1.1.1'], :type => 'A'}
+          end
+          created.each {|r| @dns.create_host_record(r) }
+          
+          # Search for records through the rubber DNS interface and make sure whe get what we expected.
+        
+          # Wildcard search.
+          records = @dns.find_host_records(:host => '*', :domain => @domain)
+          assert_equal max, records.size
+        
+          # Specific hostname search.
+          records = @dns.find_host_records(:host => 'host1', :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal 'host1', records.first[:host]
+
+          # Specific hostname search.
+          records = @dns.find_host_records(:host => "host#{max - 1}", :domain => @domain)
+          assert_equal 1, records.size
+          assert_equal "host#{max - 1}", records.first[:host]
+        end
+      
+      end
+
+   end
+  
+end

data/test/instance_test.rb

@@ -220,7 +220,16 @@ class InstanceTest < Test::Unit::TestCase
         location = "file:#{Tempfile.new('instancestorage').path}"
         Instance.any_instance.expects(:load_from_file)
         Instance.any_instance.expects(:save_to_file)
-        Instance.new(location).save        
+        Instance.new(location).save
+      end
+
+      should "create new instance in filesystem when instance file doesn't exist" do
+        tempfile = Tempfile.new('instancestorage')
+        location = "file:#{tempfile.path}"
+        tempfile.unlink
+        Instance.any_instance.expects(:load_from_file).never
+        Instance.any_instance.expects(:save_to_file)
+        Instance.new(location).save
       end
       
       should "load and save from storage when storage given" do
@@ -229,7 +238,13 @@ class InstanceTest < Test::Unit::TestCase
         Instance.any_instance.expects(:save_to_file)
         Instance.new('storage:bucket/key').save        
       end
-      
+
+      should "create a new instance in cloud storage when the instance file doesn't exist" do
+        Instance.any_instance.expects(:load_from_file).never
+        Instance.any_instance.expects(:save_to_file)
+        Instance.new('storage:bucket/key').save
+      end
+
       should "load and save from table when table given" do
         Instance.any_instance.expects(:load_from_table)
         Instance.any_instance.expects(:save_to_table)

data/test/test_helper.rb

@@ -12,11 +12,46 @@ require 'tempfile'
 require 'fog'
 
 class Test::Unit::TestCase
+  # ENV['NO_FOG_MOCK'] = 'true'
+  
   setup do
-    Fog.mock!
+    Fog.mock! unless ENV['NO_FOG_MOCK'] 
   end
   
   teardown do
-    Fog::Mock.reset    
+    Fog::Mock.reset unless ENV['NO_FOG_MOCK'] 
+  end
+end
+
+
+SECRET = YAML.load_file(File.expand_path("~/rubber-secret.yml")) rescue {}
+
+def get_secret(path)
+  parts = path.split('.')
+  result = SECRET
+  
+  parts.each do |part|
+    result = result[part] if result
+  end
+  return result
+end
+
+TEST_DOMAIN = 'rubbertester'
+
+def all_test_zones(dns)
+  dns.client.zones.all.find_all {|z| z.domain =~ /#{TEST_DOMAIN}/ }
+end
+
+def destroy_test_domains(dns)
+  all_test_zones(dns).each do |zone|
+    # hardcoded failsafe to prevent destruction of real domains
+    raise "Trying to destroy non-rubber domain!" if zone.domain !~ /rubber/
+    
+    while zone.records.all.size != 0
+      zone.records.all.each do |record|
+        record.destroy
+      end
+    end
+    zone.destroy
   end
 end