zerigo_dns 1.1.0 → 1.1.1

data/lib/zerigo_dns.rb

@@ -43,8 +43,9 @@ module Zerigo
 
   
     class Zone < Base
+      # Find or Create Zone
       def self.find_or_create(domain)
-        zones = Zerigo::DNS::Zone.find(:all) 
+        zones = find(:all) 
         zone = nil
         zones.each do |z|
           if z.domain == domain
@@ -54,7 +55,7 @@ module Zerigo
         end
 
         unless zone
-          zone = Zerigo::DNS::Zone.create(:domain=> domain, :ns_type=>'pri_sec')
+          zone = create(:domain=> domain, :ns_type=>'pri_sec')
         end  
         zone
       end
@@ -62,6 +63,37 @@ module Zerigo
     end
   
     class Host < Base
+      
+      # Update or Create Host for a zone
+      def self.update_or_create(zone, hostname, type, data)
+        hosts = find(:all) 
+        host = nil
+        hosts.each do |h|
+          if h.hostname == hostname
+            host = h
+            break;
+          end
+        end
+
+        if host
+          # update
+          host.host_type = type
+          host.data = data
+          host.ttl = 86400
+          host.save
+        else
+          # creatae
+          host = create(
+            :zone_id => zone, 
+            :hostname => hostname,
+            :host_type => type,
+            :data => data,
+            :ttl=>86400
+          )
+        end  
+        host
+      end
+      
     end
 
   end

data/spec/lib/host_spec.rb

@@ -0,0 +1,27 @@
+require File.dirname(__FILE__) + '/../spec_helper'
+
+describe "Zerigo::DNS::Host.update_or_create" do
+  
+  it 'should create host' do
+    Zerigo::DNS::Host.stub!(:find).and_return([])
+    Zerigo::DNS::Host.stub!(:create).and_return(:success => true)
+    
+    Zerigo::DNS::Host.update_or_create(1, 'www', 'A', '10.10.10.10')[:success].should be_true
+  end
+
+  it 'should update host' do
+    jackhq = mock('Zerigo::DNS::Host')
+    jackhq.stub!(:hostname).and_return('www')
+    jackhq.stub!(:host_type=)
+    jackhq.stub!(:data=)
+    jackhq.stub!(:ttl=)
+    jackhq.stub!(:save).and_return(true)
+    
+    Zerigo::DNS::Host.stub!(:find).and_return([jackhq])
+    Zerigo::DNS::Host.stub!(:create).and_return(:success => false)
+    
+    Zerigo::DNS::Host.update_or_create(1, 'www', 'A', '10.10.10.10').hostname == 'www'
+  end
+  
+  
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: zerigo_dns
 version: !ruby/object:Gem::Version 
-  version: 1.1.0
+  version: 1.1.1
 platform: ruby
 authors: 
 - Tom Wilson
@@ -77,6 +77,7 @@ signing_key:
 specification_version: 3
 summary: Zerigo DNS Gem
 test_files: 
+- spec/lib/host_spec.rb
 - spec/lib/zone_spec.rb
 - spec/spec_helper.rb
 - test/helper.rb