fixed a test

tbrand · Nov 16, 2016 · 75d2bec · 75d2bec
1 parent 765d10b
commit 75d2bec
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/support/cli.cr b/test/support/cli.cr
@@ -56,10 +56,14 @@ module Shards
             end
           elsif key.to_s == "targets"
             yml << "targets:\n"
-            unless value.nil?
+            if value.responds_to?(:each)
               value.each do |target, info|
                 yml << "  " << target.to_s << ":\n"
-                yml << "    main: " << info[:main].inspect << "\n"
+                if info.responds_to?(:each)
+                  info.each do |main, src|
+                    yml << "    main: " << src.inspect << "\n"
+                  end
+                end
               end
             end
           end