knife-google 0.0.1 → 1.0.0

data/lib/google/compute/server/serial_port_output.rb

@@ -0,0 +1,31 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+
+require 'google/compute/mixins/utils'
+
+module Google
+  module Compute
+    class SerialPortOutput
+      include Utils
+
+      attr_reader :kind, :self_link, :contents
+     
+      def initialize(data)
+        @kind = data["kind"]
+        @contents = data["contents"]
+        @self_link = data["selfLink"]
+      end
+    end
+  end
+end

data/lib/google/compute/snapshot.rb

@@ -0,0 +1,30 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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 Google
+  module Compute
+    class Snapshot < Resource
+
+      attr_reader :disk_size_gb, :status, :source_disk, :source_disk_id
+
+      def from_hash(data)
+        super(data)
+        @disk_size_gb = data["diskSizeGb"]
+        @status = data["status"]
+        @source_disk = data["sourceDisk"]
+        @source_disk_id = data["sourceDiskId"]
+      end
+    end
+  end
+end

data/lib/google/compute/version.rb

@@ -0,0 +1,19 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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 Google
+  module Compute
+    VERSION = '1.0.0'
+  end
+end

data/lib/google/compute/zone.rb

@@ -0,0 +1,32 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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 Google
+  module Compute
+    class Zone < Resource
+
+      attr_reader :maintenance_windows, :available_machine_type, :status
+      attr_reader :quotas, :deprecated
+
+      def from_hash(zone_data)
+        super(zone_data)
+        @status = zone_data["status"] 
+        @maintenance_windows = zone_data["maintenanceWindows"]
+        @available_machine_type = zone_data["availableMachineType"]
+        @quotas = zone_data["quotas"]
+        @deprecated = zone_data["deprecated"]
+      end
+    end
+  end
+end

data/lib/google/compute/zone_operation.rb

@@ -0,0 +1,60 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+
+# Google compute engine, operation resource reference
+
+require 'timeout'
+
+module Google
+  module Compute
+    class ZoneOperation < Resource
+
+      attr_reader :target_link, :target_id, :client_operation_id
+      attr_reader :status, :status_message, :user, :progress
+      attr_reader :insert_time, :start_time, :end_time
+      attr_reader :http_error_status_code, :http_error_message
+      attr_reader :error, :warnings, :operation_type, :zone
+
+      def from_hash(data)
+        super(data)
+        @target_link= data["targetLink"]
+        @target_id = data["targetId"]
+        @client_operation_id =  data["clientOperationId"]
+        @status =  data["status"]
+        @status_message = data["statusMessage"]
+        @user = data["user"]
+        @progress =  data["progress"]
+        @insert_time = Time.parse( data["insertTime"] )
+        @start_time = Time.parse( data["startTime"] )
+        @end_time = Time.parse( data["endTime"] ) if data.key?("endTime")
+        @http_error_status_code = data["httpErrorMessage"]
+        @http_error_message = data["httpErrorMessage"]
+        @error = data["error"]
+        @warnings = data["warnings"]
+        @operation_type = data["operationType"]
+        @zone = data["zone"]
+      end
+
+      def wait_for_completion!(options={})
+        timeout = options[:timeout] || 60
+        status = Timeout::timeout(timeout, OperationTimeout) do
+          until  progress==100
+            sleep 2
+            progress= get_self.progress
+          end  
+        end
+      end
+    end
+  end
+end

data/lib/knife-google/version.rb

@@ -1,3 +1,19 @@
-module KnifeGoogle
-  VERSION = "0.0.1"
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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 Knife
+  module Google
+    VERSION = "1.0.0"
+  end
 end

data/spec/chef/knife/google_base_spec.rb

