Plot/Tests/PlotTests/NodeTests.swift at 8df605e3e3ad3e5149784719e3d2d10b73e9c34a · JohnSundell/Plot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/**
*  Plot
*  Copyright (c) John Sundell 2019
*  MIT license, see LICENSE file for details
*/

import XCTest
import Plot

final class NodeTests: XCTestCase {
    func testEscapingText() {
        let node = Node<Any>.text("Hello & welcome to <Plot>!;")
        XCTAssertEqual(node.render(), "Hello &amp; welcome to &lt;Plot&gt;!;")
    }

    func testEscapingDoubleAmpersands() {
        let node = Node<Any>.text("&&")
        XCTAssertEqual(node.render(), "&amp;&amp;")
    }

    func testEscapingAmpersandFollowedByComparisonSymbols() {
        let node = Node<Any>.text("&< &>")
        XCTAssertEqual(node.render(), "&amp;&lt; &amp;&gt;")
    }

    func testNotDoubleEscapingText() {
        let node = Node<Any>.text("Hello &amp; welcome&#160;to &lt;Plot&gt;!&text")
        XCTAssertEqual(node.render(), "Hello &amp; welcome&#160;to &lt;Plot&gt;!&amp;text")
    }

    func testNotEscapingRawString() {
        let node = Node<Any>.raw("Hello & welcome to <Plot>!")
        XCTAssertEqual(node.render(), "Hello & welcome to <Plot>!")
    }

    func testGroup() {
        let node = Node<Any>.group(.text("Hello"), .text("World"))
        XCTAssertEqual(node.render(), "HelloWorld")
    }

    func testCustomElement() {
        let node = Node<Any>.element(named: "custom")
        XCTAssertEqual(node.render(), "<custom></custom>")
    }

    func testCustomAttribute() {
        let node = Node<Any>.attribute(named: "key", value: "value")
        XCTAssertEqual(node.render(), #"key="value""#)
    }

    func testEscapingDoubleQuotesInAttributeValue() {
        let node = Node<Any>.attribute(named: "key", value: #"a"b"#)
        XCTAssertEqual(node.render(), #"key="a&quot;b""#)
    }

    func testEscapingSingleQuotesInAttributeValue() {
        let node = Node<Any>.attribute(named: "key", value: "a'b")
        XCTAssertEqual(node.render(), #"key="a&#39;b""#)
    }

    func testEscapingAngleBracketsInAttributeValue() {
        let node = Node<Any>.attribute(named: "key", value: "<script>alert(1)</script>")
        XCTAssertEqual(node.render(), #"key="&lt;script&gt;alert(1)&lt;/script&gt;""#)
    }

    func testEscapingAmpersandInAttributeValue() {
        let node = Node<Any>.attribute(named: "key", value: "a&b")
        XCTAssertEqual(node.render(), #"key="a&amp;b""#)
    }

    func testNotDoubleEscapingAttributeValue() {
        let node = Node<Any>.attribute(named: "key", value: "&amp;")
        XCTAssertEqual(node.render(), #"key="&amp;""#)
    }

    func testEscapingXSSInAttributeValue() {
        let node = Node<Any>.attribute(
            named: "value",
            value: #""><img src=x onerror=alert(1)><input value=""#
        )
        XCTAssertEqual(
            node.render(),
            #"value="&quot;&gt;&lt;img src=x onerror=alert(1)&gt;&lt;input value=&quot;""#
        )
    }

    func testCustomElementWithCustomAttribute() {
        let node = Node<Any>.element(named: "custom", attributes: [
            Attribute(name: "key", value: "value")
        ])

        XCTAssertEqual(node.render(), #"<custom key="value"></custom>"#)
    }

    func testCustomElementWithCustomAttributeWithSpecificContext() {
        let node = Node<Any>.element(named: "custom", attributes: [
            Attribute<String>(name: "key", value: "value")
        ])

        XCTAssertEqual(node.render(), #"<custom key="value"></custom>"#)
    }

    func testCustomSelfClosedElementWithCustomAttribute() {
        let node = Node<Any>.selfClosedElement(named: "custom", attributes: [
            Attribute(name: "key", value: "value")
        ])

        XCTAssertEqual(node.render(), #"<custom key="value"/>"#)
    }

    func testComponents() {
        let node = Node<Any>.components {
            Paragraph("One")
            Paragraph("Two")
        }

        XCTAssertEqual(node.render(), "<p>One</p><p>Two</p>")
    }

    func testNodeComponentBodyIsEqualToSelf() {
        let node = Node.p("Text")
        XCTAssertEqual(node.render(), node.body.render())
    }
}