Fix issue chef-boneyard#19

Signed-off-by: Sebastian Gumprich <[email protected]>
rndmh3ro · Jan 11, 2018 · b78b273 · b78b273
1 parent 5d18923
commit b78b273
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/libraries/vm_advancedsetting.rb b/libraries/vm_advancedsetting.rb
@@ -19,10 +19,17 @@ def initialize(opts)
     @opts = opts
   end
 
-  def method_missing(name) # rubocop:disable Style/MethodMissing
-    advancedsetting[name.to_s]
+  def method_missing(*keys) # rubocop:disable Style/MethodMissing
+    if keys.is_a?(Array)
+      keys.shift if keys[0] == :[]
+      key = keys.first
+    else
+      key = keys
+    end
+    advancedsetting[key.to_s]
   end
 
+
   def to_s
     "vmware_vm_advancedsetting DC: #{@opts[:datacenter]} VM: #{@opts[:vm]}"
   end

diff --git a/libraries/vmhost_advancedsetting.rb b/libraries/vmhost_advancedsetting.rb
@@ -19,8 +19,14 @@ def initialize(opts)
     @opts = opts
   end
 
-  def method_missing(name) # rubocop:disable Style/MethodMissing
-    advancedsetting[name.to_s]
+  def method_missing(*keys) # rubocop:disable Style/MethodMissing
+    if keys.is_a?(Array)
+      keys.shift if keys[0] == :[]
+      key = keys.first
+    else
+      key = keys
+    end
+    advancedsetting[key.to_s]
   end
 
   private