lab-manager 1.0.5 → 1.0.6

data/lib/lab_manager.rb

@@ -19,6 +19,12 @@ require 'lab_manager/httpclient_patch'
 #
 class LabManager
 
+  @@DEBUG = 0
+
+  def self.DEBUG=(value)
+    @@DEBUG = value
+  end
+
   @@configPath = File.expand_path("~/.lab_manager")
   @@url = nil
 
@@ -108,7 +114,7 @@ class LabManager
 
     data = proxy.ListMachines(:configurationId => configurationId)
 
-    machines = Machine.fromList(data)
+    machines = Machine.from_list(data)
 
     if (!options[:exclude].nil?)
       machines = machines.find_all { |machine| 
@@ -129,7 +135,7 @@ class LabManager
   def proxy
     factory = SOAP::WSDLDriverFactory.new("#{@@url}?WSDL")
     proxy = factory.create_rpc_driver
-    #proxy.wiredump_dev = STDOUT
+    proxy.wiredump_dev = STDOUT if DEBUG
     proxy.generate_explicit_type = false  # No datatype with request
     proxy.headerhandler << LabManagerHeader.new(@organization, @workspace, @username, @password)
 

data/lib/lab_manager/machine.rb

@@ -1,13 +1,18 @@
 class Machine
 
-  def self.fromList(data)
+  def self.from_list(data)
     return [] if data.nil?
     return [] if data["ListMachinesResult"].nil?
     return [] if data["ListMachinesResult"]["Machine"].nil?
 
-    data["ListMachinesResult"]["Machine"].collect { |machine| 
-      Machine.new(machine)
-    }
+    machine_list = data["ListMachinesResult"]["Machine"]
+    if (machine_list.is_a? Array)
+      machine_list.collect { |machine| 
+        Machine.new(machine)
+      }
+    else
+      [Machine.new(machine_list)]
+    end
   end
 
   def self.to_csv(machines)

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: lab-manager
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: 
   segments: 
   - 1
   - 0
-  - 5
-  version: 1.0.5
+  - 6
+  version: 1.0.6
 platform: ruby
 authors: 
 - Ed Sumerfield