generate test data in a pythonic way

add create_test_data to conftest
remove generate_test_certificates from helper as it is on conftest already

Author: yeoldegrove

test-data/build-test-data.sh

@@ -13,8 +13,19 @@
 from cryptography.hazmat.primitives.asymmetric import rsa
 from dotenv import load_dotenv
 
-from .constants import TEST_DATA_DIR, GL_ROOT_DIR, CERT_DIR
-from .helper import call_command, spawn_background_process, generate_test_certificates
+from .constants import (
+    TEST_DATA_DIR,
+    GL_ROOT_DIR,
+    CERT_DIR,
+    GARDENLINUX_ROOT_DIR_EXAMPLE,
+    TEST_COMMIT,
+    TEST_VERSION,
+    TEST_PLATFORMS,
+    TEST_FEATURE_SET,
+    TEST_FEATURE_STRINGS_SHORT,
+    TEST_ARCHITECTURES,
+)
+from .helper import call_command, spawn_background_process
 
 
 def generate_test_certificates():
@@ -73,6 +84,48 @@ def write_zot_config(config_dict, file_path):
         json.dump(config_dict, config_file, indent=4)
 
 
+def create_test_data():
+    """Generate test data for OCI registry tests (replaces build-test-data.sh)"""
+    print("Creating fake artifacts...")
+
+    # Ensure the build directory exists
+    os.makedirs(GARDENLINUX_ROOT_DIR_EXAMPLE, exist_ok=True)
+
+    # Generate test artifacts for each combination
+    for platform in TEST_PLATFORMS:
+        for feature_string in TEST_FEATURE_STRINGS_SHORT:
+            for arch in TEST_ARCHITECTURES:
+                # Base name for the artifact
+                cname = (
+                    f"{platform}-{feature_string}-{arch}-{TEST_VERSION}-{TEST_COMMIT}"
+                )
+
+                print("Building mocked test data...")
+                # Create release file with metadata
+                release_file = f"{GARDENLINUX_ROOT_DIR_EXAMPLE}/{cname}.release"
+                with open(release_file, "w") as f:
+                    f.write(f"GARDENLINUX_FEATURES={TEST_FEATURE_SET}\n")
+                    f.write(f"GARDENLINUX_COMMIT_ID={TEST_COMMIT}\n")
+
+                # Create various file formats
+                for ext in ["raw", "tar", "qcow2", "vmdk"]:
+                    file_path = f"{GARDENLINUX_ROOT_DIR_EXAMPLE}/{cname}.{ext}"
+                    with open(file_path, "w") as f:
+                        f.write(f"dummy content for {file_path}")
+
+                # Create platform-specific files
+                if platform == "gcp":
+                    for ext in ["tar.gz", "gcpimage.tar.gz"]:
+                        file_path = f"{GARDENLINUX_ROOT_DIR_EXAMPLE}/{cname}.{ext}"
+                        with open(file_path, "w") as f:
+                            f.write(f"dummy content for {file_path}")
+
+                if platform == "azure":
+                    file_path = f"{GARDENLINUX_ROOT_DIR_EXAMPLE}/{cname}.vhd"
+                    with open(file_path, "w") as f:
+                        f.write(f"dummy content for {file_path}")
+
+
 @pytest.fixture(autouse=False, scope="function")
 def zot_session():
     load_dotenv()
@@ -108,8 +161,9 @@ def zot_session():
 
 def pytest_sessionstart(session):
     generate_test_certificates()
-    call_command("./test-data/build-test-data.sh --dummy")
-    call_command("mkdir -p manifests")
+    # Replace the bash script call with our Python function
+    create_test_data()
+    os.makedirs("./manifests", exist_ok=True)
 
 
 def pytest_sessionfinish(session):
@@ -119,3 +173,5 @@ def pytest_sessionfinish(session):
         os.remove(CERT_DIR + "/oci-sign.key")
     if os.path.isdir("./manifests"):
         shutil.rmtree("./manifests")
+    if os.path.isdir(GARDENLINUX_ROOT_DIR_EXAMPLE):
+        shutil.rmtree(GARDENLINUX_ROOT_DIR_EXAMPLE)

tests/conftest.py

@@ -1,5 +1,7 @@
 # -*- coding: utf-8 -*-
 
+from python_gardenlinux_lib.features.parse_features import get_gardenlinux_commit
+
 TEST_DATA_DIR = "test-data"
 GL_ROOT_DIR = f"{TEST_DATA_DIR}/gardenlinux"
 CERT_DIR = f"{TEST_DATA_DIR}/cert"
@@ -10,3 +12,10 @@
 REPO_NAME = "gardenlinux-example"
 CONTAINER_NAME_ZOT_EXAMPLE = f"{REGISTRY}/{REPO_NAME}"
 GARDENLINUX_ROOT_DIR_EXAMPLE = f"{TEST_DATA_DIR}/gardenlinux/.build"
+
+TEST_PLATFORMS = ["aws", "azure", "gcp", "openstack", "openstackbaremetal", "metal"]
+TEST_ARCHITECTURES = ["arm64", "amd64"]
+TEST_FEATURE_STRINGS_SHORT = ["gardener_prod"]
+TEST_FEATURE_SET = "_slim,base,container"
+TEST_COMMIT = get_gardenlinux_commit(GL_ROOT_DIR, 8)
+TEST_VERSION = "1000.0"

tests/constants.py

@@ -24,34 +24,3 @@ def call_command(cmd):
     except subprocess.CalledProcessError as e:
         error_message = e.stderr.decode("utf-8")
         return f"An error occurred: {error_message}"
-
-
-def generate_test_certificates():
-    """Generate self-signed certificates for testing"""
-    os.makedirs(CERT_DIR, exist_ok=True)
-    key_path = os.path.join(CERT_DIR, "oci-sign.key")
-    cert_path = os.path.join(CERT_DIR, "oci-sign.crt")
-    cmd = [
-        "openssl",
-        "req",
-        "-x509",
-        "-newkey",
-        "rsa:4096",
-        "-keyout",
-        key_path,
-        "-out",
-        cert_path,
-        "-days",
-        "365",
-        "-nodes",
-        "-subj",
-        "/CN=Garden Linux test signing key for oci",
-    ]
-    try:
-        subprocess.run(cmd, check=True)
-        # Set proper permissions
-        os.chmod(key_path, 0o600)
-        print(f"Generated test certificates in {CERT_DIR}")
-    except subprocess.CalledProcessError as e:
-        print(f"Error generating certificates: {e}")
-        raise