# Common helper functions and test utilities for upgrade tests
# Used by upgrade.bats and upgrade_minimal.bats

# Verify prerequisites for upgrade tests
function verify_prerequisites {
    if [ ! $(command -v curl) ]; then
        echo "you need to install curl as a prerequisite to running the tests" >&3
        return 1
    fi

    if [ ! $(command -v jq) ]; then
        echo "you need to install jq as a prerequisite to running the tests" >&3
        return 1
    fi

    return 0
}

# Common teardown function - prints zot log on failure
function teardown() {
    # conditionally printing on failure is possible from teardown but not from teardown_file
    cat ${BATS_FILE_TMPDIR}/zot.log
}

# Common teardown_file function - stops all zot instances
function teardown_file() {
    zot_stop_all
}

# ==============================================================================
# COMMON HELPER FUNCTIONS
# These are the core functions used by both release and new test functions
# ==============================================================================

# Helper: Get the zot port
function get_zot_port() {
    cat ${BATS_FILE_TMPDIR}/zot.port
}

# Helper: Push an image using skopeo
# Args: $1 = image_name, $2 = tag
function helper_push_image() {
    local image_name=${1:-golang}
    local tag=${2:-1.20}
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --dest-tls-verify=false \
        oci:${TEST_DATA_DIR}/${image_name}:${tag} \
        docker://127.0.0.1:${zot_port}/${image_name}:${tag}
    [ "$status" -eq 0 ]
    run curl http://127.0.0.1:${zot_port}/v2/_catalog
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg name "${image_name}" 'any(.repositories[]; . == $name)') = true ]
    run curl http://127.0.0.1:${zot_port}/v2/${image_name}/tags/list
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg tag "${tag}" 'any(.tags[]; . == $tag)') = true ]
}

# Helper: Pull an image using skopeo
# Args: $1 = image_name, $2 = tag
function helper_pull_image() {
    local image_name=${1:-golang}
    local tag=${2:-1.20}
    local oci_data_dir=${BATS_FILE_TMPDIR}/oci
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --src-tls-verify=false \
        docker://127.0.0.1:${zot_port}/${image_name}:${tag} \
        oci:${oci_data_dir}/${image_name}:${tag}
    [ "$status" -eq 0 ]
    run cat ${oci_data_dir}/${image_name}/index.json
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg tag "${tag}" '.manifests[].annotations."org.opencontainers.image.ref.name" == $tag') = true ]
}

# Helper: Push an image index (multi-arch)
# Args: $1 = source_image, $2 = dest_image_name, $3 = tag
function helper_push_image_index() {
    local source_image=${1:-docker://public.ecr.aws/docker/library/busybox:latest}
    local dest_name=${2:-busybox}
    local tag=${3:-latest}
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --format=oci --dest-tls-verify=false --multi-arch=all \
        ${source_image} \
        docker://127.0.0.1:${zot_port}/${dest_name}:${tag}
    [ "$status" -eq 0 ]
    run curl http://127.0.0.1:${zot_port}/v2/_catalog
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg name "${dest_name}" 'any(.repositories[]; . == $name)') = true ]
    run curl http://127.0.0.1:${zot_port}/v2/${dest_name}/tags/list
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg tag "${tag}" 'any(.tags[]; . == $tag)') = true ]
}

# Helper: Pull an image index (multi-arch)
# Args: $1 = image_name, $2 = tag
function helper_pull_image_index() {
    local image_name=${1:-busybox}
    local tag=${2:-latest}
    local oci_data_dir=${BATS_FILE_TMPDIR}/oci
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --src-tls-verify=false --multi-arch=all \
        docker://127.0.0.1:${zot_port}/${image_name}:${tag} \
        oci:${oci_data_dir}/${image_name}:${tag}
    [ "$status" -eq 0 ]
    run cat ${oci_data_dir}/${image_name}/index.json
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg tag "${tag}" '.manifests[].annotations."org.opencontainers.image.ref.name" == $tag') = true ]
    run skopeo --insecure-policy --override-arch=arm64 --override-os=linux copy --src-tls-verify=false --multi-arch=all \
        docker://127.0.0.1:${zot_port}/${image_name}:${tag} \
        oci:${oci_data_dir}/${image_name}:${tag}
    [ "$status" -eq 0 ]
    run cat ${oci_data_dir}/${image_name}/index.json
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq --arg tag "${tag}" '.manifests[].annotations."org.opencontainers.image.ref.name" == $tag') = true ]
}

