test_folder_creation.py

#!/usr/bin/env python3
 
 
 
import os
import shutil
import tempfile
import validation
import validationpath
import validationserver
import validationplots
 
from validationtestutil import check_path_exists
 
 
def main():
    """
    Runs two test validations, starts the web server and queries data
    """
 
    rev_to_gen = "test_folder_creation"
    all_tags = ["reference", rev_to_gen]
 
    # 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}")
 
        expect_html_plots_comparison_json = validationpath.get_html_plots_tag_comparison_json(
            str(tmpdir), all_tags
        )
        expect_html_plots_comparison_folder = validationpath.get_html_plots_tag_comparison_folder(
            str(tmpdir), all_tags
        )
 
        # switch to this folder
        os.chdir(str(tmpdir))
 
        validation.execute(tag=rev_to_gen, is_test=True)
 
        # todo: check if results folder has been created and is filled
        path_to_check = [
            validationpath.get_results_tag_folder(str(tmpdir), rev_to_gen),
            validationpath.get_results_tag_revision_file(
                str(tmpdir), rev_to_gen
            ),
            os.path.join(
                validationpath.get_results_tag_folder(str(tmpdir), rev_to_gen),
                "validation-test",
                "validationTestPlots.py.log",
            ),
            os.path.join(
                validationpath.get_results_tag_folder(str(tmpdir), rev_to_gen),
                "validation-test",
                "validationTestPlots.root",
            ),
            expect_html_plots_comparison_json,
            expect_html_plots_comparison_folder,
        ]
 
        check_path_exists(path_to_check)
 
        # run validation server
        # must setup all content in the html folder required
        validationserver.run_server(dry_run=True)
 
        # check if all files have been copied and the symbolic links properly
        #  set
        path_to_check = [
            os.path.join(str(tmpdir), validationpath.folder_name_html),
            os.path.join(
                str(tmpdir),
                validationpath.folder_name_html,
                validationpath.folder_name_plots,
            ),
        ]
        check_path_exists(path_to_check)
 
        # remove generated plots and use generate_new_plots function to
        # regenerate
        shutil.rmtree(expect_html_plots_comparison_folder)
 
        # recreate
        # switch to this folder
        os.chdir(str(tmpdir))
        validationplots.generate_new_plots(all_tags, tmpdir)
        check_path_exists(path_to_check)
 
 
if __name__ == "__main__":
    main()