test/components/forms/submit_test.rb in express_templates-0.7.0 vs test/components/forms/submit_test.rb in express_templates-0.7.1

@@ -9,9 +9,24 @@
 class SubmitTest < ActiveSupport::TestCase
   test "submit takes string param for value" do
     fragment = -> (ctx) {
       submit "Save it!"
     }
-    assert_match '<div class="field-wrapper"><input type="submit" name="commit" value="Save it!" /></div>', 
+    assert_match '<div class="field-wrapper"><input type="submit" name="commit" value="Save it!" /></div>',
                  arbre(&fragment)
   end
+  test "submit accepts a class option" do
+    fragment = -> (ctx) {
+      submit class: 'button'
+    }
+    assert_match '<div class="field-wrapper"><input type="submit" name="commit" value="Save" class="button" /></div>',
+                 arbre(&fragment)
+  end
+  test "submit accepts a class option when string provided as first param" do
+    fragment = -> (ctx) {
+      submit 'XYZ', class: 'button'
+    }
+    assert_match '<div class="field-wrapper"><input type="submit" name="commit" value="XYZ" class="button" /></div>',
+                 arbre(&fragment)
+  end
+
 end