@@ -0,0 +1,46 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+#
+
+require 'spec_helper'
+
+describe Chef::Knife::GoogleBase do
+  let(:knife_plugin) { Chef::Knife::GoogleServerList.new(["-Z"+stored_zone.name]) }
+
+  it "#client should return a Google::Compute::Client" do
+    Google::Compute::Client.should_receive(:from_json).
+      and_return(mock(Google::Compute::Client))
+    knife_plugin.client
+  end
+
+  it "#selflink2name should return name from a seleflink url" do
+    knife_plugin.selflink2name(
+      'https://www.googleapis.com/compute/v1beta14/projects/mock-project/category/resource').
+      should eq('resource')
+  end
+
+  it "#msg_pair should invoke ui.info with labe : value string" do
+    knife_plugin.ui.should_receive(:info).
+      with("#{knife_plugin.ui.color("label", :cyan)}: value")
+    knife_plugin.msg_pair("label","value")
+  end
+
+  it "#private_ips should extract private ip as an array from a GCE server" do
+    knife_plugin.private_ips(stored_instance).should eq(['10.100.0.10'])
+  end
+
+  it "#public_ips should extract private ip as an array from a GCE server" do
+    knife_plugin.public_ips(stored_instance).should eq(['11.1.1.11'])
+  end
+end

data/spec/chef/knife/google_disk_create_spec.rb

@@ -0,0 +1,36 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+#
+require 'spec_helper'
+
+describe Chef::Knife::GoogleDiskCreate do
+
+  let(:knife_plugin) do
+    Chef::Knife::GoogleDiskCreate.new(
+      [stored_disk.name, "-Z"+stored_zone.name, "-s14"])
+  end
+
+  it "#run should invoke compute api to create a disk" do
+    zones = mock(Google::Compute::ListableResourceCollection)
+    zones.should_receive(:get).
+      with(stored_zone.name).and_return(stored_zone)
+    disks = mock(Google::Compute::CreatableResourceCollection)
+    disks.should_receive(:create).
+      with(:zone=>stored_zone.name, :name=>stored_disk.name, :sizeGb=>"14").
+      and_return(stored_zone_operation)
+    client = mock(Google::Compute::Client, :zones=>zones, :disks=>disks)
+    Google::Compute::Client.stub!(:from_json).and_return(client)
+    knife_plugin.run
+  end
+end

data/spec/chef/knife/google_disk_delete_spec.rb

@@ -0,0 +1,65 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+
+require 'spec_helper'
+
+describe Chef::Knife::GoogleDiskDelete do
+
+  let(:knife_plugin) do
+    Chef::Knife::GoogleDiskDelete.new([stored_disk.name, "-Z"+stored_zone.name])
+  end
+
+  it "should print out error message if the disk is not found" do
+    zones = mock(Google::Compute::ListableResourceCollection)
+    zones.should_receive(:get).with(stored_zone.name).
+      and_return(stored_zone)
+    disks = mock(Google::Compute::DeletableResourceCollection)
+    disks.should_receive(:get).
+      with(:zone=>stored_zone.name, :disk=>stored_disk.name).
+      and_raise(Google::Compute::ResourceNotFound)
+    disks.should_not_receive(:delete)
+    client = mock(Google::Compute::Client,
+      :disks=>disks, :zones=>zones)
+    Google::Compute::Client.stub!(:from_json).and_return(client)
+
+    knife_plugin.config[:yes] = true
+    knife_plugin.ui.should_receive(:error).
+      with("Disk '#{stored_zone.name}:#{stored_disk.name}' not found")
+    knife_plugin.stub!(:msg_pair)
+    expect {
+      knife_plugin.run
+    }.to raise_error(SystemExit)
+  end
+
+  it "should invoke api delete method when run is called" do
+    zones = mock(Google::Compute::ListableResourceCollection)
+    zones.should_receive(:get).with(stored_zone.name).
+      and_return(stored_zone)
+    disks = mock(Google::Compute::DeletableResourceCollection)
+    disks.should_receive(:get).
+      with(:zone=>stored_zone.name, :disk=>stored_disk.name).
+      and_return(stored_disk)
+    disks.should_receive(:delete).
+      with(:zone=>stored_zone.name, :disk=>stored_disk.name)
+    client = mock(Google::Compute::Client,
+      :zones=>zones,:disks=>disks)
+    Google::Compute::Client.stub!(:from_json).
+      and_return(client)
+    knife_plugin.config[:yes] = true
+    knife_plugin.ui.should_receive(:warn).
+      with("Disk '#{stored_zone.name}:#{stored_disk.name}' deleted")
+    knife_plugin.stub!(:msg_pair)
+    knife_plugin.run
+  end
+end

