Files
container/Tests/ContainerPluginTests/PluginLoaderTest.swift
T
J Logan 6242706c66 Fixes for install root and plugin detection. (#467)
- Sets up API server as source of truth for installation root, similarly
to what was done for the data root. `system start` establishes the
install root, setting the environment variable `CONTAINER_INSTALL_ROOT`
when launching the API server.
- The API server propagates the environment variable when launching
helpers, and returns the install root to the CLI via the health check
XPC.
- Includes several fixes for detecting plugins that use app bundle
layout.
2025-08-08 21:44:26 -07:00

110 lines
4.4 KiB
Swift

//===----------------------------------------------------------------------===//
// Copyright © 2025 Apple Inc. and the container project authors. All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//===----------------------------------------------------------------------===//
//
import Foundation
import Testing
@testable import ContainerPlugin
struct PluginLoaderTest {
@Test
func testFindAll() async throws {
let tempURL = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString)
defer { try? FileManager.default.removeItem(at: tempURL) }
let factory = try setupMock(tempURL: tempURL)
let loader = try PluginLoader(
appRoot: tempURL,
installRoot: URL(filePath: "/usr/local/"),
pluginDirectories: [tempURL],
pluginFactories: [factory]
)
let plugins = loader.findPlugins()
#expect(Set(plugins.map { $0.name }) == Set(["cli", "service"]))
}
@Test
func testFindAllSymlink() async throws {
let tempURL = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString)
defer { try? FileManager.default.removeItem(at: tempURL) }
let factory = try setupMock(tempURL: tempURL)
// move the CLI plugin elsewhere and symlink it
let otherTempURL = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString)
defer { try? FileManager.default.removeItem(at: otherTempURL) }
try FileManager.default.createDirectory(at: otherTempURL, withIntermediateDirectories: true)
let srcURL = tempURL.appendingPathComponent("cli")
let dstURL = otherTempURL.appendingPathComponent("cli")
try FileManager.default.moveItem(
at: srcURL,
to: dstURL
)
try FileManager.default.createSymbolicLink(
at: srcURL,
withDestinationURL: dstURL
)
let loader = try PluginLoader(
appRoot: tempURL,
installRoot: URL(filePath: "/usr/local/"),
pluginDirectories: [tempURL],
pluginFactories: [factory]
)
let plugins = loader.findPlugins()
#expect(Set(plugins.map { $0.name }) == Set(["cli", "service"]))
}
@Test
func testFindByName() async throws {
let tempURL = FileManager.default.temporaryDirectory.appendingPathComponent(UUID().uuidString)
defer { try? FileManager.default.removeItem(at: tempURL) }
let factory = try setupMock(tempURL: tempURL)
let loader = try PluginLoader(
appRoot: tempURL,
installRoot: URL(filePath: "/usr/local/"),
pluginDirectories: [tempURL],
pluginFactories: [factory]
)
#expect(loader.findPlugin(name: "cli")?.name == "cli")
#expect(loader.findPlugin(name: "service")?.name == "service")
#expect(loader.findPlugin(name: "throw") == nil)
}
private func setupMock(tempURL: URL) throws -> MockPluginFactory {
let cliConfig = PluginConfig(abstract: "cli", author: "CLI", servicesConfig: nil)
let cliPlugin: Plugin = Plugin(binaryURL: URL(filePath: "/bin/cli"), config: cliConfig)
let serviceServicesConfig = PluginConfig.ServicesConfig(
loadAtBoot: false,
runAtLoad: false,
services: [PluginConfig.Service(type: .runtime, description: nil)],
defaultArguments: []
)
let serviceConfig = PluginConfig(abstract: "service", author: "SERVICE", servicesConfig: serviceServicesConfig)
let servicePlugin: Plugin = Plugin(binaryURL: URL(filePath: "/bin/service"), config: serviceConfig)
let mockPlugins = [
"cli": cliPlugin,
MockPluginFactory.throwSuffix: nil,
"service": servicePlugin,
]
return try MockPluginFactory(tempURL: tempURL, plugins: mockPlugins)
}
}