# Helper: Push an ORAS artifact
# Args: $1 = artifact_name, $2 = tag
function helper_push_oras_artifact() {
    local artifact_name=${1:-hello-artifact}
    local tag=${2:-v2}
    local zot_port=$(get_zot_port)
    echo "{\"name\":\"foo\",\"value\":\"bar\"}" > config.json
    echo "hello world" > artifact.txt
    run oras push --plain-http 127.0.0.1:${zot_port}/${artifact_name}:${tag} \
        --config config.json:application/vnd.acme.rocket.config.v1+json artifact.txt:text/plain -d -v
    [ "$status" -eq 0 ]
    rm -f artifact.txt
    rm -f config.json
}

# Helper: Pull an ORAS artifact
# Args: $1 = artifact_name, $2 = tag
function helper_pull_oras_artifact() {
    local artifact_name=${1:-hello-artifact}
    local tag=${2:-v2}
    local zot_port=$(get_zot_port)
    run oras pull --plain-http 127.0.0.1:${zot_port}/${artifact_name}:${tag} -d -v
    [ "$status" -eq 0 ]
    grep -q "hello world" artifact.txt
    rm -f artifact.txt
}

# Helper: Attach ORAS artifacts (signature and sbom) to an image
# Args: $1 = image_name, $2 = tag
function helper_attach_oras_artifacts() {
    local image_name=${1:-golang}
    local tag=${2:-1.20}
    local zot_port=$(get_zot_port)
    # attach signature
    echo "{\"artifact\": \"\", \"signature\": \"pat hancock\"}" > ${BATS_FILE_TMPDIR}/signature.json
    run oras attach --disable-path-validation --plain-http 127.0.0.1:${zot_port}/${image_name}:${tag} --artifact-type 'signature/example' ${BATS_FILE_TMPDIR}/signature.json:application/json
    [ "$status" -eq 0 ]
    # attach sbom
    echo "{\"version\": \"0.0.0.0\", \"artifact\": \"'127.0.0.1:${zot_port}/${image_name}:${tag}'\", \"contents\": \"good\"}" > ${BATS_FILE_TMPDIR}/sbom.json
    run oras attach --disable-path-validation --plain-http 127.0.0.1:${zot_port}/${image_name}:${tag} --artifact-type 'sbom/example' ${BATS_FILE_TMPDIR}/sbom.json:application/json
    [ "$status" -eq 0 ]
}

# Helper: Discover ORAS artifacts
# Args: $1 = image_name, $2 = tag, $3 = expected_count
function helper_discover_oras_artifacts() {
    local image_name=${1:-golang}
    local tag=${2:-1.20}
    local expected_count=${3:-2}
    local zot_port=$(get_zot_port)
    run oras discover --plain-http --format json 127.0.0.1:${zot_port}/${image_name}:${tag}
    [ "$status" -eq 0 ]
    [ $(echo "$output" | jq -r ".manifests | length") -eq ${expected_count} ]
}

# Helper: Add and list tags using ORAS
function helper_add_and_list_tags_using_oras() {
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --dest-tls-verify=false \
        oci:${TEST_DATA_DIR}/golang:1.20 \
        docker://127.0.0.1:${zot_port}/oras-tags:1.20
    [ "$status" -eq 0 ]
    run oras tag --plain-http 127.0.0.1:${zot_port}/oras-tags:1.20 1 new latest
    [ "$status" -eq 0 ]
    run oras repo tags --plain-http 127.0.0.1:${zot_port}/oras-tags
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 4 ]
    [ "${lines[-1]}" == "new" ]
    [ "${lines[-2]}" == "latest" ]
    [ "${lines[-3]}" == "1.20" ]
    [ "${lines[-4]}" == "1" ]
    run oras repo tags --plain-http --last new 127.0.0.1:${zot_port}/oras-tags
    [ "$status" -eq 0 ]
    echo "$output"
    [ -z "$output" ]
    run oras repo tags --plain-http --last latest 127.0.0.1:${zot_port}/oras-tags
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 1 ]
    [ "${lines[-1]}" == "new" ]
    run oras repo tags --plain-http --last "1.20" 127.0.0.1:${zot_port}/oras-tags
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 2 ]
    [ "${lines[-2]}" == "latest" ]
    [ "${lines[-1]}" == "new" ]
    run oras repo tags --plain-http --last "1" 127.0.0.1:${zot_port}/oras-tags
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 3 ]
    [ "${lines[-3]}" == "1.20" ]
    [ "${lines[-2]}" == "latest" ]
    [ "${lines[-1]}" == "new" ]
}

