kitchen-cloudstack 0.22.0 → 0.23.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
-SHA1:
-  metadata.gz: 865f50951ffac7b62e92d171f42fff075ad054f1
-  data.tar.gz: 0035d0cb62d978978cad8b4ed5627eaa37347015
+SHA256:
+  metadata.gz: 85e304a3e17949e1178d38b6a4bb8a79efee76cb9af48a0f9eeb6a942c416457
+  data.tar.gz: cf4363b3db44132a97cffbda84348cd553610206d3743434e34b6038033c9cc1
 SHA512:
-  metadata.gz: 3e0e41ca07d268bba030ffc40d62ecf91cbb71222498569ab0a87b6ce368a9acd3000e66b428e0506136207ae88d320695a7657b73d9279109b339fcc151b453
-  data.tar.gz: 4faa04395d32d79552ae715416e04cfdf8cbc33cf2283fe85ec282b3d25187f4ce2b6fd5c73b67a278c3fcec4bec54986d3671d69cc93f70f8b9731916445cf5
+  metadata.gz: d3100a2b43079e25074c90e61a2bf206a53267be6816ac0c0ddf93014640396a56c178bfdc956584e90e2082703df852f31403ea19a21efaec5a56ad05cc5360
+  data.tar.gz: 0d285e444bb04b5878c4d42e97d5d4fa5e12822c8a13c94da2c9f468d0cb224d2a9c4e374470a67a5b8de5dc013566e89362fde13c447b2e355bf2425d2cb2eb

data/README.md

@@ -27,6 +27,7 @@ Provide, at a minimum, the required driver options in your `.kitchen.yml` file:
       keypair_search_directory: [PATH TO DIRECTORY (other than ~, ., and ~/.ssh) WITH KEYPAIR PEM FILE]
       cloudstack_project_id: [PROJECT_ID] # To deploy VMs into project.
       cloudstack_vm_public_ip: [PUBLIC_IP] # In case you use advanced networking and do static NAT manually.
+      associate_public_ip: [TRUE/FALSE] # If you want kitchen to automatically associate a public IP, default false.
       cloudstack_userdata: "#cloud-config\npackages:\n - htop\n" # double quote required.
 
 Then to specify different OS templates,

data/lib/kitchen/driver/cloudstack.rb

@@ -146,7 +146,14 @@ module Kitchen
             info("Keypair specified but not found. Using password if enabled.")
           end
 
-          state[:hostname] = config[:cloudstack_vm_public_ip] || server_info.fetch('nic').first.fetch('ipaddress')
+          if config[:associate_public_ip]
+            info("Associating public ip...")
+            state[:hostname] = associate_public_ip(state, server_info)
+            info("Creating port forward...")
+            create_port_forward(state, server_info['id'])
+          else
+            state[:hostname] = default_public_ip(server_info) unless config[:associate_public_ip]
+          end
 
           if keypair
             debug("Using keypair: #{keypair}")
@@ -191,6 +198,10 @@ module Kitchen
 
       def destroy(state)
         return unless state[:server_id]
+        if config[:associate_public_ip]
+          delete_port_forward(state)
+          release_public_ip(state)
+        end
         debug("Destroying #{state[:server_id]}")
         server = compute.servers.get(state[:server_id])
         expunge =
@@ -302,6 +313,93 @@ module Kitchen
           Base64.encode64(user_data)
         end
       end
