test_script_discovery.py

#!/usr/bin/env python3
 
 
 
import sys
import os
from unittest.mock import Mock
import tempfile
import validationfunctions
 
from validationtestutil import create_fake_scripts
 
 
def main():
    """
    Tests the automated validation script discovery
    """
 
    # will create a temporary folder and delete it once this block is left
    with tempfile.TemporaryDirectory() as tmpdir:
        print(f"Created temporary test folder {tmpdir}")
 
        # create a couple of fake validation scripts
        create_fake_scripts(
            os.path.join(tmpdir, "pkg1", "validation"), "pkg1_a.py"
        )
        create_fake_scripts(
            os.path.join(tmpdir, "pkg1", "validation"), "pkg1_b.py"
        )
        create_fake_scripts(
            os.path.join(tmpdir, "pkg1", "validation"), "pkg1_c.py"
        )
 
        create_fake_scripts(
            os.path.join(tmpdir, "pkg2", "validation"), "pkg2_b.py"
        )
        create_fake_scripts(
            os.path.join(tmpdir, "pkg2", "validation"), "pkg2_c.py"
        )
 
        # this should not be found !
        create_fake_scripts(
            os.path.join(tmpdir, "some_other_folder", "pkgother", "validation"),
            "pkgother_b.py",
        )
 
        # fake the basepath
        basepath = {"local": str(tmpdir)}
 
        folders = validationfunctions.get_validation_folders(
            location="local", basepaths=basepath, log=Mock()
        )
 
        if (
            "pkg1" not in folders
            or "pkg2" not in folders
            or "some_other_folder" in folders
        ):
            print("scripts were discovered in the wrong folders")
            sys.exit(1)
 
        scripts = validationfunctions.scripts_in_dir(
            folders["pkg1"], Mock(), ".py"
        )
 
        if (
            len([s for s in scripts if s.endswith("pkg1/validation/pkg1_a.py")])
            == 0
        ):
            print("script file was not discovered")
            sys.exit(1)
 
        print(scripts)
 
    # test for release and local case
    with tempfile.TemporaryDirectory() as tmpdir:
        print(f"Created temporary test folder {tmpdir}")
 
        # create a couple of fake validation scripts
        create_fake_scripts(
            os.path.join(tmpdir, "pkg1", "validation"), "pkg1_a.py"
        )
        create_fake_scripts(
            os.path.join(tmpdir, "pkg1", "validation"), "pkg1_b.py"
        )
        create_fake_scripts(
            os.path.join(tmpdir, "pkg1", "validation"), "pkg1_c.py"
        )
 
        create_fake_scripts(
            os.path.join(tmpdir, "pkg2", "validation"), "pkg2_b.py"
        )
        create_fake_scripts(
            os.path.join(tmpdir, "pkg2", "validation"), "pkg2_c.py"
        )
 
        # this should not be found !
        create_fake_scripts(
            os.path.join(tmpdir, "some_other_folder", "pkgother", "validation"),
            "pkgother_b.py",
        )
 
        # fake the basepath
        basepath = {"local": str(tmpdir), "central": str(tmpdir)}
 
        folders = validationfunctions.get_validation_folders(
            location="central", basepaths=basepath, log=Mock()
        )
 
        if (
            "pkg1" not in folders
            or "pkg2" not in folders
            or "some_other_folder" in folders
        ):
            print("scripts were discovered in the wrong folders")
            sys.exit(1)
 
 
if __name__ == "__main__":
    main()