Added config test for :simpletext formatter

robd · May 6, 2015 · e4e0a1a · e4e0a1a
1 parent 92590aa
commit e4e0a1a
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/test/unit/test_configuration.rb b/test/unit/test_configuration.rb
@@ -51,15 +51,17 @@ def test_command_map
       assert_equal "/opt/sites/example/current/bin ruby", SSHKit.config.command_map[:ruby]
     end
 
-    def test_setting_formatter_to_dot
-      assert SSHKit.config.format = :dot
-      assert SSHKit.config.output.is_a? SSHKit::Formatter::Dot
+    def test_setting_formatter_types
+      {
+        dot:        SSHKit::Formatter::Dot,
+        blackhole:  SSHKit::Formatter::BlackHole,
+        simpletext: SSHKit::Formatter::SimpleText,
+      }.each do |format, expected_class|
+        SSHKit.config.format = format
+        assert SSHKit.config.output.is_a? expected_class
+      end
     end
 
-    def test_setting_formatter_to_blackhole
-      assert SSHKit.config.format = :BlackHole
-      assert SSHKit.config.output.is_a? SSHKit::Formatter::BlackHole
-    end
   end
 
 end