CodeEditLanguages/Tests/CodeEditLanguagesTests/LanguageDetectionTests.swift at main · CodeEditApp/CodeEditLanguages · 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
125
//
//  LanguageDetectionTests.swift
//  CodeEditLanguages
//
//  Created by Khan Winter on 6/17/23.
//

import XCTest
@testable import CodeEditLanguages
import RegexBuilder

// swiftlint:disable all
final class LanguageDetectionTests: XCTestCase {
    func test_shebang() {
        let validCases = [
            "#! /usr/bin/env swift",
            "#! /usr/bin/env -S swift",
            "#!/usr/bin/env swift",
            "#!     /usr/bin/env         swift",
            "#! swift",
            "#!swift"
        ]

        let invalidCases = [
            "1234",
            "import CodeEditLanguages",
            "",
            "\n",
            "# This is a regular comment",
            "#!",
            "#",
            "!"
        ]

        for validCase in validCases {
            let detectedLanguage = CodeLanguage.detectLanguageFrom(url: URL(filePath: ""), prefixBuffer: validCase)
            XCTAssert(
                detectedLanguage == .swift,
                "Parser failed to find valid shebang language. Given \"\(validCase)\", found \(detectedLanguage), expected swift."
            )
        }

        for invalidCase in invalidCases {
            let detectedLanguage = CodeLanguage.detectLanguageFrom(url: URL(filePath: ""), prefixBuffer: invalidCase)
            XCTAssert(
                detectedLanguage == .default,
                "Parser unexpectedly found valid shebang language. Given \"\(invalidCase)\", found \(detectedLanguage), expected default."
            )
        }
    }

    func test_vimModeline() {
        let validCases = [
            "// vim: ft=swift",
            "/* vim: ft=swift */",
            "/* vim: other=param a=b ft=swift b=d\n*/",
            "// vim: other=param a=b ft=swift b=d",
        ]

        let invalidCases = [
            "//vim:",
            "//vim",
            "// This is a normal comment",
            "/*vim",
            "/* vim",
            "/* this is also a normal comment */",
            "1234",
            "import CodeEditLanguages",
            "",
            "\n"
        ]

        for validCase in validCases {
            let detectedLanguage = CodeLanguage.detectLanguageFrom(url: URL(filePath: ""), prefixBuffer: validCase)
            XCTAssert(
                detectedLanguage == .swift,
                "Parser failed to find valid shebang language. Given \"\(validCase)\", found \(detectedLanguage), expected swift."
            )
        }

        for invalidCase in invalidCases {
            let detectedLanguage = CodeLanguage.detectLanguageFrom(url: URL(filePath: ""), prefixBuffer: invalidCase)
            XCTAssert(
                detectedLanguage == .default,
                "Parser unexpectedly found valid shebang language. Given \"\(invalidCase)\", found \(detectedLanguage), expected default."
            )
        }
    }

    func test_emacsModeline() {
        let validCases = [
            "-*- mode:swift -*-",
            "-*- indent-style: tabs mode: swift -*-",
            "-*-mode: swift-*-"
        ]

        let invalidCases = [
            "1234",
            "import CodeEditLanguages",
            "",
            "\n",
            "-*-",
            "-*- -*-",
            "-- -*-",
            "mode:swift"
        ]

        for validCase in validCases {
            let detectedLanguage = CodeLanguage.detectLanguageFrom(url: URL(filePath: ""), prefixBuffer: validCase)
            XCTAssert(
                detectedLanguage == .swift,
                "Parser failed to find valid shebang language. Given \"\(validCase)\", found \(detectedLanguage), expected swift."
            )
        }

        for invalidCase in invalidCases {
            let detectedLanguage = CodeLanguage.detectLanguageFrom(url: URL(filePath: ""), prefixBuffer: invalidCase)
            XCTAssert(
                detectedLanguage == .default,
                "Parser unexpectedly found valid shebang language. Given \"\(invalidCase)\", found \(detectedLanguage), expected default."
            )
        }
    }
}
// swiftlint:enable all