Add Sanity tests

tests/.copyright.tmpl

@@ -0,0 +1 @@
+Copyright ${years} ${owner}.

tests/requirements-dev.txt

@@ -0,0 +1,5 @@
+black==24.3.0
+codespell==2.2.4
+flake8==6.0.0
+isort==5.12.0
+licenseheaders==0.8.8

tests/requirements-test.txt

@@ -0,0 +1,5 @@
+coverage[toml]==7.2.5
+pytest==7.3.1
+PyYAML==6.0.1
+tenacity==8.2.3
+git+https://github.com/canonical/k8s-test-harness.git@main

tests/sanity/test_rock.py

@@ -0,0 +1,20 @@
+#
+# Copyright 2024 Canonical, Ltd.
+#
+import subprocess
+import pytest
+from k8s_test_harness.util import docker_util, env_util
+
+
+@pytest.mark.parametrize("image_version", ["0.8.0"])
+def test_sanity(image_version):
+    rock = env_util.get_build_meta_info_for_rock_version(
+        "rawfile-localpv", image_version, "amd64"
+    )
+    image = rock.image
+    args = ["python3", "-c", "import app.consts; print(app.consts.PROVISIONER_VERSION)"]
+    try:
+        process = docker_util.run_in_docker(image, args)
+    except subprocess.CalledProcessError as e:
+        assert False, e.stderr or e.stdout
+    assert image_version in process.stdout, process.stderr

tests/tox.ini

@@ -0,0 +1,59 @@
+[tox]
+no_package = True
+skip_missing_interpreters = True
+env_list = format, lint, integration
+min_version = 4.0.0
+
+[testenv]
+set_env =
+    PYTHONBREAKPOINT=pdb.set_trace
+    PY_COLORS=1
+pass_env =
+    PYTHONPATH
+
+[testenv:format]
+description = Apply coding style standards to code
+deps = -r {tox_root}/requirements-dev.txt
+commands =
+    licenseheaders -t {tox_root}/.copyright.tmpl -cy -o 'Canonical, Ltd' -d {tox_root}/sanity
+    isort {tox_root}/sanity --profile=black
+    black {tox_root}/sanity
+
+[testenv:lint]
+description = Check code against coding style standards
+deps = -r {tox_root}/requirements-dev.txt
+commands =
+    codespell {tox_root}/sanity
+    flake8 {tox_root}/sanity
+    licenseheaders -t {tox_root}/.copyright.tmpl -cy -o 'Canonical, Ltd' -d {tox_root}/sanity --dry
+
+[testenv:sanity]
+description = Run sanity tests
+deps =
+    -r {tox_root}/requirements-test.txt
+commands =
+    pytest -v \
+        --maxfail 1 \
+        --tb native \
+        --log-cli-level DEBUG \
+        --disable-warnings \
+        {posargs} \
+        {tox_root}/sanity
+pass_env =
+    TEST_*
+    ROCK_*
+    BUILT_ROCKS_METADATA
+
+[testenv: integration]
+allowlist_externals =
+    echo
+commands =
+# TODO: Implement integration tests here
+    echo "WARNING: This is a placeholder test - no test is implemented here."
+
+[flake8]
+max-line-length = 120
+select = E,W,F,C,N
+ignore = W503
+exclude = venv,.git,.tox,.tox_env,.venv,build,dist,*.egg_info
+show-source = true