mirror of
https://github.com/apple/container.git
synced 2026-07-13 13:07:06 +00:00
J Logan
85 lines
3.4 KiB
Swift
85 lines
3.4 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
// Copyright © 2026 Apple Inc. and the container project authors.
|
|
//
|
|
// 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
|
|
|
|
/// Tests for the `--init-image` flag which allows specifying a custom init filesystem
|
|
/// image for microvms.
|
|
///
|
|
/// Note: A full integration test that verifies custom init behavior would require
|
|
/// a pre-built test init image that writes a marker to /dev/kmsg. This can be added
|
|
/// once a test init image is published to the registry.
|
|
@Suite
|
|
struct TestCLIRunInitImage {
|
|
private let alpine = ContainerFixture.warmupImages[0]
|
|
|
|
@Test func testRunWithNonExistentInitImage() async throws {
|
|
try await ContainerFixture.with { f in
|
|
let image = try f.copyWarmupImage(alpine)
|
|
let c = "\(f.testID)-c"
|
|
f.addCleanup { try? f.doRemove(c, force: true) }
|
|
let result = try f.run([
|
|
"run", "--rm", "--name", c, "-d",
|
|
"--init-image", "nonexistent.invalid/init-image:does-not-exist",
|
|
image, "sleep", "infinity",
|
|
])
|
|
#expect(result.status != 0, "run with non-existent init-image should fail")
|
|
}
|
|
}
|
|
|
|
@Test func testInitImageFlagInHelp() async throws {
|
|
try await ContainerFixture.with { f in
|
|
let result = try f.run(["run", "--help"]).check()
|
|
#expect(result.output.contains("--init-image"))
|
|
#expect(result.output.contains("custom init image"))
|
|
}
|
|
}
|
|
|
|
@Test func testCreateWithNonExistentInitImage() async throws {
|
|
try await ContainerFixture.with { f in
|
|
let image = try f.copyWarmupImage(alpine)
|
|
let c = "\(f.testID)-c"
|
|
f.addCleanup { try? f.doRemove(c, force: true) }
|
|
let result = try f.run([
|
|
"create", "--name", c,
|
|
"--init-image", "nonexistent.invalid/init-image:does-not-exist",
|
|
image, "echo", "hello",
|
|
])
|
|
#expect(result.status != 0, "create with non-existent init-image should fail")
|
|
}
|
|
}
|
|
|
|
@Test func testRunWithExplicitDefaultInitImage() async throws {
|
|
try await ContainerFixture.with { f in
|
|
let image = try f.copyWarmupImage(alpine)
|
|
let c = "\(f.testID)-c"
|
|
let config = try f.getSystemConfig()
|
|
try f.doLongRun(
|
|
name: c, image: image,
|
|
args: ["--init-image", config.vminit.image], autoRemove: false)
|
|
try await f.waitForContainerRunning(c)
|
|
f.addCleanup {
|
|
try? f.doStop(c)
|
|
try? f.doRemove(c)
|
|
}
|
|
|
|
let output = try f.doExec(c, cmd: ["echo", "hello"])
|
|
#expect(output.trimmingCharacters(in: .whitespacesAndNewlines) == "hello")
|
|
}
|
|
}
|
|
}
|