Added an attribute to represent the button types. (#69)

* Added `HTMLButtonType`. * Added a test for button type.
JohnSundell · Jun 1, 2021 · 0320a8d · 0320a8d
1 parent 1b10c15
commit 0320a8d
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 2 deletions.
diff --git a/Sources/Plot/API/HTMLAttributes.swift b/Sources/Plot/API/HTMLAttributes.swift
@@ -311,6 +311,14 @@ public extension Attribute where Context == HTML.InputContext {
     }
 }
 
+public extension Node where Context == HTML.ButtonContext {
+    /// Assign a button type to the element.
+    /// - parameter type: The button type to assign.
+    static func type(_ type: HTMLButtonType) -> Node {
+        .attribute(named: "type", value: type.rawValue)
+    }
+}
+
 public extension Node where Context == HTML.TextAreaContext {
     /// Specify the number of columns that the text area should contain.
     /// - parameter columns: The number of columns to specify.

diff --git a/Sources/Plot/API/HTMLButtonType.swift b/Sources/Plot/API/HTMLButtonType.swift
@@ -0,0 +1,15 @@
+/**
+*  Plot
+*  Copyright (c) John Sundell 2019
+*  MIT license, see LICENSE file for details
+*/
+
+import Foundation
+
+/// Enum that defines various button types that can be used with the
+/// `<button/>` HTML element. For example `.button(.type(.submit))`.
+public enum HTMLButtonType: String {
+    case button
+    case reset
+    case submit
+}
diff --git a/Tests/PlotTests/HTMLTests.swift b/Tests/PlotTests/HTMLTests.swift
@@ -608,11 +608,15 @@ final class HTMLTests: XCTestCase {
 
     func testButton() {
         let html = HTML(.body(
-            .button(.name("Name"), .value("Value"), .text("Text"))
+            .button(.type(.button), .name("Name"), .value("Value"), .text("Text")),
+            .button(.type(.submit), .text("Submit"))
         ))
 
         assertEqualHTMLContent(html, """
-        <body><button name="Name" value="Value">Text</button></body>
+        <body>\
+        <button type="button" name="Name" value="Value">Text</button>\
+        <button type="submit">Submit</button>\
+        </body>
         """)
     }