krz/hutch

 1import Foundation
 2import Testing
 3import UIKit
 4@testable import Hutch
 5
 6@MainActor
 7struct SyntaxHighlighterTests {
 8    @Test
 9    func mapsFileNamesToLanguages() {
10        #expect(SyntaxHighlighter.language(forFileName: "Sources/App/Main.swift") == "swift")
11        #expect(SyntaxHighlighter.language(forFileName: "server.py") == "python")
12        #expect(SyntaxHighlighter.language(forFileName: "index.tsx") == "typescript")
13        #expect(SyntaxHighlighter.language(forFileName: "Dockerfile") == "dockerfile")
14        #expect(SyntaxHighlighter.language(forFileName: "Makefile") == "makefile")
15    }
16
17    @Test
18    func returnsNilForUnknownExtensions() {
19        #expect(SyntaxHighlighter.language(forFileName: "notes.unknownext") == nil)
20        #expect(SyntaxHighlighter.language(forFileName: "LICENSE") == nil)
21    }
22
23    @Test
24    func highlightsKnownLanguageAsColoredSpans() {
25        let highlighter = SyntaxHighlighter(theme: .light)
26        let html = highlighter.highlightedHTML(for: "let answer = 42", language: "swift")
27
28        #expect(html?.contains("<span style=\"color:") == true)
29        #expect(html?.contains("answer") == true)
30    }
31
32    @Test
33    func fallsBackToNilForUnsupportedLanguage() {
34        let highlighter = SyntaxHighlighter(theme: .dark)
35        #expect(highlighter.highlightedHTML(for: "plain text", language: nil) == nil)
36        #expect(highlighter.highlightedHTML(for: "plain text", language: "totally-not-a-language") == nil)
37    }
38
39    @Test
40    func producesAttributedTextForNativeViewer() {
41        let highlighter = SyntaxHighlighter(theme: .light)
42        let font = UIFont.monospacedSystemFont(ofSize: 12, weight: .regular)
43        let attributed = highlighter.attributedText(for: "print(\"hi\")", language: "python", font: font)
44
45        #expect(attributed?.string.contains("print") == true)
46    }
47}