+
+      def associate_public_ip(state, server_info)
+        options = {
+          'zoneid' => config[:cloudstack_zone_id],
+          'vpcid' => get_vpc_id
+        }
+        res = compute.associate_ip_address(options)
+        job_status = compute.query_async_job_result(res['associateipaddressresponse']['jobid'])
+        if job_status['queryasyncjobresultresponse'].fetch('jobstatus').to_i == 1
+          save_ipaddress_id(state, job_status)
+          get_public_ip(res['associateipaddressresponse']['id'])
+        else
+          error(job_status['queryasyncjobresultresponse'].fetch('jobresult'))
+        end
+      end
+
+      def create_port_forward(state, virtualmachineid)
+        options = {
+          'ipaddressid' => state[:ipaddressid],
+          'privateport' => 22,
+          'protocol' => "TCP",
+          'publicport' => 22,
+          'virtualmachineid' => virtualmachineid,
+          'networkid' => config[:cloudstack_network_id],
+          'openfirewall' => false
+        }
+        res = compute.create_port_forwarding_rule(options)
+        job_status = compute.query_async_job_result(res['createportforwardingruleresponse']['jobid'])
+        unless job_status['queryasyncjobresultresponse'].fetch('jobstatus').to_i == 0
+          error("Error creating port forwarding rules")
+        end
+        save_forwarding_port_rule_id(state, res['createportforwardingruleresponse']['id'])
+      end
+
+      def release_public_ip(state)
+        info("Disassociating public ip...")
+        begin
+          res = compute.disassociate_ip_address(state[:ipaddressid])
+        rescue Fog::Compute::Cloudstack::BadRequest => e
+          error(e) unless e.to_s.match?(/does not exist/)
+        else
+          job_status = compute.query_async_job_result(res['disassociateipaddressresponse']['jobid'])
+          unless job_status['queryasyncjobresultresponse'].fetch('jobstatus').to_i == 0
+            error("Error disassociating public ip")
+          end
+        end
+      end
+
+      def delete_port_forward(state)
+        info("Deleting port forwarding rules...")
+        begin
+          res = compute.delete_port_forwarding_rule(state[:forwardingruleid])
+        rescue Fog::Compute::Cloudstack::BadRequest => e
+          error(e) unless e.to_s.match?(/does not exist/)
+        else
+          job_status = compute.query_async_job_result(res['deleteportforwardingruleresponse']['jobid'])
+          unless job_status['queryasyncjobresultresponse'].fetch('jobstatus').to_i == 0
+            error("Error deleting port forwarding rules")
+          end
+        end
+      end
+
+      def get_vpc_id
+        compute.list_networks['listnetworksresponse']['network']
+          .select{|e| e['id'] == config[:cloudstack_network_id]}.first['vpcid']
+      end
+
+      def get_public_ip(public_ip_uuid)
+        compute.list_public_ip_addresses['listpublicipaddressesresponse']['publicipaddress']
+          .select{|e| e['id'] == public_ip_uuid}
+          .first['ipaddress']
+      end
+
+      def save_ipaddress_id(state, job_status)
+        state[:ipaddressid] = job_status['queryasyncjobresultresponse']
+                                .fetch('jobresult')
+                                .fetch('ipaddress')
+                                .fetch('id')
+      end
+
+      def save_forwarding_port_rule_id(state, uuid)
+        state[:forwardingruleid] = uuid
+      end
+
+      def default_public_ip(server_info)
+        config[:cloudstack_vm_public_ip] || server_info.fetch('nic').first.fetch('ipaddress')
+      end
     end
   end
 end

data/lib/kitchen/driver/cloudstack_version.rb

@@ -21,6 +21,6 @@ module Kitchen
   module Driver
 
     # Version string for Cloudstack Kitchen driver
-    CLOUDSTACK_VERSION = "0.22.0"
+    CLOUDSTACK_VERSION = "0.23.0"
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: kitchen-cloudstack
 version: !ruby/object:Gem::Version
-  version: 0.22.0
+  version: 0.23.0
 platform: ruby
 authors:
 - Jeff Moody
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-01-11 00:00:00.000000000 Z
+date: 2018-08-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: test-kitchen
@@ -173,7 +173,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 2.5.1
+rubygems_version: 2.7.6
 signing_key: 
 specification_version: 4
 summary: Provides an interface for Test Kitchen to be able to run jobs against an