# Helper: Push a Helm chart
function helper_push_helm_chart() {
    local zot_port=$(get_zot_port)
    run helm package ${BATS_FILE_TMPDIR}/helm-charts/charts/zot -d ${BATS_FILE_TMPDIR}
    [ "$status" -eq 0 ]
    local chart_version=$(awk '/version/{printf $2}' ${BATS_FILE_TMPDIR}/helm-charts/charts/zot/Chart.yaml)
    run helm push ${BATS_FILE_TMPDIR}/zot-${chart_version}.tgz oci://localhost:${zot_port}/zot-chart
    [ "$status" -eq 0 ]
}

# Helper: Pull a Helm chart
function helper_pull_helm_chart() {
    local zot_port=$(get_zot_port)
    local chart_version=$(awk '/version/{printf $2}' ${BATS_FILE_TMPDIR}/helm-charts/charts/zot/Chart.yaml)
    run helm pull oci://localhost:${zot_port}/zot-chart/zot --version ${chart_version} -d ${BATS_FILE_TMPDIR}
    [ "$status" -eq 0 ]
}

# Helper: Push image with regclient
function helper_push_image_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl registry set localhost:${zot_port} --tls disabled
    [ "$status" -eq 0 ]
    run regctl image copy ocidir://${TEST_DATA_DIR}/golang:1.20 localhost:${zot_port}/test-regclient
    [ "$status" -eq 0 ]
}

# Helper: Pull image with regclient
function helper_pull_image_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl image copy localhost:${zot_port}/test-regclient ocidir://${TEST_DATA_DIR}/golang:1.20
    [ "$status" -eq 0 ]
}

# Helper: List repositories with regclient
# Args: $1 = limit (optional), $2 = expected_first_repos (optional, space separated)
function helper_list_repositories_with_regclient() {
    local limit=${1:-2}
    local first_repo=${2:-busybox}
    local second_repo=${3:-golang}
    local zot_port=$(get_zot_port)
    run regctl repo ls localhost:${zot_port}
    [ "$status" -eq 0 ]

    found=0
    for i in "${lines[@]}"
    do
        if [ "$i" = 'test-regclient' ]; then
            found=1
        fi
    done
    [ "$found" -eq 1 ]

    run regctl repo ls --limit ${limit} localhost:${zot_port}
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq ${limit} ]
}

# Helper: List image tags with regclient
function helper_list_image_tags_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl tag ls localhost:${zot_port}/test-regclient
    [ "$status" -eq 0 ]

    found=0
    for i in "${lines[@]}"
    do
        if [ "$i" = 'latest' ]; then
            found=1
        fi
    done
    [ "$found" -eq 1 ]
}

# Helper: Push manifest with regclient
function helper_push_manifest_with_regclient() {
    local zot_port=$(get_zot_port)
    manifest=$(regctl manifest get localhost:${zot_port}/test-regclient --format=raw-body)
    run regctl manifest put localhost:${zot_port}/test-regclient:1.0.0 --format oci --content-type application/vnd.oci.image.manifest.v1+json --format oci <<JSON
    $manifest
JSON
    [ "$status" -eq 0 ]
}

# Helper: Pull manifest with regclient
function helper_pull_manifest_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl manifest get localhost:${zot_port}/test-regclient
    [ "$status" -eq 0 ]
}

# Helper: Pull manifest with docker client
function helper_pull_manifest_with_docker_client() {
    local zot_port=$(get_zot_port)
    run docker pull localhost:${zot_port}/test-regclient
    [ "$status" -eq 0 ]
}

# Helper: Pull manifest with crictl
function helper_pull_manifest_with_crictl() {
    local zot_port=$(get_zot_port)
    run crictl pull localhost:${zot_port}/test-regclient
    [ "$status" -eq 0 ]
}

# Helper: Push OCI artifact with regclient
function helper_push_oci_artifact_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl artifact put localhost:${zot_port}/artifact:demo <<TXT
this is an artifact
TXT
    [ "$status" -eq 0 ]
}

