From 3c638adcf57f06aa15f872fa154e93acd0d7536c Mon Sep 17 00:00:00 2001
From: Marco Capano <marco24capano@icloud.com>
Date: Wed, 5 Sep 2018 18:43:22 +0200
Subject: [PATCH] Adding tests for repeat, while, continue and Fallthrough

---
 Tests/SplashTests/Tests/StatementTests.swift | 111 +++++++++++++++++++
 1 file changed, 111 insertions(+)

diff --git a/Tests/SplashTests/Tests/StatementTests.swift b/Tests/SplashTests/Tests/StatementTests.swift
index 514c1d2..d8bd33e 100644
--- a/Tests/SplashTests/Tests/StatementTests.swift
+++ b/Tests/SplashTests/Tests/StatementTests.swift
@@ -134,6 +134,40 @@ final class StatementTests: SyntaxHighlighterTestCase {
             .plainText("}")
         ])
     }
+        
+    func testSwitchStatementWithFallthrough() {
+        let components = highlighter.highlight("""
+        switch variable {
+        case .one: fallthrough
+        default:
+            callB()
+        }
+        """)
+        
+        XCTAssertEqual(components, [
+            .token("switch", .keyword),
+            .whitespace(" "),
+            .plainText("variable"),
+            .whitespace(" "),
+            .plainText("{"),
+            .whitespace("\n"),
+            .token("case", .keyword),
+            .whitespace(" "),
+            .plainText("."),
+            .token("one", .dotAccess),
+            .plainText(":"),
+            .whitespace(" "),
+            .token("fallthrough", .keyword),
+            .whitespace("\n"),
+            .token("default", .keyword),
+            .plainText(":"),
+            .whitespace("\n    "),
+            .token("callB", .call),
+            .plainText("()"),
+            .whitespace("\n"),
+            .plainText("}")
+            ])
+    }
 
     func testForStatementWithStaticProperty() {
         let components = highlighter.highlight("for value in Enum.allCases { }")
@@ -154,6 +188,83 @@ final class StatementTests: SyntaxHighlighterTestCase {
             .plainText("}")
         ])
     }
+        
+    func testForStatementWithContinue() {
+        let components = highlighter.highlight("for value in Enum.allCases { continue }")
+        
+        XCTAssertEqual(components, [
+            .token("for", .keyword),
+            .whitespace(" "),
+            .plainText("value"),
+            .whitespace(" "),
+            .token("in", .keyword),
+            .whitespace(" "),
+            .token("Enum", .type),
+            .plainText("."),
+            .token("allCases", .property),
+            .whitespace(" "),
+            .plainText("{"),
+            .whitespace(" "),
+            .token("continue",.keyword),
+            .plainText("}")
+            ])
+    }
+    
+    func testRepeatWhileStatement() {
+        let components = highlighter.highlight("""
+        var x = 5
+        repeat {
+            print(x)
+            x = x - 1
+        } while x > 1
+        """)
+        
+        XCTAssertEqual(components, [
+            .token("var", .keyword),
+            .whitespace(" "),
+            .plainText("x"),
+            .whitespace(" "),
+            .plainText("="),
+            .whitespace(" "),
+            .token("5", .number),
+            .whitespace("\n"),
+            .token("repeat", .keyword),
+            .whitespace(" "),
+            .plainText("{"),
+            .whitespace("\n"),
+            .whitespace(" "),
+            .whitespace(" "),
+            .whitespace(" "),
+            .whitespace(" "),
+            .token("print", .call),
+            .plainText("(x)"),
+            .whitespace("\n"),
+            .whitespace(" "),
+            .whitespace(" "),
+            .whitespace(" "),
+            .whitespace(" "),
+            .plainText("x"),
+            .whitespace(" "),
+            .plainText("="),
+            .whitespace(" "),
+            .plainText("x"),
+            .whitespace(" "),
+            .plainText("-"),
+            .whitespace(" "),
+            .token("1", .number),
+            .whitespace("\n"),
+            .plainText("}"),
+            .whitespace(" "),
+            .token("while", .keyword),
+            .whitespace(" "),
+            .plainText("x"),
+            .whitespace(" "),
+            .plainText(">"),
+            .whitespace(" "),
+            .token("1", .number)
+            ])
+        
+    }
 
     func testAllTestsRunOnLinux() {
         XCTAssertTrue(TestCaseVerifier.verifyLinuxTests((type(of: self)).allTests))