crossmate

ShareLinkShortenerTests.swift (7569B)

---

 import Foundation
 import Testing
 
 @testable import Crossmate
 
 @Suite("Share link shortener")
 struct ShareLinkShortenerTests {
 
     private let base = URL(string: "https://crossmate.example.net")!
     private let token = "0a1BcDeFgHiJkLmNoPqRsTuVw"
     private var share: URL { URL(string: "https://www.icloud.com/share/\(token)")! }
 
     @Test("rewrites an iCloud share URL to the worker short form")
     func rewritesShareURL() {
         let short = ShareLinkShortener.shortURL(for: share, title: nil, shape: nil, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)")
     }
 
     @Test("drops the title-slug fragment from the short link")
     func dropsFragment() {
         let fragmented = URL(string: "https://www.icloud.com/share/\(token)#Saturday_Stumper")!
         let short = ShareLinkShortener.shortURL(for: fragmented, title: nil, shape: nil, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)")
     }
 
     @Test("encodes a custom game title as a t-tagged base64url segment")
     func encodesTitleSegment() {
         let short = ShareLinkShortener.shortURL(
             for: share, title: "Saturday Stumper", shape: nil, baseURL: base
         )
         #expect(
             short.absoluteString
                 == "https://crossmate.example.net/s/\(token)/tU2F0dXJkYXkgU3R1bXBlcg"
         )
     }
 
     @Test("maps base64 plus and slash into the URL-safe alphabet")
     func encodesTitleURLSafely() {
         // The em dash and π force '+' into the standard base64 encoding.
         let short = ShareLinkShortener.shortURL(
             for: share, title: "Sunday? Yes — π!", shape: nil, baseURL: base
         )
         #expect(
             short.absoluteString
                 == "https://crossmate.example.net/s/\(token)/tU3VuZGF5PyBZZXMg4oCUIM-AIQ"
         )
     }
 
     @Test("collapses a standard NYT weekday title to a single digit")
     func encodesWeekdayTitle() {
         let cases: [(String, String)] = [
             ("Sunday Crossword", "0"),
             ("Monday Crossword", "1"),
             ("Saturday Crossword", "6"),
             ("  Monday Crossword  ", "1")
         ]
         for (title, digit) in cases {
             let short = ShareLinkShortener.shortURL(for: share, title: title, shape: nil, baseURL: base)
             #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)/\(digit)")
         }
     }
 
     @Test("a near-miss weekday title falls back to the base64url form")
     func nonWeekdayTitleStaysCustom() {
         // No trailing "Crossword", so it is not the NYT shortcut.
         let short = ShareLinkShortener.shortURL(for: share, title: "Monday", shape: nil, baseURL: base)
         #expect(
             short.absoluteString == "https://crossmate.example.net/s/\(token)/tTW9uZGF5"
         )
     }
 
     @Test("truncates an over-long custom title to the cap")
     func truncatesLongTitle() {
         let long = String(repeating: "A", count: ShareLinkShortener.maxCustomTitleLength + 12)
         let capped = String(long.prefix(ShareLinkShortener.maxCustomTitleLength))
         let expected = "t" + GridSilhouette.base64URLEncode([UInt8](capped.utf8))
         let short = ShareLinkShortener.shortURL(for: share, title: long, shape: nil, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)/\(expected)")
     }
 
     @Test("appends the grid silhouette after the title")
     func appendsShapeAfterTitle() {
         // 3×3, centre cell blocked → symmetric → "s3CA".
         var blocks = [Bool](repeating: false, count: 9)
         blocks[4] = true
         let grid = GridSilhouette.Grid(side: 3, blocks: blocks)
         let short = ShareLinkShortener.shortURL(
             for: share, title: "Monday Crossword", shape: grid, baseURL: base
         )
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)/1/s3CA")
     }
 
     @Test("emits the silhouette segment even when there is no title")
     func appendsShapeWithoutTitle() {
         var blocks = [Bool](repeating: false, count: 9)
         blocks[4] = true
         let grid = GridSilhouette.Grid(side: 3, blocks: blocks)
         let short = ShareLinkShortener.shortURL(for: share, title: nil, shape: grid, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)/s3CA")
     }
 
     @Test("appends a rectangular silhouette under the uppercase form")
     func appendsRectangularShape() {
         // 2×3, all empty → symmetric → "S23AA".
         let grid = GridSilhouette.Grid(width: 2, height: 3, blocks: [Bool](repeating: false, count: 6))
         let short = ShareLinkShortener.shortURL(for: share, title: nil, shape: grid, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)/S23AA")
     }
 
     @Test("omits a silhouette whose block count doesn't match its dimensions")
     func omitsMismatchedShape() {
         let grid = GridSilhouette.Grid(side: 3, blocks: [Bool](repeating: false, count: 6))
         let short = ShareLinkShortener.shortURL(for: share, title: nil, shape: grid, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)")
     }
 
     @Test("omits the title segment for an empty or whitespace title")
     func omitsBlankTitle() {
         for title in [nil, "", "   ", "\n"] as [String?] {
             let short = ShareLinkShortener.shortURL(for: share, title: title, shape: nil, baseURL: base)
             #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)")
         }
     }
 
     @Test("accepts the bare icloud.com host")
     func acceptsBareHost() {
         let bare = URL(string: "https://icloud.com/share/\(token)")!
         let short = ShareLinkShortener.shortURL(for: bare, title: nil, shape: nil, baseURL: base)
         #expect(short.absoluteString == "https://crossmate.example.net/s/\(token)")
     }
 
     @Test("passes through unchanged when no base URL is configured")
     func passesThroughWithoutBase() {
         let fragmented = URL(string: "https://www.icloud.com/share/\(token)#Title")!
         #expect(
             ShareLinkShortener.shortURL(for: fragmented, title: "Title", shape: nil, baseURL: nil)
                 == fragmented
         )
     }
 
     @Test(
         "passes through URLs that are not a CKShare link",
         arguments: [
             "https://example.com/share/0a1BcDeFgHiJkLmNoPqRsTuVw",
             "http://www.icloud.com/share/0a1BcDeFgHiJkLmNoPqRsTuVw",
             "https://www.icloud.com/notes/0a1BcDeFgHiJkLmNoPqRsTuVw",
             "https://www.icloud.com/share",
             "https://www.icloud.com/share/0a1BcDeF/extra"
         ]
     )
     func passesThroughForeignURLs(raw: String) {
         let url = URL(string: raw)!
         #expect(ShareLinkShortener.shortURL(for: url, title: nil, shape: nil, baseURL: base) == url)
     }
 
     @Test("rejects tokens outside the unreserved charset or length bounds")
     func rejectsMalformedTokens() {
         // The encoded slash decodes into the last path component; passing it
         // through to the worker would change the redirect target's shape.
         let smuggled = URL(string: "https://www.icloud.com/share/0a1BcDeF%2F..%2FbAd")!
         #expect(ShareLinkShortener.shortURL(for: smuggled, title: nil, shape: nil, baseURL: base) == smuggled)
 
         let tooShort = URL(string: "https://www.icloud.com/share/0a1Bc")!
         #expect(ShareLinkShortener.shortURL(for: tooShort, title: nil, shape: nil, baseURL: base) == tooShort)
     }
 }