kitchen-google 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 76a9c3e0e6785ad0ce9e9a5f73eb41ffedd5ff13
-  data.tar.gz: 345f185a9329e09575fbc5247b8818a62c65f4e8
+  metadata.gz: 82664c984b7db17f6d04c2844f382e18e87047d2
+  data.tar.gz: a47939f2f42029ee55400669a5daee07848023b6
 SHA512:
-  metadata.gz: 427b6a8e6c0f9485e8072de9c6b68c5762a3f8fa991c747c88539a22242e6e76c64d2f98b8184b70f588c29632bdfcf69df4289610aca1f9149292530c557c28
-  data.tar.gz: dc81a39c4cf56add4985c31ea750b4a55be452cc354659dcda30e84a47d073bcb419b6a40572b47f1ba861725fd1de33e5d1dd018dc36fde529f9da972af56d5
+  metadata.gz: 52267771b0165f69641258d29dd85603574e402b5c506dcae6a2c5e3b4a1332fce62cb31826ae6a22139e98d5792abcc5fccde5d14ff0be110cdb4964f44c7f8
+  data.tar.gz: 7ff9cb61420f0d29c4f71b0dfe4678f5a76d755efa5e0a4b80f12efb496189c962f3b275ae801927cc3c0680d2765282e76f56f295d6ba42536db4d2d7ca9873

data/CHANGELOG.md

@@ -1,3 +1,10 @@
+## 1.1.0 / 2016-03-17
+
+ * #32: Rubocop and Rake fixes to address Travis test issues
+ * #33: Automatically disable auto-restart and auto-migrate when using preemptible instances
+ * #34: Support for using subnetworks
+ * #35: Support for gcloud-style image aliases (i.e. "centos-7" will get you the latest CentOS image)
+
 ## 1.0.0 / 2016-03-10
 
 ### New Features

data/README.md

@@ -125,6 +125,10 @@ public image, the appropriate public project will be searched.
 You can override the project in which to search for the image with the
 `image_project` parameter.
 
+Additionally, you can supply an image alias supported by the `gcloud compute instances create`
+command and kitchen-google will find the latest version of that image to use.
+For a full list of aliases, see the output of `gcloud compute instances create --help`.
+
 ### `zone`
 
 **Required if `region` is left blank.** The name of the GCE zone in which to
@@ -183,6 +187,10 @@ GCE instance type (size) to launch; default: `n1-standard-1`
 
 GCE network that instance will be attached to; default: `default`
 
+### `subnet`
+
+GCE subnetwork that instance will be attached to. Only applies to custom networks.
+
 ### `preemptible`
 
 If set to `true`, GCE instance will be brought up as a  [preemptible](https://cloud.google.com/compute/docs/instances/preemptible) virtual machine,

data/kitchen-google.gemspec

@@ -20,7 +20,7 @@ Gem::Specification.new do |s|
 
   s.add_development_dependency "bundler"
   s.add_development_dependency "pry"
-  s.add_development_dependency "rake"
+  s.add_development_dependency "rake", "~> 10.5"
   s.add_development_dependency "rspec"
   s.add_development_dependency "rubocop"
 

data/lib/kitchen/driver/gce.rb

@@ -53,6 +53,27 @@ module Kitchen
         "userinfo-email"     => "userinfo.email",
       }
 
