83475cc9f4
Treat outputs like we treat trees: store them in the object store. This simplifies using osbuild and allows returning a cached version if one is available. This makes the `--output` parameter redundant. Remove it.
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
from dataclasses import dataclass
|
|
from enum import Enum
|
|
from typing import List, Callable, Any
|
|
|
|
from . import evaluate_test, rel_path
|
|
from .build import run_osbuild
|
|
from .run import run_image, extract_image
|
|
from test.integration_tests.config import *
|
|
|
|
|
|
class IntegrationTestType(Enum):
|
|
EXTRACT=0
|
|
BOOT_WITH_QEMU=1
|
|
|
|
|
|
@dataclass
|
|
class IntegrationTestCase:
|
|
name: str
|
|
pipeline: str
|
|
build_pipeline: str
|
|
output_image: str
|
|
test_cases: List[Callable[[Any], None]]
|
|
type: IntegrationTestType
|
|
|
|
def run(self):
|
|
tree_id, output_id = run_osbuild(rel_path(f"pipelines/{self.pipeline}"), self.build_pipeline)
|
|
if self.type == IntegrationTestType.BOOT_WITH_QEMU:
|
|
self.run_and_test(output_id)
|
|
else:
|
|
self.extract_and_test(output_id)
|
|
|
|
def run_and_test(self, output_id):
|
|
r = run_image(f"{OBJECTS}/refs/{output_id}/{self.output_image}")
|
|
for test in self.test_cases:
|
|
evaluate_test(test, r.stdout)
|
|
|
|
def extract_and_test(self, output_id):
|
|
with extract_image(f"{OBJECTS}/refs/{output_id}/{self.output_image}") as fstree:
|
|
for test in self.test_cases:
|
|
evaluate_test(lambda: test(fstree), name=test.__name__)
|