kitchen-google 0.3.0 → 1.0.0

data/lib/kitchen/driver/gce_version.rb

@@ -0,0 +1,23 @@
+#
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
+# Copyright:: Copyright (c) 2015 Chef Software, Inc.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+module Kitchen
+  module Driver
+    GCE_VERSION = "1.0.0".freeze
+  end
+end

data/spec/kitchen/driver/gce_spec.rb

@@ -1,8 +1,9 @@
 # -*- coding: utf-8 -*-
 #
 # Author:: Andrew Leonard (<andy@hurricane-ridge.com>)
+# Author:: Chef Partner Engineering (<partnereng@chef.io>)
 #
-# Copyright (C) 2013-2014, Andrew Leonard
+# Copyright (C) 2013-2016, Andrew Leonard and Chef Software, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -16,383 +17,945 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-require_relative '../../spec_helper.rb'
+require "spec_helper"
+require "google/apis/compute_v1"
+require "kitchen/driver/gce"
+require "kitchen/provisioner/dummy"
+require "kitchen/transport/dummy"
+require "kitchen/verifier/dummy"
+
+shared_examples_for "a validity checker" do |config_key, api_method, *args|
+  it "returns false if the config value is nil" do
+    expect(driver).to receive(:config).and_return({})
+    expect(subject).to eq(false)
+  end
 
-require 'resolv'
+  it "checks the outcome of the API call" do
+    connection = double("connection")
+    allow(driver).to receive(:config).and_return({ config_key => "test_value" })
+    expect(driver).to receive(:connection).and_return(connection)
+    expect(connection).to receive(api_method).with(*args, "test_value")
+    expect(driver).to receive(:check_api_call).and_call_original
+    expect(subject).to eq(true)
+  end
+end
 
 describe Kitchen::Driver::Gce do
+  let(:logged_output) { StringIO.new }
+  let(:logger)        { Logger.new(logged_output) }
+  let(:platform)      { Kitchen::Platform.new(name: "fake_platform") }
+  let(:transport)     { Kitchen::Transport::Dummy.new }
+  let(:driver)        { Kitchen::Driver::Gce.new(config) }
+
+  let(:project)       { "test_project" }
+  let(:zone)          { "test_zone" }
 
   let(:config) do