+      IMAGE_ALIAS_MAP = {
+        "centos-6"           => { project: "centos-cloud",      prefix: "centos-6" },
+        "centos-7"           => { project: "centos-cloud",      prefix: "centos-7" },
+        "container-vm"       => { project: "google-containers", prefix: "container-vm" },
+        "coreos"             => { project: "coreos-cloud",      prefix: "coreos-stable" },
+        "debian-7"           => { project: "debian-cloud",      prefix: "debian-7-wheezy" },
+        "debian-7-backports" => { project: "debian-cloud",      prefix: "backports-debian-7-wheezy" },
+        "debian-8"           => { project: "debian-cloud",      prefix: "debian-8-jessie" },
+        "opensuse-13"        => { project: "opensuse-cloud",    prefix: "opensuse-13" },
+        "rhel-6"             => { project: "rhel-cloud",        prefix: "rhel-6" },
+        "rhel-7"             => { project: "rhel-cloud",        prefix: "rhel-7" },
+        "sles-11"            => { project: "suse-cloud",        prefix: "sles-11" },
+        "sles-12"            => { project: "suse-cloud",        prefix: "sles-12" },
+        "ubuntu-12-04"       => { project: "ubuntu-os-cloud",   prefix: "ubuntu-1204-precise" },
+        "ubuntu-14-04"       => { project: "ubuntu-os-cloud",   prefix: "ubuntu-1404-trusty" },
+        "ubuntu-15-04"       => { project: "ubuntu-os-cloud",   prefix: "ubuntu-1504-vivid" },
+        "ubuntu-15-10"       => { project: "ubuntu-os-cloud",   prefix: "ubuntu-1510-wily" },
+        "windows-2008-r2"    => { project: "windows-cloud",     prefix: "windows-server-2008-r2" },
+        "windows-2012-r2"    => { project: "windows-cloud",     prefix: "windows-server-2012-r2" },
+      }
+
       kitchen_driver_api_version 2
       plugin_version Kitchen::Driver::GCE_VERSION
 
@@ -67,6 +88,7 @@ module Kitchen
       default_config :disk_type, "pd-standard"
       default_config :machine_type, "n1-standard-1"
       default_config :network, "default"
+      default_config :subnet, nil
       default_config :inst_name, nil
       default_config :service_account_name, "default"
       default_config :service_account_scopes, []
@@ -144,10 +166,14 @@ module Kitchen
         raise "Region #{config[:region]} is not a valid region" if config[:region] && !valid_region?
         raise "Machine type #{config[:machine_type]} is not valid" unless valid_machine_type?
         raise "Disk type #{config[:disk_type]} is not valid" unless valid_disk_type?
+        raise "Disk image #{config[:image_name]} is not valid - check your image name and image project" if boot_disk_source_image.nil?
         raise "Network #{config[:network]} is not valid" unless valid_network?
+        raise "Subnet #{config[:subnet]} is not valid" if config[:subnet] and !valid_subnet?
         raise "Email address of GCE user is not set" if winrm_transport? && config[:email].nil?
 
         warn("Both zone and region specified - region will be ignored.") if config[:zone] && config[:region]
+        warn("Auto-migrate disabled for preemptible instance") if preemptible? && config[:auto_migrate]
+        warn("Auto-restart disabled for preemptible instance") if preemptible? && config[:auto_restart]
       end
 
       def connection
@@ -195,7 +221,7 @@ module Kitchen
       end
 
       def check_api_call(&block)
-        block.call
+        yield
       rescue Google::Apis::ClientError => e
         debug("API error: #{e.message}")
         false
@@ -217,6 +243,11 @@ module Kitchen
         check_api_call { connection.get_network(project, config[:network]) }
       end
 
+      def valid_subnet?
+        return false if config[:subnet].nil?
+        check_api_call { connection.get_subnetwork(project, region, config[:subnet]) }
+      end
+
       def valid_zone?
         return false if config[:zone].nil?
         check_api_call { connection.get_zone(project, config[:zone]) }
@@ -245,7 +276,11 @@ module Kitchen
       end
 
       def region
-        config[:region]
+        config[:region].nil? ? region_for_zone : config[:region]
+      end
+
+      def region_for_zone
+        @region_for_zone ||= connection.get_zone(project, zone).region.split("/").last
       end
 
       def zone
@@ -320,7 +355,7 @@ module Kitchen
         params.disk_name    = server_name
         params.disk_size_gb = config[:disk_size]
         params.disk_type    = disk_type_url_for(config[:disk_type])
-        params.source_image = disk_image_url
+        params.source_image = boot_disk_source_image
 
         disk.initialize_params = params
         disk
