Add memory configuration

README.md

@@ -101,6 +101,7 @@ This provider has the following settings, all are required unless noted:
 * `proxy_host` - _Optional_ proxy host name for connecting to vSphere via proxy
 * `proxy_port` - _Optional_ proxy port number for connecting to vSphere via proxy
 * `vlan` - _Optional_ vlan to connect the first NIC to
+* `memory_mb` - _Optional_ Configure the amount of memory (in MB) for the new VM
 
 ### Cloning from a VM rather than a template
 

lib/vSphere/action/clone.rb

@@ -28,10 +28,12 @@ def call(env)
           begin
             location = get_location connection, machine, config, template
             spec = RbVmomi::VIM.VirtualMachineCloneSpec :location => location, :powerOn => true, :template => false
+            spec[:config] = RbVmomi::VIM.VirtualMachineConfigSpec
             customization_info = get_customization_spec_info_by_name connection, machine
 
             spec[:customization] = get_customization_spec(machine, customization_info) unless customization_info.nil?
             add_custom_vlan(template, dc, spec, config.vlan) unless config.vlan.nil?
+            add_custom_memory(spec, config.memory_mb) unless config.memory_mb.nil?
 
             env[:ui].info I18n.t('vsphere.creating_cloned_vm')
             env[:ui].info " -- #{config.clone_from_vm ? "Source" : "Template"} VM: #{template.pretty_path}"
@@ -137,7 +139,7 @@ def get_vm_base_folder(dc, template, config)
         end
 
         def add_custom_vlan(template, dc, spec, vlan)
-          spec[:config] = RbVmomi::VIM.VirtualMachineConfigSpec(:deviceChange => Array.new)
+          spec[:config][:deviceChange] = []
           network = get_network_by_name(dc, vlan)
           config = template.config
           card = config.hardware.device.grep(RbVmomi::VIM::VirtualEthernetCard).first or fail Errors::VSphereError, :missing_network_card
@@ -151,6 +153,10 @@ def add_custom_vlan(template, dc, spec, vlan)
           dev_spec = RbVmomi::VIM.VirtualDeviceConfigSpec(:device => card, :operation => "edit")
           spec[:config][:deviceChange].push dev_spec
         end
+
+        def add_custom_memory(spec, memory_mb)
+          spec[:config][:memoryMB] = Integer(memory_mb)
+        end
       end
     end
   end

lib/vSphere/config.rb

@@ -20,6 +20,7 @@ class Config < Vagrant.plugin('2', :config)
       attr_accessor :proxy_host
       attr_accessor :proxy_port
       attr_accessor :vlan
+      attr_accessor :memory_mb
 
       def validate(machine)
         errors = _detected_errors

spec/clone_spec.rb

@@ -12,7 +12,8 @@
     expect(@template).to have_received(:CloneVM_Task).with({
       :folder => @data_center,
       :name => NAME,
-      :spec => {:location => {:pool => @child_resource_pool} }
+      :spec => {:location => {:pool => @child_resource_pool}, 
+        :config => RbVmomi::VIM.VirtualMachineConfigSpec }
     })
   end
 
@@ -25,7 +26,8 @@
     expect(@template).to have_received(:CloneVM_Task).with({
       :folder => custom_base_folder,
       :name => NAME,
-      :spec => {:location => {:pool => @child_resource_pool} }
+      :spec => {:location => {:pool => @child_resource_pool},
+        :config => RbVmomi::VIM.VirtualMachineConfigSpec }
     })
   end
 
@@ -55,11 +57,22 @@
       :name => NAME,
       :spec => {:location => 
         {:pool => @child_resource_pool},
-        :config => expected_config
+          :config => expected_config
       }
     })
   end
 
+  it 'should create a CloneVM spec with configured memory_mb' do
+    @machine.provider_config.stub(:memory_mb).and_return(2048)
+    call
+    expect(@template).to have_received(:CloneVM_Task).with({
+      :folder => @data_center,
+      :name => NAME,
+      :spec => {:location => {:pool => @child_resource_pool}, 
+        :config => RbVmomi::VIM.VirtualMachineConfigSpec(:memoryMB => 2048) },
+    })
+  end
+
   it 'should set static IP when given config spec' do
     @machine.provider_config.stub(:customization_spec_name).and_return('spec')
     call
@@ -72,7 +85,8 @@
     expect(@template).to have_received(:CloneVM_Task).with({
       :folder => @data_center,
       :name => NAME,
-      :spec => {:location => {:pool => @root_resource_pool } }
+      :spec => {:location => {:pool => @root_resource_pool }, 
+        :config => RbVmomi::VIM.VirtualMachineConfigSpec }
     })
   end
 end

spec/spec_helper.rb

@@ -53,7 +53,8 @@ def call
         :linked_clone => nil,
         :proxy_host => nil,
         :proxy_port => nil,
-        :vlan => nil)
+        :vlan => nil,
+        :memory_mb => nil)
     vm_config = double(
       :vm => double('config_vm',
                     :box => nil,