# Helper: Pull OCI artifact with regclient
function helper_pull_oci_artifact_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl manifest get localhost:${zot_port}/artifact:demo
    [ "$status" -eq 0 ]
    run regctl artifact get localhost:${zot_port}/artifact:demo
    [ "$status" -eq 0 ]
    [ "${lines[-1]}" == "this is an artifact" ]
}

# Helper: Push OCI artifact references with regclient
# Args: $1 = expected_initial_count (0 for release, 1 for new)
function helper_push_oci_artifact_references_with_regclient() {
    local expected_initial_count=${1:-0}
    local zot_port=$(get_zot_port)
    run regctl artifact put localhost:${zot_port}/manifest-ref:demo <<TXT
test artifact
TXT
    [ "$status" -eq 0 ]
    run regctl artifact list localhost:${zot_port}/manifest-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq ${expected_initial_count} ]
    run regctl artifact put --annotation  demo=true --annotation format=oci --artifact-type "application/vnd.example.icecream.v1" --subject localhost:${zot_port}/manifest-ref:demo << TXT
test reference
TXT
    [ "$status" -eq 0 ]
    # with artifact media-type
    run regctl artifact put localhost:${zot_port}/artifact-ref:demo <<TXT
test artifact
TXT
    [ "$status" -eq 0 ]
    run regctl artifact list localhost:${zot_port}/artifact-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq ${expected_initial_count} ]
    run regctl artifact put --annotation  demo=true --annotation format=oci --artifact-type "application/vnd.example.icecream.v1" --subject localhost:${zot_port}/artifact-ref:demo << TXT
test reference
TXT
    [ "$status" -eq 0 ]
}

# Helper: Pull OCI artifact references with regclient
# Args: $1 = expected_count (1 for release, 1 for new - same after adding reference)
function helper_pull_oci_artifact_references_with_regclient() {
    local expected_count=${1:-1}
    local zot_port=$(get_zot_port)
    run regctl artifact list localhost:${zot_port}/manifest-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq ${expected_count} ]
    run regctl artifact list --filter-artifact-type "application/vnd.example.icecream.v1" localhost:${zot_port}/manifest-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq ${expected_count} ]
    run regctl artifact list --filter-artifact-type "application/invalid" localhost:${zot_port}/manifest-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq 0 ]
    # with artifact media-type
    run regctl artifact list localhost:${zot_port}/artifact-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq ${expected_count} ]
    run regctl artifact list --filter-artifact-type "application/vnd.example.icecream.v1" localhost:${zot_port}/artifact-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq ${expected_count} ]
    run regctl artifact list --filter-artifact-type "application/invalid" localhost:${zot_port}/artifact-ref:demo --format raw-body
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests | length') -eq 0 ]
}

# Helper: Push docker image
function helper_push_docker_image() {
    local zot_port=$(get_zot_port)
    cat > Dockerfile <<DOCKERFILE
    FROM ghcr.io/project-zot/test-images/busybox-docker:1.37
    RUN echo "hello world" > /testfile
DOCKERFILE
    docker build -f Dockerfile . -t localhost:${zot_port}/test
    run docker push localhost:${zot_port}/test
    [ "$status" -eq 1 ]
    run docker pull localhost:${zot_port}/test
    [ "$status" -eq 1 ]
}

# ==============================================================================
# RELEASE TEST FUNCTIONS
# These functions are used to test the released version of zot before upgrade
# ==============================================================================

function test_release_push_image() {
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --dest-tls-verify=false \
        oci:${TEST_DATA_DIR}/golang:1.20 \
        docker://127.0.0.1:${zot_port}/golang:1.20
    [ "$status" -eq 0 ]
    run curl http://127.0.0.1:${zot_port}/v2/_catalog
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.repositories[]') = '"golang"' ]
    run curl http://127.0.0.1:${zot_port}/v2/golang/tags/list
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.tags[]') = '"1.20"' ]
}

function test_release_pull_image() {
    helper_pull_image golang 1.20
}

function test_release_push_image_index() {
    helper_push_image_index docker://public.ecr.aws/docker/library/busybox:latest busybox latest
}

function test_release_pull_image_index() {
    helper_pull_image_index busybox latest
}

function test_release_push_oras_artifact() {
    helper_push_oras_artifact hello-artifact v2
}

function test_release_pull_oras_artifact() {
    helper_pull_oras_artifact hello-artifact v2
}