@@ -330,21 +365,40 @@ module Kitchen
         "zones/#{zone}/diskTypes/#{type}"
       end
 
+      def boot_disk_source_image
+        @boot_disk_source ||= disk_image_url
+      end
+
       def disk_image_url
         # if the user provided an image_project, assume they want it, no questions asked
-        return image_url_for(config[:image_project], config[:image_name]) unless config[:image_project].nil?
+        unless config[:image_project].nil?
+          debug("Searching project #{config[:image_project]} for image #{config[:image_name]}")
+          return image_url_for(config[:image_project], config[:image_name])
+        end
+
+        # No image project has been provided. Check to see if the image is a known alias.
+        alias_url = image_alias_url
+        unless alias_url.nil?
+          debug("Image #{config[:image_name]} is a known alias - using image URL: #{alias_url}")
+          return alias_url
+        end
 
-        # no image project has been provided. We'll first check the user's project for the image.
+        # Doesn't match an alias. Let's check the user's project for the image.
         url = image_url_for(project, config[:image_name])
-        return url unless url.nil?
+        unless url.nil?
+          debug("Located image #{config[:image_name]} in project #{project} - using image URL: #{url}")
+          return url
+        end
 
         # Image not found in user's project. Is there a public project this image might exist in?
         public_project = public_project_for_image(config[:image_name])
         if public_project
+          debug("Searching public image project #{public_project} for image #{config[:image_name]}")
           return image_url_for(public_project, config[:image_name])
         end
 
         # No image in user's project or public project, so it doesn't exist.
+        error("Image search failed for image #{config[:image_name]} - no suitable image found")
         nil
       end
 
@@ -352,6 +406,23 @@ module Kitchen
         return "projects/#{image_project}/global/images/#{image_name}" if image_exist?(image_project, image_name)
       end
 
+      def image_alias_url
+        image_alias = config[:image_name]
+        return unless IMAGE_ALIAS_MAP.key?(image_alias)
+
+        image_project = IMAGE_ALIAS_MAP[image_alias][:project]
+        image_prefix  = IMAGE_ALIAS_MAP[image_alias][:prefix]
+
+        latest_image = connection.list_images(image_project).items
+          .select { |image| image.name.start_with?(image_prefix) }
+          .sort { |a, b| a.name <=> b.name }
+          .last
+
+        return if latest_image.nil?
+
+        latest_image.self_link
+      end
+
       def public_project_for_image(image)
         case image
         when /centos/
@@ -403,6 +474,7 @@ module Kitchen
       def instance_network_interfaces
         interface                = Google::Apis::ComputeV1::NetworkInterface.new
         interface.network        = network_url
+        interface.subnetwork     = subnet_url if subnet_url
         interface.access_configs = interface_access_configs
 
         Array(interface)
@@ -412,6 +484,12 @@ module Kitchen
         "projects/#{project}/global/networks/#{config[:network]}"
       end
 
+      def subnet_url
+        return unless config[:subnet]
+
+        "projects/#{project}/regions/#{region}/subnetworks/#{config[:subnet]}"
+      end
+
       def interface_access_configs
         return [] if config[:use_private_ip]
 
@@ -424,14 +502,26 @@ module Kitchen
 
       def instance_scheduling
         Google::Apis::ComputeV1::Scheduling.new.tap do |scheduling|
-          scheduling.automatic_restart   = config[:auto_restart].to_s
-          scheduling.preemptible         = config[:preemptible].to_s
+          scheduling.automatic_restart   = auto_restart?.to_s
+          scheduling.preemptible         = preemptible?.to_s
           scheduling.on_host_maintenance = migrate_setting
         end
       end
 
+      def preemptible?
+        config[:preemptible]
+      end
+
+      def auto_migrate?
+        preemptible? ? false : config[:auto_migrate]
+      end
+
+      def auto_restart?
+        preemptible? ? false : config[:auto_restart]
+      end
+
       def migrate_setting
