Update facts_for_platform_by_name to take an array

danieldreier · Dec 15, 2014 · 356e3ee · 356e3ee
1 parent 8056a12
commit 356e3ee
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/lib/puppet_spec_facts.rb b/lib/puppet_spec_facts.rb
@@ -54,8 +54,13 @@ def self.facts_for_platform_by_fact(factstyle: nil, select_facts: {})
     }
   end
 
-  def self.facts_for_platform_by_name(platform)
-    puppet_platforms[platform]
+  def self.facts_for_platform_by_name(platforms)
+    platforms = [platforms] if platforms.is_a?(String)
+    results = {}
+    platforms.each do |name|
+      results[:"#{name}"] = puppet_platforms[name]
+    end
+    results
   end
 
   def self.resave_all_facts