function test_release_attach_oras_artifacts() {
    helper_attach_oras_artifacts golang 1.20
}

function test_release_discover_oras_artifacts() {
    helper_discover_oras_artifacts golang 1.20 2
}

function test_release_add_and_list_tags_using_oras() {
    helper_add_and_list_tags_using_oras
}

function test_release_push_helm_chart() {
    helper_push_helm_chart
}

function test_release_pull_helm_chart() {
    helper_pull_helm_chart
}

function test_release_push_image_with_regclient() {
    helper_push_image_with_regclient
}

function test_release_pull_image_with_regclient() {
    helper_pull_image_with_regclient
}

function test_release_list_repositories_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl repo ls localhost:${zot_port}
    [ "$status" -eq 0 ]

    found=0
    for i in "${lines[@]}"
    do
        if [ "$i" = 'test-regclient' ]; then
            found=1
        fi
    done
    [ "$found" -eq 1 ]

    run regctl repo ls --limit 2 localhost:${zot_port}
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 2 ]
    [ "${lines[-2]}" == "busybox" ]
    [ "${lines[-1]}" == "golang" ]

    run regctl repo ls --last busybox --limit 1 localhost:${zot_port}
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 1 ]
    [ "${lines[-1]}" == "golang" ]
}

function test_release_list_image_tags_with_regclient() {
    helper_list_image_tags_with_regclient
}

function test_release_push_manifest_with_regclient() {
    helper_push_manifest_with_regclient
}

function test_release_pull_manifest_with_regclient() {
    helper_pull_manifest_with_regclient
}

function test_release_pull_manifest_with_docker_client() {
    helper_pull_manifest_with_docker_client
}

function test_release_pull_manifest_with_crictl() {
    helper_pull_manifest_with_crictl
}

function test_release_push_oci_artifact_with_regclient() {
    helper_push_oci_artifact_with_regclient
}

function test_release_pull_oci_artifact_with_regclient() {
    helper_pull_oci_artifact_with_regclient
}

function test_release_push_oci_artifact_references_with_regclient() {
    helper_push_oci_artifact_references_with_regclient 0
}

function test_release_pull_oci_artifact_references_with_regclient() {
    helper_pull_oci_artifact_references_with_regclient 1
}

function test_release_push_docker_image() {
    helper_push_docker_image
}

# ==============================================================================
# NEW (POST-UPGRADE) TEST FUNCTIONS
# These functions are used to test the new version of zot after upgrade
# ==============================================================================

function test_new_existing_pull_image() {
    helper_pull_image golang 1.20
}

function test_new_existing_pull_image_index() {
    helper_pull_image_index busybox latest
}

function test_new_existing_pull_oras_artifact() {
    helper_pull_oras_artifact hello-artifact v2
}

function test_new_push_image() {
    local zot_port=$(get_zot_port)
    # first check existing images
    run curl http://127.0.0.1:${zot_port}/v2/_catalog
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq 'any(.repositories[]; . == "golang")') = true ] 
    run skopeo --insecure-policy copy --dest-tls-verify=false \
        oci:${TEST_DATA_DIR}/golang:1.20 \
        docker://127.0.0.1:${zot_port}/golang:1.20
    [ "$status" -eq 0 ]
    run skopeo --insecure-policy copy --dest-tls-verify=false \
        docker://ghcr.io/project-zot/test-images/alpine:3.17.3 \
        docker://127.0.0.1:${zot_port}/alpine:3.17.3
    [ "$status" -eq 0 ]
    run curl http://127.0.0.1:${zot_port}/v2/_catalog
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq 'any(.repositories[]; . == "golang")') = true ] 
    [ $(echo "${lines[-1]}" | jq 'any(.repositories[]; . == "alpine")') = true ] 
    run curl http://127.0.0.1:${zot_port}/v2/golang/tags/list
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.tags[]') = '"1.20"' ]
}

function test_new_pull_image() {
    helper_pull_image golang 1.20
}

function test_new_push_image_index() {
    local zot_port=$(get_zot_port)
    # --multi-arch below pushes an image index (containing many images) instead
    # of an image manifest (single image)
    run skopeo --insecure-policy copy --format=oci --dest-tls-verify=false --multi-arch=all \
        docker://public.ecr.aws/docker/library/busybox:latest \
        docker://127.0.0.1:${zot_port}/busybox:latest
    [ "$status" -eq 0 ]
    run curl http://127.0.0.1:${zot_port}/v2/_catalog
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq 'any(.repositories[]; . == "busybox")') = true ] 
    run curl http://127.0.0.1:${zot_port}/v2/busybox/tags/list
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.tags[]') = '"latest"' ]
}