-        config[:auto_migrate] ? "MIGRATE" : "TERMINATE"
+        auto_migrate? ? "MIGRATE" : "TERMINATE"
       end
 
       def instance_service_accounts
@@ -471,7 +561,7 @@ module Kitchen
         begin
           Timeout.timeout(wait_time) do
             loop do
-              item = block.call
+              item = yield
               current_status = item.status
 
               unless last_status == current_status

data/lib/kitchen/driver/gce_version.rb

@@ -18,6 +18,6 @@
 
 module Kitchen
   module Driver
-    GCE_VERSION = "1.0.0".freeze
+    GCE_VERSION = "1.1.0".freeze
   end
 end

data/spec/kitchen/driver/gce_spec.rb

@@ -71,6 +71,7 @@ describe Kitchen::Driver::Gce do
     allow(driver).to receive(:instance).and_return(instance)
     allow(driver).to receive(:project).and_return("test_project")
     allow(driver).to receive(:zone).and_return("test_zone")
+    allow(driver).to receive(:region).and_return("test_region")
   end
 
   it "driver API version is 2" do
@@ -193,6 +194,7 @@ describe Kitchen::Driver::Gce do
         zone:         "test_zone",
         machine_type: "test_machine_type",
         disk_type:    "test_disk_type",
+        image_name:   "test_image",
         network:      "test_network",
       }
     end
@@ -203,7 +205,9 @@ describe Kitchen::Driver::Gce do
       allow(driver).to receive(:valid_region?).and_return(true)
       allow(driver).to receive(:valid_machine_type?).and_return(true)
       allow(driver).to receive(:valid_disk_type?).and_return(true)
+      allow(driver).to receive(:boot_disk_source_image).and_return("image")
       allow(driver).to receive(:valid_network?).and_return(true)
+      allow(driver).to receive(:valid_subnet?).and_return(true)
       allow(driver).to receive(:winrm_transport?).and_return(false)
       allow(driver).to receive(:config).and_return(config)
     end
@@ -253,6 +257,24 @@ describe Kitchen::Driver::Gce do
       end
     end
 
+    context "when subnet is set" do
+      let(:config) do
+        {
+          project:      "test_project",
+          zone:         "test_zone",
+          machine_type: "test_machine_type",
+          disk_type:    "test_disk_type",
+          network:      "test_network",
+          subnet:       "test_subnet",
+        }
+      end
+
+      it "raises an exception if the subnet is invalid" do
+        expect(driver).to receive(:valid_subnet?).and_return(false)
+        expect { driver.validate! }.to raise_error(RuntimeError, "Subnet test_subnet is not valid")
+      end
+    end
+
     it "raises an exception if the project is invalid" do
       expect(driver).to receive(:valid_project?).and_return(false)
       expect { driver.validate! }.to raise_error(RuntimeError, "Project test_project is not a valid project")
@@ -268,6 +290,11 @@ describe Kitchen::Driver::Gce do
       expect { driver.validate! }.to raise_error(RuntimeError, "Disk type test_disk_type is not valid")
     end
 
+    it "raises an exception if the boot disk source image is invalid" do
+      expect(driver).to receive(:boot_disk_source_image).and_return(nil)
+      expect { driver.validate! }.to raise_error(RuntimeError, "Disk image test_image is not valid - check your image name and image project")
+    end
+
     it "raises an exception if the network is invalid" do
       expect(driver).to receive(:valid_network?).and_return(false)
       expect { driver.validate! }.to raise_error(RuntimeError, "Network test_network is not valid")
@@ -372,6 +399,11 @@ describe Kitchen::Driver::Gce do
     it_behaves_like "a validity checker", :network, :get_network, "test_project"
   end
 
+  describe '#valid_subnet?' do
+    subject { driver.valid_subnet? }
+    it_behaves_like "a validity checker", :subnet, :get_subnetwork, "test_project", "test_region"
+  end
+
   describe '#valid_zone?' do
     subject { driver.valid_zone? }
     it_behaves_like "a validity checker", :zone, :get_zone, "test_project"
