Merge pull request #40 from JohnSundell/xctassert

Add special case for `XCTAssert` family of functions

Sources/Splash/Grammar/SwiftGrammar.swift

@@ -256,6 +256,13 @@ private extension SwiftGrammar {
                 return false
             }
 
+            // The XCTAssert family of functions is a bit of an edge case,
+            // since they start with capital letters. Since they are so
+            // commonly used, we'll add a special case for them here:
+            guard !segment.tokens.current.starts(with: "XCTAssert") else {
+                return false
+            }
+
             // In a generic declaration, only highlight constraints
             if segment.tokens.previous.isAny(of: "<", ",") {
                 var foundOpeningBracket = false

Tests/SplashTests/Tests/FunctionCallTests.swift

@@ -130,6 +130,15 @@ final class FunctionCallTests: SyntaxHighlighterTestCase {
         ])
     }
 
+    func testXCTAssertCalls() {
+        let components = highlighter.highlight("XCTAssertTrue(variable)")
+
+        XCTAssertEqual(components, [
+            .token("XCTAssertTrue", .call),
+            .plainText("(variable)")
+        ])
+    }
+
     func testAllTestsRunOnLinux() {
         XCTAssertTrue(TestCaseVerifier.verifyLinuxTests((type(of: self)).allTests))
     }
@@ -146,7 +155,8 @@ extension FunctionCallTests {
             ("testAccessingPropertyAfterFunctionCallWithArguments", testAccessingPropertyAfterFunctionCallWithArguments),
             ("testCallingStaticMethodOnGenericType", testCallingStaticMethodOnGenericType),
             ("testPassingTypeToFunction", testPassingTypeToFunction),
-            ("testPassingBoolToUnnamedArgument", testPassingBoolToUnnamedArgument)
+            ("testPassingBoolToUnnamedArgument", testPassingBoolToUnnamedArgument),
+            ("testXCTAssertCalls", testXCTAssertCalls)
         ]
     }
 }