function test_new_pull_image_index() {
    local oci_data_dir=${BATS_FILE_TMPDIR}/oci
    local zot_port=$(get_zot_port)
    run skopeo --insecure-policy copy --src-tls-verify=false --multi-arch=all \
        docker://127.0.0.1:${zot_port}/busybox:latest \
        oci:${oci_data_dir}/busybox:latest
    [ "$status" -eq 0 ]
    run cat ${BATS_FILE_TMPDIR}/oci/busybox/index.json
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests[].annotations."org.opencontainers.image.ref.name"') = '"latest"' ]
    run skopeo --insecure-policy --override-arch=arm64 --override-os=linux copy --src-tls-verify=false --multi-arch=all \
        docker://127.0.0.1:${zot_port}/busybox:latest \
        oci:${oci_data_dir}/busybox:latest
    [ "$status" -eq 0 ]
    run cat ${BATS_FILE_TMPDIR}/oci/busybox/index.json
    [ "$status" -eq 0 ]
    [ $(echo "${lines[-1]}" | jq '.manifests[].annotations."org.opencontainers.image.ref.name"') = '"latest"' ]
    run curl -X DELETE http://127.0.0.1:${zot_port}/v2/busybox/manifests/latest
    [ "$status" -eq 0 ]
}

function test_new_push_oras_artifact() {
    helper_push_oras_artifact hello-artifact v2
}

function test_new_pull_oras_artifact() {
    helper_pull_oras_artifact hello-artifact v2
}

function test_new_attach_oras_artifacts() {
    helper_attach_oras_artifacts golang 1.20
}

# Note: expected_count parameter allows different counts for full vs minimal zot
function test_new_discover_oras_artifacts() {
    local expected_count=${1:-4}
    helper_discover_oras_artifacts golang 1.20 ${expected_count}
}

function test_new_add_and_list_tags_using_oras() {
    helper_add_and_list_tags_using_oras
}

function test_new_push_helm_chart() {
    helper_push_helm_chart
}

function test_new_pull_helm_chart() {
    helper_pull_helm_chart
}

function test_new_push_image_with_regclient() {
    helper_push_image_with_regclient
}

function test_new_pull_image_with_regclient() {
    helper_pull_image_with_regclient
}

function test_new_list_repositories_with_regclient() {
    local zot_port=$(get_zot_port)
    run regctl repo ls localhost:${zot_port}
    [ "$status" -eq 0 ]

    found=0
    for i in "${lines[@]}"
    do
        if [ "$i" = 'test-regclient' ]; then
            found=1
        fi
    done
    [ "$found" -eq 1 ]

    run regctl repo ls --limit 4 localhost:${zot_port}
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 4 ]
    [ "${lines[0]}" == "alpine" ]
    [ "${lines[-1]}" == "busybox" ]

    run regctl repo ls --last busybox --limit 1 localhost:${zot_port}
    [ "$status" -eq 0 ]
    echo "$output"
    [ $(echo "$output" | wc -l) -eq 1 ]
    [ "${lines[-1]}" == "golang" ]
}

function test_new_list_image_tags_with_regclient() {
    helper_list_image_tags_with_regclient
}

function test_new_push_manifest_with_regclient() {
    helper_push_manifest_with_regclient
}

function test_new_pull_manifest_with_regclient() {
    helper_pull_manifest_with_regclient
}

function test_new_pull_manifest_with_docker_client() {
    helper_pull_manifest_with_docker_client
}

function test_new_pull_manifest_with_crictl() {
    helper_pull_manifest_with_crictl
}

function test_new_push_oci_artifact_with_regclient() {
    helper_push_oci_artifact_with_regclient
}

function test_new_pull_oci_artifact_with_regclient() {
    helper_pull_oci_artifact_with_regclient
}

function test_new_push_oci_artifact_references_with_regclient() {
    helper_push_oci_artifact_references_with_regclient 1
}

function test_new_pull_oci_artifact_references_with_regclient() {
    helper_pull_oci_artifact_references_with_regclient 1
}

function test_new_push_docker_image() {
    helper_push_docker_image
}