[vsphere] Supplied a mock implementation for cloudinit_to_customspec.

lautis · Feb 8, 2015 · f161203 · f161203
1 parent a05e7db
commit f161203
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/lib/fog/vsphere/requests/compute/cloudinit_to_customspec.rb b/lib/fog/vsphere/requests/compute/cloudinit_to_customspec.rb
@@ -3,7 +3,7 @@ module Compute
     class Vsphere
       class Real
         def cloudinit_to_customspec(user_data)
-          raise ArgumentError, "user_data cant be nil" if user_data.nil?
+          raise ArgumentError, "user_data can't be nil" if user_data.nil?
           custom_spec = { 'customization_spec' => Hash.new }
           user_data = YAML.load(user_data)
           custom_spec['hostname']                    =  user_data['hostname'] if user_data.key?('hostname')
@@ -19,8 +19,17 @@ def cloudinit_to_customspec(user_data)
 
       class Mock
         def cloudinit_to_customspec(user_data)
-          raise ArgumentError, "user_data cant be nil" if user_data.nil?
-          true
+          raise ArgumentError, "user_data can't be nil" if user_data.nil?
+          custom_spec = { 'customization_spec' => Hash.new }
+          user_data = YAML.load(user_data)
+          custom_spec['hostname']                    =  user_data['hostname'] if user_data.key?('hostname')
+          custom_spec['ipsettings']                  =  { 'ip' => user_data['ip'] } if user_data.key?('ip')
+          custom_spec['ipsettings']['subnetMask']    =  user_data['netmask'] if user_data.key?('netmask')
+          custom_spec['ipsettings']['dnsServerList'] =  user_data['dns'] if user_data.key?('dns')
+          custom_spec['domain']                      =  user_data['domain'] if user_data.key?('domain')
+          custom_spec['dnsSuffixList']               =  user_data['domain'] if user_data.key?('domain')
+          custom_spec['time_zone']                   =  user_data['timezone'] if user_data.key?('timezone')
+          custom_spec
         end
       end
     end