data/spec/chef/knife/google_disk_list_spec.rb

@@ -0,0 +1,36 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+#
+require 'spec_helper'
+
+describe Chef::Knife::GoogleDiskList do
+
+  let(:knife_plugin) do
+    Chef::Knife::GoogleDiskList.new(["-Z"+stored_zone.name])
+  end
+
+  it "should enlist all the GCE disks when run invoked" do
+    zones = mock(Google::Compute::ListableResourceCollection)
+    zones.should_receive(:get).with(stored_zone.name).
+      and_return(stored_zone)
+    disks = mock(Google::Compute::ListableResourceCollection)
+    disks.should_receive(:list).with(:zone=>stored_zone.name).
+      and_return([stored_disk])
+
+    client = mock(Google::Compute::Client, :disks=>disks, :zones=>zones)
+    Google::Compute::Client.stub!(:from_json).and_return(client)
+    $stdout.should_receive(:write).with(kind_of(String))
+    knife_plugin.run
+  end
+end

data/spec/chef/knife/google_server_create_spec.rb

@@ -0,0 +1,84 @@
+# Copyright 2013 Google Inc. All Rights Reserved.
+#
+# 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.
+#
+require 'spec_helper'
+
+describe Chef::Knife::GoogleServerCreate do
+
+  let(:knife_plugin) do
+    Chef::Knife::GoogleServerCreate.new(["-m"+stored_machine_type.name,
+      "-I"+stored_image.name, "-n"+stored_network.name,
+      "-Z"+stored_zone.name, stored_instance.name])
+  end
+
+  it "#run should invoke compute api to create an server" do
+    zones = mock(Google::Compute::ListableResourceCollection)
+    zones.should_receive(:get).with(stored_zone.name).
+      and_return(stored_zone)
+
+    machine_types = mock(Google::Compute::ListableResourceCollection)
+    machine_types.should_receive(:get).with(stored_machine_type.name).
+      and_return(stored_machine_type)
+
+    images = mock(Google::Compute::ListableResourceCollection)
+    images.should_receive(:get).
+      with({:project=>"google", :name=>stored_image.name}).
+      and_return(stored_image)
+
+    networks = mock(Google::Compute::ListableResourceCollection)
+    networks.should_receive(:get).with(stored_network.name).
+      and_return(stored_network)
+
+    instances = mock(Google::Compute::ListableResourceCollection)
+    instances.should_receive(:create).with(
+      {:name=>stored_instance.name, :image=>stored_image.self_link,
+      :machineType=>stored_machine_type.self_link, :disks=>[],
+      :metadata=>{"items"=>[]}, :zone=>stored_zone.name,
+      :networkInterfaces=>[
+        {"network"=>stored_network.self_link,
+        "accessConfigs"=>[
+          {"name"=>"External NAT", "type"=>"ONE_TO_ONE_NAT"}]}],
+      :tags=>nil}).and_return(stored_zone_operation)
+
+    instances.should_receive(:get).
+      with(:zone=>stored_zone.name, :name=>stored_instance.name).
+      and_return(stored_instance)
+
+    client = mock(Google::Compute::Client, :instances=>instances,
+      :images=>images, :zones=>zones,:machine_types=>machine_types,
+      :networks=>networks)
+    Google::Compute::Client.stub!(:from_json).and_return(client)
+
+    knife_plugin.config[:disks]=[]
+    knife_plugin.config[:metadata]=[]
+    knife_plugin.config[:public_ip]='EPHEMERAL'
+    knife_plugin.ui.stub!(:info)
+
+    knife_plugin.stub!(:wait_for_sshd)
+    knife_plugin.should_receive(:bootstrap_for_node).
+      with(stored_instance,'10.100.0.10').
+      and_return(mock("Chef::Knife::Bootstrap",:run=>true))
+
+    knife_plugin.run
+  end
+
+  describe "without appropriate command line options" do
+    it "should throw exception when required params are not passed" do
+      $stdout.stub!(:write) # lets not print those error messages
+      expect {
+        Chef::Knife::GoogleServerCreate.new([ "NAME"])
+      }.to raise_error(SystemExit)
+    end
+  end
+end