Files
sourcekit-lsp/Tests/SourceKitDTests/SourceKitDRegistryTests.swift
Alex Hoppen be546308ca Use URL in many cases where we used AbsolutePath
We made quite a few fixes recently to make sure that path handling works correctly using `URL` on Windows. Use `URL` in most places to have a single type that represents file paths instead of sometimes using `AbsolutePath`.

While doing so, also remove usages of `TSCBasic.FileSystem` an `InMemoryFileSystem`. The pattern of using `InMemoryFileSystem` for tests was never consistently used and it was a little confusing that some types took a `FileSystem` parameter while other always assumed to work on the local file system.
2024-11-18 18:19:48 -08:00

87 lines
3.2 KiB
Swift

//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2020 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
import Csourcekitd
import LanguageServerProtocolExtensions
import SKTestSupport
import SourceKitD
import SwiftExtensions
import TSCBasic
import XCTest
final class SourceKitDRegistryTests: XCTestCase {
func testAdd() async throws {
let registry = SourceKitDRegistry()
let a = await FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry)
let b = await FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/b"), in: registry)
let a2 = await FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry)
XCTAssert(a === a2)
XCTAssert(a !== b)
}
func testRemove() async throws {
let registry = SourceKitDRegistry()
let a = await FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry)
await assertTrue(registry.remove(URL(fileURLWithPath: "/a")) === a)
await assertNil(registry.remove(URL(fileURLWithPath: "/a")))
}
func testRemoveResurrect() async throws {
let registry = SourceKitDRegistry()
@inline(never)
func scope(registry: SourceKitDRegistry) async throws -> UInt32 {
let a = await FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry)
await assertTrue(a === FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry))
await assertTrue(registry.remove(URL(fileURLWithPath: "/a")) === a)
// Resurrected.
await assertTrue(a === FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry))
// Remove again.
await assertTrue(registry.remove(URL(fileURLWithPath: "/a")) === a)
return (a as! FakeSourceKitD).token
}
let id = try await scope(registry: registry)
let a2 = await FakeSourceKitD.getOrCreate(URL(fileURLWithPath: "/a"), in: registry)
XCTAssertNotEqual(id, (a2 as! FakeSourceKitD).token)
}
}
private let nextToken = AtomicUInt32(initialValue: 0)
final class FakeSourceKitD: SourceKitD {
let token: UInt32
var api: sourcekitd_api_functions_t { fatalError() }
var keys: sourcekitd_api_keys { fatalError() }
var requests: sourcekitd_api_requests { fatalError() }
var values: sourcekitd_api_values { fatalError() }
func addNotificationHandler(_ handler: SKDNotificationHandler) { fatalError() }
func removeNotificationHandler(_ handler: SKDNotificationHandler) { fatalError() }
private init() {
token = nextToken.fetchAndIncrement()
}
static func getOrCreate(_ url: URL, in registry: SourceKitDRegistry) async -> SourceKitD {
return await registry.getOrAdd(url, create: { Self.init() })
}
package func log(request: SKDRequestDictionary) {}
package func log(response: SKDResponse) {}
package func log(crashedRequest: SKDRequestDictionary, fileContents: String?) {}
package func logRequestCancellation(request: SKDRequestDictionary) {}
}