Use only ZoneDistribution

dnsimple · Jul 19, 2018 · b1ee738 · b1ee738
1 parent f53541d
commit b1ee738
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 16 deletions.
diff --git a/lib/dnsimple/client/zones_distributions.rb b/lib/dnsimple/client/zones_distributions.rb
@@ -32,14 +32,14 @@ def zone_distribution(account_id, zone_id, options = {})
       # @param  [String] zone_id the zone name
       # @param  [Integer] record_id the record ID
       # @param  [Hash] options
-      # @return [Dnsimple::Response<Dnsimple::Struct::ZoneRecordDistribution>]
+      # @return [Dnsimple::Response<Dnsimple::Struct::ZoneDistribution>]
       #
       # @raise  [Dnsimple::NotFoundError]
       # @raise  [Dnsimple::RequestError]
       def zone_record_distribution(account_id, zone_id, record_id, options = {})
         response = client.get(Client.versioned("/%s/zones/%s/records/%s/distribution" % [account_id, zone_id, record_id]), options)
 
-        Dnsimple::Response.new(response, Struct::ZoneRecordDistribution.new(response["data"]))
+        Dnsimple::Response.new(response, Struct::ZoneDistribution.new(response["data"]))
       end
 
     end

diff --git a/lib/dnsimple/struct.rb b/lib/dnsimple/struct.rb
@@ -43,6 +43,5 @@ def initialize(attributes = {})
 require_relative 'struct/zone'
 require_relative 'struct/zone_file'
 require_relative 'struct/zone_distribution'
-require_relative 'struct/zone_record_distribution'
 require_relative 'struct/webhook'
 require_relative 'struct/whoami'
diff --git a/lib/dnsimple/struct/zone_record_distribution.rb b/lib/dnsimple/struct/zone_record_distribution.rb
diff --git a/spec/dnsimple/client/zones_distributions_spec.rb b/spec/dnsimple/client/zones_distributions_spec.rb
@@ -84,7 +84,7 @@
       expect(response).to be_a(Dnsimple::Response)
 
       result = response.data
-      expect(result).to be_a(Dnsimple::Struct::ZoneRecordDistribution)
+      expect(result).to be_a(Dnsimple::Struct::ZoneDistribution)
       expect(result.distributed).to be(true)
     end
 
@@ -96,7 +96,7 @@
       expect(response).to be_a(Dnsimple::Response)
 
       result = response.data
-      expect(result).to be_a(Dnsimple::Struct::ZoneRecordDistribution)
+      expect(result).to be_a(Dnsimple::Struct::ZoneDistribution)
       expect(result.distributed).to be(false)
     end