lib/terraforming/resource/route53_zone.rb in terraforming-0.0.5 vs lib/terraforming/resource/route53_zone.rb in terraforming-0.1.0

@@ -1,27 +1,27 @@
 module Terraforming
   module Resource
     class Route53Zone
       include Terraforming::Util
 
-      def self.tf(client = Aws::Route53::Client.new)
+      def self.tf(client: Aws::Route53::Client.new)
         self.new(client).tf
       end
 
-      def self.tfstate(client = Aws::Route53::Client.new)
-        self.new(client).tfstate
+      def self.tfstate(client: Aws::Route53::Client.new, tfstate_base: nil)
+        self.new(client).tfstate(tfstate_base)
       end
 
       def initialize(client)
         @client = client
       end
 
       def tf
         apply_template(@client, "tf/route53_zone")
       end
 
-      def tfstate
+      def tfstate(tfstate_base)
         resources = hosted_zones.inject({}) do |result, hosted_zone|
           zone_id = zone_id_of(hosted_zone)
 
           attributes = {
             "id"=> zone_id,
@@ -39,10 +39,10 @@
           }
 
           result
         end
 
-        generate_tfstate(resources)
+        generate_tfstate(resources, tfstate_base)
       end
 
       private
 
       def hosted_zones