-    { google_client_email: '123456789012@developer.gserviceaccount.com',
-      google_project: 'alpha-bravo-123'
+    {
+      project:    project,
+      zone:       zone,
+      image_name: "test_image",
     }
   end
 
-  let(:state) { Hash.new }
+  let(:instance) do
+    instance_double(Kitchen::Instance,
+                    logger:    logger,
+                    transport: transport,
+                    platform:  platform,
+                    to_str:    "instance_str"
+                   )
+  end
 
-  let(:logged_output) { StringIO.new }
-  let(:logger) { Logger.new(logged_output) }
+  before 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")
+  end
 
-  let(:instance) do
-    double(
-      logger: logger,
-      name: 'default-distro-12'
-    )
-  end
-
-  let(:driver) do
-    d = Kitchen::Driver::Gce.new(config)
-    allow(d).to receive(:instance) { instance }
-    allow(d).to receive(:wait_for_sshd) { true }
-    d
-  end
-
-  let(:fog) do
-    Fog::Compute::Google::Mock.new({})
-  end
-
-  let(:disk) do
-    fog.disks.create(
-      name: 'rspec-test-disk',
-      size_gb: 10,
-      zone_name: 'us-central1-b',
-      source_image: 'debian-7-wheezy-v20130816'
-    )
-  end
-
-  let(:server) do
-    fog.servers.create(
-      name: 'rspec-test-instance',
-      disks: [disk],
-      machine_type: 'n1-standard-1',
-      zone_name: 'us-central1-b'
-    )
-  end
-
-  before(:each) do
-    Fog.mock!
-    Fog::Mock.reset
-    Fog::Mock.delay = 0
-  end
-
-  describe '#initialize' do
-    context 'with default options' do
-
-      defaults = {
-        area: 'us-central1',
-        autodelete_disk: true,
-        disk_size: 10,
-        inst_name: nil,
-        machine_type: 'n1-standard-1',
-        network: 'default',
-        region: nil,
-        service_accounts: nil,
-        tags: [],
-        username: ENV['USER'],
-        zone_name: nil,
-        google_key_location: nil,
-        google_json_key_location: nil,
-        preemptible: false,
-        auto_restart: false
-      }
+  it "driver API version is 2" do
+    expect(driver.diagnose_plugin[:api_version]).to eq(2)
+  end
 
-      defaults.each do |k, v|
-        it "sets the correct default for #{k}" do
-          expect(driver[k]).to eq(v)
-        end
-      end
+  describe '#name' do
+    it "has an overridden name" do
+      expect(driver.name).to eq("Google Compute (GCE)")
+    end
+  end
+
+  describe '#create' do
+    let(:connection) { double("connection") }
+    let(:operation)  { double("operation", name: "test_operation") }
+    let(:state)      { {} }
+
+    before do
+      allow(driver).to receive(:validate!)
+      allow(driver).to receive(:connection).and_return(connection)
+      allow(driver).to receive(:generate_server_name)
+      allow(driver).to receive(:wait_for_operation)
+      allow(driver).to receive(:server_instance)
+      allow(driver).to receive(:create_instance_object)
+      allow(driver).to receive(:ip_address_for)
+      allow(driver).to receive(:update_windows_password)
+      allow(driver).to receive(:wait_for_server)
+      allow(connection).to receive(:insert_instance).and_return(operation)
+    end
+
+    it "does not create the server if the hostname is in the state file" do
+      expect(connection).not_to receive(:insert_instance)
+      driver.create(server_name: "server_exists")
+    end
+
+    it "generates a unique server name and sets the state" do
+      expect(driver).to receive(:generate_server_name).and_return("server_1")
+      driver.create(state)
+      expect(state[:server_name]).to eq("server_1")
     end
 
-    context 'with overriden options' do
-      overrides = {
-        area: 'europe-west',
-        autodelete_disk: false,
-        disk_size: 15,
-        inst_name: 'ci-instance',
-        machine_type: 'n1-highmem-8',
-        network: 'dev-net',
-        region: 'asia-east1',
-        service_accounts: %w(userdata.email compute.readonly),
-        tags: %w(qa integration),
-        username: 'root',
-        zone_name: 'europe-west1-a',
-        google_key_location: '/path/to/foo.p12',
-        google_json_key_location: '/path/to/bar.json',
-        preemptible: true,
-        auto_restart: false # because of preemptible, see kitchen-google/pull/22
+    it "creates the instance via the API and waits for it to complete" do
+      expect(driver).to receive(:generate_server_name).and_return("server_1")
+      expect(driver).to receive(:create_instance_object).with("server_1").and_return("create_obj")
+      expect(connection).to receive(:insert_instance).with("test_project", "test_zone", "create_obj").and_return(operation)
+      expect(driver).to receive(:wait_for_operation).with(operation)
+
+      driver.create(state)
+    end
+
+    it "sets the correct data in the state object" do
+      expect(driver).to receive(:generate_server_name).and_return("server_1")
+      expect(driver).to receive(:server_instance).with("server_1").and_return("server_obj")
+      expect(driver).to receive(:ip_address_for).with("server_obj").and_return("1.2.3.4")
+      driver.create(state)
+
+      expect(state[:server_name]).to eq("server_1")
+      expect(state[:hostname]).to eq("1.2.3.4")
+      expect(state[:zone]).to eq("test_zone")
+    end
+
+    it "updates the windows password" do
+      expect(driver).to receive(:generate_server_name).and_return("server_1")
+      expect(driver).to receive(:update_windows_password).with("server_1")
+      driver.create(state)
+    end
+
+    it "waits for the server to be ready" do
+      expect(driver).to receive(:wait_for_server)
+      driver.create(state)
+    end
+
+    it "destroys the server if any exceptions are raised" do
+      expect(connection).to receive(:insert_instance).and_raise(RuntimeError)
+      expect(driver).to receive(:destroy).with(state)
+      expect { driver.create(state) }.to raise_error(RuntimeError)
+    end
+  end
+
+  describe '#destroy' do
+    let(:connection) { double("connection") }
+    let(:state)      { { server_name: "server_1", hostname: "test_host", zone: "test_zone" } }
+
+    before do
+      allow(driver).to receive(:connection).and_return(connection)
+      allow(driver).to receive(:server_exist?).and_return(true)
+      allow(driver).to receive(:wait_for_operation)
+      allow(connection).to receive(:delete_instance)
+    end
+
+    it "does not attempt to delete the instance if there is no server_name" do
+      expect(connection).not_to receive(:delete_instance)
+      driver.destroy({})
+    end
+
+    it "does not attempt to delete the instance if it does not exist" do
+      expect(driver).to receive(:server_exist?).with("server_1").and_return(false)
+      expect(connection).not_to receive(:delete_instance)
+      driver.destroy(state)
+    end
+
+    it "deletes the instance via the API and waits for it to complete" do
+      expect(connection).to receive(:delete_instance).with("test_project", "test_zone", "server_1").and_return("operation")
+      expect(driver).to receive(:wait_for_operation).with("operation")
+      driver.destroy(state)
+    end
+
+    it "deletes the state keys" do
+      driver.destroy(state)
+      expect(state.key?(:server_name)).to eq(false)
+      expect(state.key?(:hostname)).to eq(false)
+      expect(state.key?(:zone)).to eq(false)
+    end
+  end
+
+  describe '#validate!' do
+    let(:config) do
+      {
+        project:      "test_project",
+        zone:         "test_zone",
+        machine_type: "test_machine_type",
+        disk_type:    "test_disk_type",
+        network:      "test_network",
       }
+    end
 
-      let(:config) { overrides }
+    before do
+      allow(driver).to receive(:valid_project?).and_return(true)
+      allow(driver).to receive(:valid_zone?).and_return(true)
+      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(:valid_network?).and_return(true)
+      allow(driver).to receive(:winrm_transport?).and_return(false)
+      allow(driver).to receive(:config).and_return(config)
+    end
 
-      overrides.each do |k, v|
-        it "overrides the default value for #{k}" do
-          expect(driver[k]).to eq(v)
-        end
+    it "does not raise an exception when all validations are successful" do
+      expect { driver.validate! }.not_to raise_error
+    end
+
+    context "when neither zone nor region are specified" do
+      let(:config) { {} }
+      it "raises an exception" do
+        expect { driver.validate! }.to raise_error(RuntimeError, "Either zone or region must be specified")
       end
     end
-  end
 
-  describe '#connection' do
-    context 'with required variables set' do
-      it 'returns a Fog Compute object' do
-        expect(driver.send(:connection)).to be_a(Fog::Compute::Google::Mock)
+    context "when zone and region are both set" do
+      let(:config) { { zone: "test_zone", region: "test_region" } }
+
+      it "warns the user that the region will be ignored" do
+        expect(driver).to receive(:warn).with("Both zone and region specified - region will be ignored.")
+        driver.validate!
       end
+    end
 
-      it 'uses the v1 api version' do
-        conn = driver.send(:connection)
-        expect(conn.api_version).to eq('v1')
+    context "when region is set to 'any'" do
+      let(:config) { { region: "any" } }
+      it "raises an exception" do
+        expect { driver.validate! }.to raise_error(RuntimeError, "'any' is no longer a valid region")
       end
     end
 
-    context 'without required variables set' do
-      let(:config) { Hash.new }
+    context "when zone is set" do
+      let(:config) { { zone: "test_zone" } }
 
-      it 'raises an error' do
-        expect { driver.send(:connection) }.to raise_error(ArgumentError)
+      it "raises an exception if the zone is not valid" do
+        expect(driver).to receive(:valid_zone?).and_return(false)
+        expect { driver.validate! }.to raise_error(RuntimeError, "Zone test_zone is not a valid zone")
       end
     end
-  end
 
-  describe '#create' do
-    context 'with an existing server' do
-      let(:state) do
-        s = Hash.new
-        s[:server_id] = 'default-distro-12345678'
-        s
+    context "when region is set" do
+      let(:config) { { region: "test_region" } }
+
+      it "raises an exception if the region is not valid" do
+        expect(driver).to receive(:valid_region?).and_return(false)
+        expect { driver.validate! }.to raise_error(RuntimeError, "Region test_region is not a valid region")
       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")
+    end
+
+    it "raises an exception if the machine_type is invalid" do
+      expect(driver).to receive(:valid_machine_type?).and_return(false)
+      expect { driver.validate! }.to raise_error(RuntimeError, "Machine type test_machine_type is not valid")
+    end
 
-      it 'returns if server_id already exists' do
-        expect(driver.create(state)).to equal nil
+    it "raises an exception if the disk_type is invalid" do
+      expect(driver).to receive(:valid_disk_type?).and_return(false)
+      expect { driver.validate! }.to raise_error(RuntimeError, "Disk type test_disk_type is not valid")
+    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")
+    end
+
+    it "raises an exception if WinRM transport is used but no email is set" do
+      expect(driver).to receive(:winrm_transport?).and_return(true)
+      expect { driver.validate! }.to raise_error(RuntimeError, "Email address of GCE user is not set")
+    end
+  end
+
+  describe '#connection' do
+    it "returns a properly configured ComputeService" do
+      compute_service = double("compute_service")
+      client_options  = double("client_options")
+
+      expect(Google::Apis::ClientOptions).to receive(:new).and_return(client_options)
+      expect(client_options).to receive(:application_name=).with("kitchen-google")
+      expect(client_options).to receive(:application_version=).with(Kitchen::Driver::GCE_VERSION)
+
+      expect(Google::Apis::ComputeV1::ComputeService).to receive(:new).and_return(compute_service)
+      expect(driver).to receive(:authorization).and_return("authorization_object")
+      expect(compute_service).to receive(:authorization=).with("authorization_object")
+      expect(compute_service).to receive(:client_options=).with(client_options)
+
+      expect(driver.connection).to eq(compute_service)
+    end
+  end
+
+  describe '#authorization' do
+    it "returns a Google::Auth authorization object" do
+      auth_object = double("auth_object")
+      expect(Google::Auth).to receive(:get_application_default).and_return(auth_object)
+      expect(driver.authorization).to eq(auth_object)
+    end
+  end
+
+  describe '#winrm_transport?' do
+    it "returns true if the transport name is Winrm" do
+      expect(transport).to receive(:name).and_return("Winrm")
+      expect(driver.winrm_transport?).to eq(true)
+    end
+
+    it "returns false if the transport name is not Winrm" do
+      expect(transport).to receive(:name).and_return("Ssh")
+      expect(driver.winrm_transport?).to eq(false)
+    end
+  end
+
+  describe '#update_windows_password' do
+    it "does not attempt to reset the password if the transport is not WinRM" do
+      expect(driver).to receive(:winrm_transport?).and_return(false)
+      expect(GoogleComputeWindowsPassword).not_to receive(:new)
+
+      driver.update_windows_password("server_1")
+    end
+
+    it "resets the password and puts it in the state object if the transport is WinRM" do
+      state          = {}
+      winpass        = double("winpass")
+      winpass_config = {
+        project:       "test_project",
+        zone:          "test_zone",
+        instance_name: "server_1",
+        email:         "test_email",
+        username:      "test_username",
+      }
+
+      allow(driver).to receive(:state).and_return(state)
+      expect(transport).to receive(:config).and_return(username: "test_username")
+      expect(driver).to receive(:config).and_return(email: "test_email")
+      expect(driver).to receive(:winrm_transport?).and_return(true)
+      expect(GoogleComputeWindowsPassword).to receive(:new).with(winpass_config).and_return(winpass)
+      expect(winpass).to receive(:new_password).and_return("password123")
+      driver.update_windows_password("server_1")
+      expect(state[:password]).to eq("password123")
+    end
+  end
+
+  describe '#check_api_call' do
+    it "returns false and logs a debug message if the block raises a ClientError" do
+      expect(driver).to receive(:debug).with("API error: whoops")
+      expect(driver.check_api_call { raise Google::Apis::ClientError.new("whoops") }).to eq(false)
+    end
+
+    it "raises an exception if the block raises something other than a ClientError" do
+      expect { driver.check_api_call { raise RuntimeError.new("whoops") } }.to raise_error(RuntimeError)
+    end
+
+    it "returns true if the block does not raise an exception" do
+      expect(driver.check_api_call { true }).to eq(true)
+    end
+  end
+
+  describe '#valid_machine_type?' do
+    subject { driver.valid_machine_type? }
+    it_behaves_like "a validity checker", :machine_type, :get_machine_type, "test_project", "test_zone"
+  end
+
+  describe '#valid_network?' do
+    subject { driver.valid_network? }
+    it_behaves_like "a validity checker", :network, :get_network, "test_project"
+  end
+
+  describe '#valid_zone?' do
+    subject { driver.valid_zone? }
+    it_behaves_like "a validity checker", :zone, :get_zone, "test_project"
+  end
+
+  describe '#valid_region?' do
+    subject { driver.valid_region? }
+    it_behaves_like "a validity checker", :region, :get_region, "test_project"
+  end
+
+  describe '#valid_disk_type?' do
+    subject { driver.valid_disk_type? }
+    it_behaves_like "a validity checker", :disk_type, :get_disk_type, "test_project", "test_zone"
+  end
+
+  describe '#image_exist?' do
+    it "checks the outcome of the API call" do
+      connection = double("connection")
+      expect(driver).to receive(:connection).and_return(connection)
+      expect(connection).to receive(:get_image).with("image_project", "image_name")
+      expect(driver).to receive(:check_api_call).and_call_original
+      expect(driver.image_exist?("image_project", "image_name")).to eq(true)
+    end
+  end
+
+  describe '#server_exist?' do
+    it "checks the outcome of the API call" do
+      expect(driver).to receive(:server_instance).with("server_1")
+      expect(driver).to receive(:check_api_call).and_call_original
+      expect(driver.server_exist?("server_1")).to eq(true)
+    end
+  end
+
+  describe '#project' do
+    it "returns the project from the config" do
+      allow(driver).to receive(:project).and_call_original
+      expect(driver).to receive(:config).and_return(project: "my_project")
+      expect(driver.project).to eq("my_project")
+    end
+  end
+
+  describe '#region' do
+    it "returns the region from the config" do
+      expect(driver).to receive(:config).and_return(region: "my_region")
+      expect(driver.region).to eq("my_region")
+    end
+  end
+
+  describe '#zone' do
+    before do
+      allow(driver).to receive(:zone).and_call_original
+    end
+
+    context "when a zone exists in the state" do
+      let(:state) { { zone: "state_zone" } }
+
+      it "returns the zone from the state" do
+        expect(driver).to receive(:state).and_return(state)
+        expect(driver.zone).to eq("state_zone")
       end
     end
 
-    context 'when an instance is successfully created' do
+    context "when a zone does not exist in the state" do
+      let(:state) { {} }
 
-      let(:driver) do
-        d = Kitchen::Driver::Gce.new(config)
-        allow(d).to receive(:create_instance) { server }
-        allow(d).to receive(:wait_for_up_instance) { nil }
-        d
+      before do
+        allow(driver).to receive(:state).and_return(state)
       end
 
-      it 'sets a value for server_id in the state hash' do
-        driver.send(:create, state)
-        expect(state[:server_id]).to eq('rspec-test-instance')
+      it "returns the zone from the config if it exists" do
+        expect(driver).to receive(:config).and_return(zone: "config_zone")
+        expect(driver.zone).to eq("config_zone")
       end
 
-      it 'returns nil' do
-        expect(driver.send(:create, state)).to equal(nil)
+      it "returns the zone from find_zone if it does not exist in the config" do
+        expect(driver).to receive(:config).and_return({})
+        expect(driver).to receive(:find_zone).and_return("found_zone")
+        expect(driver.zone).to eq("found_zone")
       end
     end
   end
 
-  describe '#create_disk' do
-    context 'with defaults and required options' do
-      it 'returns a Google Disk object' do
-        config[:image_name] = 'debian-7-wheezy-v20130816'
-        config[:inst_name] = 'rspec-disk'
-        config[:zone_name] = 'us-central1-a'
-        expect(driver.send(:create_disk)).to be_a(Fog::Compute::Google::Disk)
-      end
+  describe '#find_zone' do
+    let(:zones_in_region) { double("zones_in_region") }
+
+    before do
+      expect(driver).to receive(:zones_in_region).and_return(zones_in_region)
     end
 
-    context 'without required options' do
-      it 'returns a Fog NotFound Error' do
-        expect { driver.send(:create_disk) }.to raise_error(
-          Fog::Errors::NotFound)
-      end
+    it "returns a random zone from the list of zones in the region" do
+      zone = double("zone", name: "random_zone")
+      expect(zones_in_region).to receive(:sample).and_return(zone)
+      expect(driver.find_zone).to eq("random_zone")
+    end
+
+    it "raises an exception if no zones are found" do
+      expect(zones_in_region).to receive(:sample).and_return(nil)
+      expect(driver).to receive(:region).and_return("test_region")
+      expect { driver.find_zone }.to raise_error(RuntimeError, "Unable to find a suitable zone in test_region")
     end
   end
 
-  describe '#create_instance' do
-    context 'with default options' do
-      it 'returns a Fog Compute Server object' do
-        expect(driver.send(:create_instance)).to be_a(
-          Fog::Compute::Google::Server)
-      end
+  describe '#zones_in_region' do
+    it "returns a correct list of available zones" do
+      zone1      = double("zone1", status: "UP", region: "a/b/c/test_region")
+      zone2      = double("zone2", status: "UP", region: "a/b/c/test_region")
+      zone3      = double("zone3", status: "DOWN", region: "a/b/c/test_region")
+      zone4      = double("zone4", status: "UP", region: "a/b/c/wrong_region")
+      zone5      = double("zone5", status: "UP", region: "a/b/c/test_region")
+      connection = double("connection")
+      response   = double("response", items: [zone1, zone2, zone3, zone4, zone5])
+
+      allow(driver).to receive(:region).and_return("test_region")
+      expect(driver).to receive(:connection).and_return(connection)
+      expect(connection).to receive(:list_zones).and_return(response)
+      expect(driver.zones_in_region).to eq([zone1, zone2, zone5])
+    end
+  end
 
-      it 'sets the region to the default "us-central1"' do
-        driver.send(:create_instance)
-        expect(config[:region]).to eq('us-central1')
-      end
+  describe '#server_instance' do
+    it "returns the instance from the API" do
+      connection = double("connection")
+      expect(driver).to receive(:connection).and_return(connection)
+      expect(connection).to receive(:get_instance).with("test_project", "test_zone", "server_1").and_return("instance")
+      expect(driver.server_instance("server_1")).to eq("instance")
     end
+  end
 
-    context 'area set, region unset' do
-      let(:config) do
-        { area: 'europe-west1',
-          google_client_email: '123456789012@developer.gserviceaccount.com',
-          google_key_location: '/home/user/gce/123456-privatekey.p12',
-          google_project: 'alpha-bravo-123'
-        }
-      end
+  describe '#ip_address_for' do
+    it "returns the private IP if use_private_ip is true" do
+      expect(driver).to receive(:config).and_return(use_private_ip: true)
+      expect(driver).to receive(:private_ip_for).with("server").and_return("1.2.3.4")
+      expect(driver.ip_address_for("server")).to eq("1.2.3.4")
+    end
 
-      it 'sets region to the area value' do
-        driver.send(:create_instance)
-        expect(config[:region]).to eq(config[:area])
-      end
+    it "returns the public IP if use_private_ip is false" do
+      expect(driver).to receive(:config).and_return(use_private_ip: false)
+      expect(driver).to receive(:public_ip_for).with("server").and_return("4.3.2.1")
+      expect(driver.ip_address_for("server")).to eq("4.3.2.1")
     end
+  end
 
-    context 'area set, region set' do
-      let(:config) do
-        { area: 'fugazi',
-          google_client_email: '123456789012@developer.gserviceaccount.com',
-          google_key_location: '/home/user/gce/123456-privatekey.p12',
-          google_project: 'alpha-bravo-123',
-          region: 'europe-west1'
-        }
-      end
+  describe '#private_ip_for' do
+    it "returns the IP address if it exists" do
+      network_interface = double("network_interface", network_ip: "1.2.3.4")
+      server            = double("server", network_interfaces: [network_interface])
 
-      it 'sets the region independent of the area value' do
-        driver.send(:create_instance)
-        expect(config[:region]).to eq('europe-west1')
-      end
+      expect(driver.private_ip_for(server)).to eq("1.2.3.4")
+    end
 
+    it "raises an exception if the IP cannot be found" do
+      server = double("server")
+
+      expect(server).to receive(:network_interfaces).and_raise(NoMethodError)
+      expect { driver.private_ip_for(server) }.to raise_error(RuntimeError, "Unable to determine private IP for instance")
     end
   end
 
-  describe '#create_server' do
-    context 'with default options' do
-      it 'returns a Fog Compute Server object' do
-        expect(driver.send(:create_instance)).to be_a(
-          Fog::Compute::Google::Server)
-      end
+  describe '#public_ip_for' do
+    it "returns the IP address if it exists" do
+      access_config     = double("access_config", nat_ip: "4.3.2.1")
+      network_interface = double("network_interface", access_configs: [access_config])
+      server            = double("server", network_interfaces: [network_interface])
+
+      expect(driver.public_ip_for(server)).to eq("4.3.2.1")
+    end
+
+    it "raises an exception if the IP cannot be found" do
+      network_interface = double("network_interface")
+      server            = double("server", network_interfaces: [network_interface])
+
+      expect(network_interface).to receive(:access_configs).and_raise(NoMethodError)
+      expect { driver.public_ip_for(server) }.to raise_error(RuntimeError, "Unable to determine public IP for instance")
     end
   end
 
-  describe '#destroy' do
-    let(:state) do
-      s = Hash.new
-      s[:server_id] = 'rspec-test-instance'
-      s[:hostname] = '198.51.100.17'
-      s
+  describe '#generate_server_name' do
+    it "generates and returns a server name" do
+      expect(instance).to receive(:name).and_return("ABC123")
+      expect(SecureRandom).to receive(:hex).with(3).and_return("abcdef")
+      expect(driver.generate_server_name).to eq("tk-abc123-abcdef")
     end
 
-    it 'returns if server_id does not exist' do
-      expect(driver.destroy({})).to equal nil
+    it "uses a UUID-based server name if the instance name is too long" do
+      expect(instance).to receive(:name).twice.and_return("123456789012345678901234567890123456789012345678901235467890")
+      expect(driver).to receive(:warn)
+      expect(SecureRandom).to receive(:hex).with(3).and_return("abcdef")
+      expect(SecureRandom).to receive(:uuid).and_return("lmnop")
+      expect(driver.generate_server_name).to eq("tk-lmnop")
     end
+  end
 
-    it 'removes the server state information' do
-      driver.destroy(state)
-      expect(state[:hostname]).to equal(nil)
-      expect(state[:server_id]).to equal(nil)
+  describe '#boot_disk' do
+    it "sets up a disk object and returns it" do
+      disk    = double("disk")
+      params  = double("params")
+
+      config  = {
+        autodelete_disk: "auto_delete",
+        disk_size: "test_size",
+        disk_type: "test_type",
+      }
+
+      allow(driver).to receive(:config).and_return(config)
+      expect(driver).to receive(:disk_type_url_for).with("test_type").and_return("disk_url")
+      expect(driver).to receive(:disk_image_url).and_return("disk_image_url")
+
+      expect(Google::Apis::ComputeV1::AttachedDisk).to receive(:new).and_return(disk)
+      expect(Google::Apis::ComputeV1::AttachedDiskInitializeParams).to receive(:new).and_return(params)
+      expect(disk).to receive(:boot=).with(true)
+      expect(disk).to receive(:auto_delete=).with("auto_delete")
+      expect(disk).to receive(:initialize_params=).with(params)
+      expect(params).to receive(:disk_name=).with("server_1")
+      expect(params).to receive(:disk_size_gb=).with("test_size")
+      expect(params).to receive(:disk_type=).with("disk_url")
+      expect(params).to receive(:source_image=).with("disk_image_url")
+
+      expect(driver.boot_disk("server_1")).to eq(disk)
     end
   end
 
-  describe '#generate_inst_name' do
-    context 'with a name less than 28 characters' do
-      it 'concatenates the name and a UUID' do
-        expect(driver.send(:generate_inst_name)).to match(
-          /^default-distro-12-[a-f0-9]{8}-([a-f0-9]{4}-){3}[a-f0-9]{12}$/)
-      end
+  describe '#disk_type_url_for' do
+    it "returns a disk URL" do
+      expect(driver.disk_type_url_for("my_type")).to eq("zones/test_zone/diskTypes/my_type")
     end
+  end
 
-    context 'with a name 27 characters or longer' do
-      let(:instance) do
-        double(name: 'a23456789012345678901234567')
-      end
+  describe '#disk_image_url' do
+    before do
+      allow(driver).to receive(:config).and_return(config)
+    end
 
-      it 'shortens the base name and appends a UUID' do
-        expect(driver.send(:generate_inst_name).length).to eq 63
-        expect(driver.send(:generate_inst_name)).to match(
-          /^a2345678901234567890123456
-            -[a-f0-9]{8}-([a-f0-9]{4}-){3}[a-f0-9]{12}$/x)
+    context "when the user supplies an image project" do
+      let(:config) { { image_project: "my_image_project", image_name: "my_image" } }
+
+      it "returns the image URL based on the image project" do
+        expect(driver).to receive(:image_url_for).with("my_image_project", "my_image").and_return("image_url")
+        expect(driver.disk_image_url).to eq("image_url")
       end
     end
 
-    context 'with a "name" value containing an invalid leading character' do
-      let(:instance) do
-        double(name: '12345')
+    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")
+        end
       end
 
-      it 'adds a leading "t"' do
-        expect(driver.send(:generate_inst_name)).to match(
-          /^t12345
-            -[a-f0-9]{8}-([a-f0-9]{4}-){3}[a-f0-9]{12}$/x)
+      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
+  end
 
-    context 'with a "name" value containing uppercase letters' do
-      let(:instance) do
-        double(name: 'AbCdEf')
-      end
+  describe '#image_url_for' do
+    it "returns nil if the image does not exist" do
+      expect(driver).to receive(:image_exist?).with("image_project", "image_name").and_return(false)
+      expect(driver.image_url_for("image_project", "image_name")).to eq(nil)
+    end
 
-      it 'downcases the "name" characters in the instance name' do
-        expect(driver.send(:generate_inst_name)).to match(
-          /^abcdef
-            -[a-f0-9]{8}-([a-f0-9]{4}-){3}[a-f0-9]{12}$/x)
-      end
+    it "returns a properly formatted image URL if the image exists" do
+      expect(driver).to receive(:image_exist?).with("image_project", "image_name").and_return(true)
+      expect(driver.image_url_for("image_project", "image_name")).to eq("projects/image_project/global/images/image_name")
     end
+  end
 
-    context 'with a name value containing invalid characters' do
-      let(:instance) do
-        double(name: 'a!b@c#d$e%f^g&h*i(j)')
+  describe '#public_project_for_image' do
+    {
+      "centos"         => "centos-cloud",
+      "container-vm"   => "google-containers",
+      "coreos"         => "coreos-cloud",
+      "debian"         => "debian-cloud",
+      "opensuse-cloud" => "opensuse-cloud",
+      "rhel"           => "rhel-cloud",
+      "sles"           => "suse-cloud",
+      "ubuntu"         => "ubuntu-os-cloud",
+      "windows"        => "windows-cloud",
+    }.each do |image_name, project_name|
+      it "returns project #{project_name} for an image named #{image_name}" do
+        expect(driver.public_project_for_image(image_name)).to eq(project_name)
       end
+    end
+  end
 
-      it 'replaces the invalid characters with dashes' do
-        expect(driver.send(:generate_inst_name)).to match(
-          /^a-b-c-d-e-f-g-h-i-j-
-            -[a-f0-9]{8}-([a-f0-9]{4}-){3}[a-f0-9]{12}$/x)
-      end
+  describe '#machine_type_url' do
+    it "returns a machine type URL" do
+      expect(driver).to receive(:config).and_return(machine_type: "machine_type")
+      expect(driver.machine_type_url).to eq("zones/test_zone/machineTypes/machine_type")
     end
   end
 
-  describe '#select_zone' do
-    context 'when choosing from any region' do
-      let(:config) do
-        { region: 'any',
-          google_client_email: '123456789012@developer.gserviceaccount.com',
-          google_key_location: '/home/user/gce/123456-privatekey.p12',
-          google_project: 'alpha-bravo-123'
-        }
-      end
+  describe '#instance_metadata' do
+    it "returns a properly-configured metadata object" do
+      item1    = double("item1")
+      item2    = double("item2")
+      item3    = double("item3")
+      metadata = double("metadata")
+
+      expect(instance).to receive(:name).and_return("instance_name")
+      expect(driver).to receive(:env_user).and_return("env_user")
+      expect(Google::Apis::ComputeV1::Metadata).to receive(:new).and_return(metadata)
+      expect(Google::Apis::ComputeV1::Metadata::Item).to receive(:new).and_return(item1)
+      expect(Google::Apis::ComputeV1::Metadata::Item).to receive(:new).and_return(item2)
+      expect(Google::Apis::ComputeV1::Metadata::Item).to receive(:new).and_return(item3)
+      expect(item1).to receive(:key=).with("created-by")
+      expect(item1).to receive(:value=).with("test-kitchen")
+      expect(item2).to receive(:key=).with("test-kitchen-instance")
+      expect(item2).to receive(:value=).with("instance_name")
+      expect(item3).to receive(:key=).with("test-kitchen-user")
+      expect(item3).to receive(:value=).with("env_user")
+      expect(metadata).to receive(:items=).with([item1, item2, item3])
+
+      expect(driver.instance_metadata).to eq(metadata)
+    end
+  end
 
-      it 'chooses from all zones' do
-        expect(driver.send(:select_zone)).to satisfy do |zone|
-          %w(europe-west1-a us-central1-a us-central1-b
-             us-central2-a).include?(zone)
-        end
+  describe '#env_user' do
+    it "returns the current user from the environment" do
+      expect(ENV).to receive(:[]).with("USER").and_return("test_user")
+      expect(driver.env_user).to eq("test_user")
+    end
+
+    it "returns 'unknown' if there is no USER present" do
+      expect(ENV).to receive(:[]).with("USER").and_return(nil)
+      expect(driver.env_user).to eq("unknown")
+    end
+  end
+
+  describe '#instance_network_interfaces' do
+    it "returns an array containing a properly-formatted interface" do
+      interface = double("interface")
+
+      expect(driver).to receive(:network_url).and_return("network_url")
+      expect(driver).to receive(:interface_access_configs).and_return("access_configs")
+
+      expect(Google::Apis::ComputeV1::NetworkInterface).to receive(:new).and_return(interface)
+      expect(interface).to receive(:network=).with("network_url")
+      expect(interface).to receive(:access_configs=).with("access_configs")
+
+      expect(driver.instance_network_interfaces).to eq([interface])
+    end
+  end
+
+  describe '#network_url' do
+    it "returns a network URL" do
+      expect(driver).to receive(:config).and_return(network: "test_network")
+      expect(driver.network_url).to eq("projects/test_project/global/networks/test_network")
+    end
+  end
+
+  describe '#interface_access_configs' do
+    it "returns a properly-configured access config object" do
+      access_config = double("access_config")
+
+      expect(driver).to receive(:config).and_return({})
+      expect(Google::Apis::ComputeV1::AccessConfig).to receive(:new).and_return(access_config)
+      expect(access_config).to receive(:name=).with("External NAT")
+      expect(access_config).to receive(:type=).with("ONE_TO_ONE_NAT")
+
+      expect(driver.interface_access_configs).to eq([access_config])
+    end
+
+    it "returns an empty array if use_private_ip is true" do
+      expect(driver).to receive(:config).and_return(use_private_ip: true)
+      expect(driver.interface_access_configs).to eq([])
+    end
+  end
+
+  describe '#instance_scheduling' 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(: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(:on_host_maintenance=).with("host_maintenance")
+      expect(driver.instance_scheduling).to eq(scheduling)
+    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.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.migrate_setting).to eq("TERMINATE")
+    end
+  end
+
+  describe '#instance_service_accounts' do
+    it "returns nil if service_account_scopes is nil" do
+      allow(driver).to receive(:config).and_return({})
+      expect(driver.instance_service_accounts).to eq(nil)
+    end
+
+    it "returns nil if service_account_scopes is empty" do
+      allow(driver).to receive(:config).and_return(service_account_scopes: [])
+      expect(driver.instance_service_accounts).to eq(nil)
+    end
+
+    it "returns an array containing a properly-formatted service account" do
+      service_account = double("service_account")
+
+      allow(driver).to receive(:config).and_return(service_account_name: "account_name", service_account_scopes: %w{scope1 scope2})
+      expect(Google::Apis::ComputeV1::ServiceAccount).to receive(:new).and_return(service_account)
+      expect(service_account).to receive(:email=).with("account_name")
+      expect(driver).to receive(:service_account_scope_url).with("scope1").and_return("https://www.googleapis.com/auth/scope1")
+      expect(driver).to receive(:service_account_scope_url).with("scope2").and_return("https://www.googleapis.com/auth/scope2")
+      expect(service_account).to receive(:scopes=).with([
+        "https://www.googleapis.com/auth/scope1",
+        "https://www.googleapis.com/auth/scope2",
+      ])
+
+      expect(driver.instance_service_accounts).to eq([service_account])
+    end
+  end
+
+  describe '#service_account_scope_url' do
+    it "returns the passed-in scope if it already looks like a scope URL" do
+      scope = "https://www.googleapis.com/auth/fake_scope"
+      expect(driver.service_account_scope_url(scope)).to eq(scope)
+    end
+
+    it "returns a properly-formatted scope URL if a short-name or alias is provided" do
+      expect(driver).to receive(:translate_scope_alias).with("scope_alias").and_return("real_scope")
+      expect(driver.service_account_scope_url("scope_alias")).to eq("https://www.googleapis.com/auth/real_scope")
+    end
+  end
+
+  describe '#translate_scope_alias' do
+    it "returns a scope for a given alias" do
+      expect(driver.translate_scope_alias("storage-rw")).to eq("devstorage.read_write")
+    end
+
+    it "returns the passed-in scope alias if nothing matches in the alias map" do
+      expect(driver.translate_scope_alias("fake_scope")).to eq("fake_scope")
+    end
+  end
+
+  describe '#instance_tags' do
+    it "returns a properly-formatted tags object" do
+      tags_obj = double("tags_obj")
+
+      expect(driver).to receive(:config).and_return(tags: "test_tags")
+      expect(Google::Apis::ComputeV1::Tags).to receive(:new).and_return(tags_obj)
+      expect(tags_obj).to receive(:items=).with("test_tags")
+
+      expect(driver.instance_tags).to eq(tags_obj)
+    end
+  end
+
+  describe '#wait_time' do
+    it "returns the configured wait time" do
+      expect(driver).to receive(:config).and_return(wait_time: 123)
+      expect(driver.wait_time).to eq(123)
+    end
+  end
+
+  describe '#refresh_rate' do
+    it "returns the configured refresh rate" do
+      expect(driver).to receive(:config).and_return(refresh_rate: 321)
+      expect(driver.refresh_rate).to eq(321)
+    end
+  end
+
+  describe '#wait_for_status' do
+    let(:item) { double("item") }
+
+    before do
+      allow(driver).to receive(:wait_time).and_return(600)
+      allow(driver).to receive(:refresh_rate).and_return(2)
+
+      # don"t actually sleep
+      allow(driver).to receive(:sleep)
+    end
+
+    context "when the items completes normally, 3 loops" do
+      it "only refreshes the item 3 times" do
+        allow(item).to receive(:status).exactly(3).times.and_return("PENDING", "RUNNING", "DONE")
+
+        driver.wait_for_status("DONE") { item }
       end
     end
 
-    context 'when choosing from the "europe-west1" region' do
-      let(:config) do
-        { region: 'europe-west1',
-          google_client_email: '123456789012@developer.gserviceaccount.com',
-          google_key_location: '/home/user/gce/123456-privatekey.p12',
-          google_project: 'alpha-bravo-123'
-        }
+    context "when the item is completed on the first loop" do
+      it "only refreshes the item 1 time" do
+        allow(item).to receive(:status).once.and_return("DONE")
+
+        driver.wait_for_status("DONE") { item }
       end
+    end
 
-      it 'chooses a zone in europe-west1' do
-        expect(driver.send(:select_zone)).to satisfy do |zone|
-          %w(europe-west1-a).include?(zone)
-        end
+    context "when the timeout is exceeded" do
+      it "prints a warning and exits" do
+        allow(Timeout).to receive(:timeout).and_raise(Timeout::Error)
+        expect(driver).to receive(:error)
+          .with("Request did not complete in 600 seconds. Check the Google Cloud Console for more info.")
+        expect { driver.wait_for_status("DONE") { item } }.to raise_error(RuntimeError)
       end
     end
 
-    context 'when choosing from the default "us-central1" region' do
-      let(:config) do
-        { region: 'us-central1',
-          google_client_email: '123456789012@developer.gserviceaccount.com',
-          google_key_location: '/home/user/gce/123456-privatekey.p12',
-          google_project: 'alpha-bravo-123'
-        }
+    context "when a non-timeout exception is raised" do
+      it "raises the original exception" do
+        allow(item).to receive(:status).and_raise(NoMethodError)
+        expect { driver.wait_for_status("DONE") { item } }.to raise_error(NoMethodError)
       end
+    end
+  end
 
-      it 'chooses a zone in us-central1' do
-        expect(driver.send(:select_zone)).to satisfy do |zone|
-          %w(us-central1-a us-central1-b us-central2-a).include?(zone)
-        end
+  describe '#wait_for_operation' do
+    let(:operation) { double("operation", name: "operation-123") }
 
-      end
+    it "raises a properly-formatted exception when errors exist" do
+      error1 = double("error1", code: "ERROR1", message: "error 1")
+      error2 = double("error2", code: "ERROR2", message: "error 2")
+      expect(driver).to receive(:wait_for_status).with("DONE")
+      expect(driver).to receive(:operation_errors).with("operation-123").and_return([error1, error2])
+      expect(driver).to receive(:error).with("ERROR1: error 1")
+      expect(driver).to receive(:error).with("ERROR2: error 2")
+
+      expect { driver.wait_for_operation(operation) }.to raise_error(RuntimeError, "Operation operation-123 failed.")
+    end
+
+    it "does not raise an exception if no errors are encountered" do
+      expect(driver).to receive(:wait_for_status).with("DONE")
+      expect(driver).to receive(:operation_errors).with("operation-123").and_return([])
+      expect(driver).not_to receive(:error)
+
+      expect { driver.wait_for_operation(operation) }.not_to raise_error
     end
   end
 
-  describe '#wait_for_up_instance' do
-    it 'sets the hostname' do
-      driver.send(:wait_for_up_instance, server, state)
-      # Mock instance gives us a random IP each time:
-      expect(state[:hostname]).to match(Resolv::IPv4::Regex)
+  describe '#zone_operation' do
+    it "fetches the operation from the API and returns it" do
+      connection = double("connection")
+      expect(driver).to receive(:connection).and_return(connection)
+      expect(connection).to receive(:get_zone_operation).with(project, zone, "operation-123").and_return("operation")
+      expect(driver.zone_operation("operation-123")).to eq("operation")
+    end
+  end
+
+  describe '#operation_errors' do
+    let(:operation) { double("operation") }
+    let(:error_obj) { double("error_obj") }
+
+    before do
+      expect(driver).to receive(:zone_operation).with("operation-123").and_return(operation)
+    end
+
+    it "returns an empty array if there are no errors" do
+      expect(operation).to receive(:error).and_return(nil)
+      expect(driver.operation_errors("operation-123")).to eq([])
+    end
+
+    it "returns the errors from the operation if they exist" do
+      expect(operation).to receive(:error).twice.and_return(error_obj)
+      expect(error_obj).to receive(:errors).and_return("some errors")
+      expect(driver.operation_errors("operation-123")).to eq("some errors")
     end
   end
 end