@@ -414,10 +446,29 @@ describe Kitchen::Driver::Gce do
   end
 
   describe '#region' do
-    it "returns the region from the config" do
-      expect(driver).to receive(:config).and_return(region: "my_region")
+    it "returns the region from the config if specified" do
+      allow(driver).to receive(:region).and_call_original
+      allow(driver).to receive(:config).and_return(region: "my_region")
       expect(driver.region).to eq("my_region")
     end
+
+    it "returns the region for the zone if no region is specified" do
+      allow(driver).to receive(:region).and_call_original
+      allow(driver).to receive(:config).and_return({})
+      expect(driver).to receive(:region_for_zone).and_return("zone_region")
+      expect(driver.region).to eq("zone_region")
+    end
+  end
+
+  describe '#region_for_zone' do
+    it "returns the region for a given zone" do
+      connection = double("connection")
+      zone_obj   = double("zone_obj", region: "/path/to/test_region")
+
+      expect(driver).to receive(:connection).and_return(connection)
+      expect(connection).to receive(:get_zone).with(project, zone).and_return(zone_obj)
+      expect(driver.region_for_zone).to eq("test_region")
+    end
   end
 
   describe '#zone' do
@@ -614,33 +665,49 @@ describe Kitchen::Driver::Gce do
     end
 
     context "when the user does not supply an image project" do
-      let(:config) { { image_name: "my_image" } }
 
-      context "when the image exists in the user's project" do
-        it "returns the image URL" do
-          expect(driver).to receive(:image_url_for).with(project, "my_image").and_return("image_url")
-          expect(driver.disk_image_url).to eq("image_url")
+      context "when the image provided is an alias" do
+        let(:config) { { image_name: "image_alias" } }
+
+        it "returns the alias URL" do
+          expect(driver).to receive(:image_alias_url).and_return("image_alias_url")
+          expect(driver.disk_image_url).to eq("image_alias_url")
         end
       end
 
-      context "when the image does not exist in the user's project" do
+      context "when the image provided is not an alias" do
+        let(:config) { { image_name: "my_image" } }
+
         before do
-          expect(driver).to receive(:image_url_for).with(project, "my_image").and_return(nil)
+          expect(driver).to receive(:image_alias_url).and_return(nil)
         end
 
-        context "when the image matches a known public project" do
-          it "returns the image URL from the public project" do
-            expect(driver).to receive(:public_project_for_image).with("my_image").and_return("public_project")
-            expect(driver).to receive(:image_url_for).with("public_project", "my_image").and_return("image_url")
+        context "when the image exists in the user's project" do
+          it "returns the image URL" do
+            expect(driver).to receive(:image_url_for).with(project, "my_image").and_return("image_url")
             expect(driver.disk_image_url).to eq("image_url")
           end
         end
 
-        context "when the image does not match a known project" do
-          it "returns nil" do
-            expect(driver).to receive(:public_project_for_image).with("my_image").and_return(nil)
-            expect(driver).not_to receive(:image_url_for)
-            expect(driver.disk_image_url).to eq(nil)
+        context "when the image does not exist in the user's project" do
+          before do
+            expect(driver).to receive(:image_url_for).with(project, "my_image").and_return(nil)
+          end
+
+          context "when the image matches a known public project" do
+            it "returns the image URL from the public project" do
+              expect(driver).to receive(:public_project_for_image).with("my_image").and_return("public_project")
+              expect(driver).to receive(:image_url_for).with("public_project", "my_image").and_return("image_url")
+              expect(driver.disk_image_url).to eq("image_url")
+            end
+          end
+
+          context "when the image does not match a known project" do
+            it "returns nil" do
+              expect(driver).to receive(:public_project_for_image).with("my_image").and_return(nil)
+              expect(driver).not_to receive(:image_url_for)
+              expect(driver.disk_image_url).to eq(nil)
+            end
           end
         end
       end
