mirror of
https://github.com/apple/container.git
synced 2026-07-14 21:47:01 +00:00
J Logan
88ab3192f9
- Part of #1833. - Distributes fixture extensions for images and containers added as part of previous builder PR to more sensible locations.
93 lines
3.5 KiB
Swift
93 lines
3.5 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
|
|
|
|
// MARK: - Image inspect types
|
|
|
|
extension ContainerFixture {
|
|
/// Decoded output of `container image inspect` or `container image list --format json`.
|
|
struct ImageInspectOutput: Codable {
|
|
struct Configuration: Codable { let name: String }
|
|
struct Variant: Codable {
|
|
struct Platform: Codable {
|
|
let os: String
|
|
let architecture: String
|
|
}
|
|
let platform: Platform
|
|
}
|
|
let configuration: Configuration
|
|
let variants: [Variant]
|
|
}
|
|
}
|
|
|
|
// MARK: - Image lifecycle helpers
|
|
|
|
extension ContainerFixture {
|
|
|
|
/// Pulls an image. Passes optional extra args (e.g. `["--platform", "linux/amd64"]`).
|
|
func doPull(_ imageName: String, args: [String] = []) throws {
|
|
var pullArgs = ["image", "pull"] + args
|
|
pullArgs.append(imageName)
|
|
try run(pullArgs).check()
|
|
}
|
|
|
|
/// Returns all images currently in the local store.
|
|
func doListImages() throws -> [ImageInspectOutput] {
|
|
let result = try run(["image", "list", "--format", "json"]).check()
|
|
return try JSONDecoder().decode([ImageInspectOutput].self, from: result.outputData)
|
|
}
|
|
|
|
/// Returns true if an image with the given exact reference is present.
|
|
func isImagePresent(_ targetImage: String) throws -> Bool {
|
|
try doListImages().contains { $0.configuration.name == targetImage }
|
|
}
|
|
|
|
/// Tags `image` with `newName`.
|
|
func doImageTag(_ image: String, newName: String) throws {
|
|
try run(["image", "tag", image, newName]).check()
|
|
}
|
|
|
|
/// Removes the given images, or all images when `images` is `nil`.
|
|
func doRemoveImages(_ images: [String]? = nil) throws {
|
|
var args = ["image", "rm"]
|
|
if let images { args.append(contentsOf: images) } else { args.append("--all") }
|
|
try run(args).check()
|
|
}
|
|
|
|
/// Returns the full inspect output for an image, including variant information.
|
|
func doInspectImages(_ name: String) throws -> [ImageInspectOutput] {
|
|
let result = try run(["image", "inspect", name]).check()
|
|
return try JSONDecoder().decode([ImageInspectOutput].self, from: result.outputData)
|
|
}
|
|
|
|
/// Returns the `configuration.name` of an image.
|
|
func inspectImage(_ name: String) throws -> String {
|
|
let outputs = try doInspectImages(name)
|
|
guard let first = outputs.first else {
|
|
throw CommandError.executionFailed("image '\(name)' not found in inspect output")
|
|
}
|
|
return first.configuration.name
|
|
}
|
|
|
|
/// Asserts that the image was successfully built and is present in the image store.
|
|
func assertImageBuilt(_ image: String) throws {
|
|
let name = try inspectImage(image)
|
|
#expect(name == image, "expected image \(image) to be present")
|
|
}
|
|
}
|