@@ -659,6 +726,60 @@ describe Kitchen::Driver::Gce do
     end
   end
 
+  describe '#image_alias_url' do
+    before do
+      allow(driver).to receive(:config).and_return(config)
+    end
+
+    context "when the image_alias is not a valid alias" do
+      let(:config) { { image_name: "fake_alias" } }
+
+      it "returns nil" do
+        expect(driver.image_alias_url).to eq(nil)
+      end
+    end
+
+    context "when the image_alias is a valid alias" do
+      let(:config)     { { image_name: "centos-7" } }
+      let(:connection) { double("connection") }
+
+      before do
+        allow(driver).to receive(:connection).and_return(connection)
+        allow(connection).to receive(:list_images).and_return(response)
+      end
+
+      context "when the response contains no images" do
+        let(:response) { double("response", items: []) }
+
+        it "returns nil" do
+          expect(driver.image_alias_url).to eq(nil)
+        end
+      end
+
+      context "when the response contains images but none match the name" do
+        let(:image1)   { double("image1", name: "centos-6-v20150101") }
+        let(:image2)   { double("image2", name: "centos-6-v20150202") }
+        let(:image3)   { double("image3", name: "ubuntu-14-v20150303") }
+        let(:response) { double("response", items: [ image1, image2, image3 ]) }
+
+        it "returns nil" do
+          expect(driver.image_alias_url).to eq(nil)
+        end
+      end
+
+      context "when the response contains images that match the name" do
+        let(:image1)   { double("image1", name: "centos-7-v20160201", self_link: "image1_selflink") }
+        let(:image2)   { double("image2", name: "centos-7-v20160301", self_link: "image2_selflink") }
+        let(:image3)   { double("image3", name: "centos-6-v20160401", self_link: "image3_selflink") }
+        let(:response) { double("response", items: [ image1, image2, image3 ]) }
+
+        it "returns the link for image2 which is the most recent image" do
+          expect(driver.image_alias_url).to eq("image2_selflink")
+        end
+      end
+    end
+  end
+
   describe '#public_project_for_image' do
     {
       "centos"         => "centos-cloud",
@@ -722,17 +843,45 @@ describe Kitchen::Driver::Gce do
   end
 
   describe '#instance_network_interfaces' do
-    it "returns an array containing a properly-formatted interface" do
-      interface = double("interface")
+    let(:interface) { double("interface") }
 
-      expect(driver).to receive(:network_url).and_return("network_url")
-      expect(driver).to receive(:interface_access_configs).and_return("access_configs")
+    before do
+      allow(Google::Apis::ComputeV1::NetworkInterface).to receive(:new).and_return(interface)
+      allow(driver).to receive(:network_url)
+      allow(driver).to receive(:subnet_url)
+      allow(driver).to receive(:interface_access_configs)
+      allow(interface).to receive(:network=)
+      allow(interface).to receive(:subnetwork=)
+      allow(interface).to receive(:access_configs=)
+    end
 
+    it "creates a network interface object and returns it" do
       expect(Google::Apis::ComputeV1::NetworkInterface).to receive(:new).and_return(interface)
+      expect(driver.instance_network_interfaces).to eq([interface])
+    end
+
+    it "sets the network" do
+      expect(driver).to receive(:network_url).and_return("network_url")
       expect(interface).to receive(:network=).with("network_url")
+      driver.instance_network_interfaces
+    end
+
+    it "sets the access configs" do
+      expect(driver).to receive(:interface_access_configs).and_return("access_configs")
       expect(interface).to receive(:access_configs=).with("access_configs")
+      driver.instance_network_interfaces
+    end
 
-      expect(driver.instance_network_interfaces).to eq([interface])
+    it "does not set a subnetwork by default" do
+      allow(driver).to receive(:subnet_url).and_return(nil)
+      expect(interface).not_to receive(:subnetwork=)
+      driver.instance_network_interfaces
+    end
+
+    it "sets a subnetwork if one was specified" do
+      allow(driver).to receive(:subnet_url).and_return("subnet_url")
+      expect(interface).to receive(:subnetwork=).with("subnet_url")
+      driver.instance_network_interfaces
     end
   end
 
@@ -743,6 +892,19 @@ describe Kitchen::Driver::Gce do
     end
   end
 
+  describe '#subnet_url_for' do
+    it "returns nil if no subnet is specified" do
+      expect(driver).to receive(:config).and_return({})
+      expect(driver.subnet_url).to eq(nil)
+    end
+
+    it "returns a properly-formatted subnet URL" do
+      allow(driver).to receive(:config).and_return(subnet: "test_subnet")
+      expect(driver).to receive(:region).and_return("test_region")
+      expect(driver.subnet_url).to eq("projects/test_project/regions/test_region/subnetworks/test_subnet")
+    end
+  end
+
   describe '#interface_access_configs' do
     it "returns a properly-configured access config object" do
       access_config = double("access_config")
@@ -765,24 +927,58 @@ describe Kitchen::Driver::Gce do
     it "returns a properly-configured scheduling object" do
       scheduling = double("scheduling")
 
-      allow(driver).to receive(:config).and_return(auto_restart: true, preemptible: false)
+      expect(driver).to receive(:auto_restart?).and_return("restart")
+      expect(driver).to receive(:preemptible?).and_return("preempt")
       expect(driver).to receive(:migrate_setting).and_return("host_maintenance")
       expect(Google::Apis::ComputeV1::Scheduling).to receive(:new).and_return(scheduling)
-      expect(scheduling).to receive(:automatic_restart=).with("true")
-      expect(scheduling).to receive(:preemptible=).with("false")
+      expect(scheduling).to receive(:automatic_restart=).with("restart")
+      expect(scheduling).to receive(:preemptible=).with("preempt")
       expect(scheduling).to receive(:on_host_maintenance=).with("host_maintenance")
       expect(driver.instance_scheduling).to eq(scheduling)
     end
   end
 
+  describe '#preemptible?' do
+    it "returns the preemptible setting from the config" do
+      expect(driver).to receive(:config).and_return(preemptible: "test_preempt")
+      expect(driver.preemptible?).to eq("test_preempt")
+    end
+  end
+
+  describe '#auto_migrate?' do
+    it "returns false if the instance is preemptible" do
+      expect(driver).to receive(:preemptible?).and_return(true)
+      expect(driver.auto_migrate?).to eq(false)
+    end
+
+    it "returns the setting from the config if preemptible is false" do
+      expect(driver).to receive(:config).and_return(auto_migrate: "test_migrate")
+      expect(driver).to receive(:preemptible?).and_return(false)
+      expect(driver.auto_migrate?).to eq("test_migrate")
+    end
+  end
+
+  describe '#auto_restart?' do
+    it "returns false if the instance is preemptible" do
+      expect(driver).to receive(:preemptible?).and_return(true)
+      expect(driver.auto_restart?).to eq(false)
+    end
+
+    it "returns the setting from the config if preemptible is false" do
+      expect(driver).to receive(:config).and_return(auto_restart: "test_restart")
+      expect(driver).to receive(:preemptible?).and_return(false)
+      expect(driver.auto_restart?).to eq("test_restart")
+    end
+  end
+
   describe '#migrate_setting' do
     it "returns MIGRATE if auto_migrate is true" do
-      expect(driver).to receive(:config).and_return(auto_migrate: true)
+      expect(driver).to receive(:auto_migrate?).and_return(true)
       expect(driver.migrate_setting).to eq("MIGRATE")
     end
 
     it "returns TERMINATE if auto_migrate is false" do
-      expect(driver).to receive(:config).and_return(auto_migrate: false)
+      expect(driver).to receive(:auto_migrate?).and_return(false)
       expect(driver.migrate_setting).to eq("TERMINATE")
     end
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: kitchen-google
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Andrew Leonard
@@ -85,16 +85,16 @@ dependencies:
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '10.5'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '10.5'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement