test_variable_substitutions.py

#!/bin/python
# Tests for variable substitutions
# A. Schlemmer, 05/2022

from newcrawler import Crawler
from newcrawler.structure_elements import File, DictTextElement, DictListElement
from newcrawler.identifiable_adapters import IdentifiableAdapter, LocalStorageIdentifiableAdapter
from functools import partial
from copy import deepcopy
from unittest.mock import MagicMock, Mock
from os.path import join, dirname, basename
import yaml
import caosdb as db
from caosdb.apiutils import compare_entities

import pytest
from pytest import raises


def rfp(*pathcomponents):
    """
    Return full path.
    Shorthand convenience function.
    """
    return join(dirname(__file__), *pathcomponents)

def dircheckstr(element_type, *pathcomponents):
    """
    Return the debug tree identifier for a given path.
    """
    return "newcrawler.structure_elements." + element_type + ": " + basename(join(*pathcomponents)) + ", " + rfp("test_directories", "example_substitutions", *pathcomponents)


@pytest.fixture
def crawler():
    crawler = Crawler(debug=True)
    crawler.crawl_directory(rfp("test_directories", "example_substitutions", "ExperimentalData"),
                            rfp("test_directories", "example_substitutions", "substitutions.yml"))
    return crawler


def test_substitutions(crawler):
    for i in range(2):
        subd = crawler.debug_tree[dircheckstr("File", "ExperimentalData", "220512_data.dat")]
        assert subd[i]["Experiment"].get_property("date").value == "2022-05-12"
        assert isinstance(subd[i]["ExperimentSeries"].get_property("Experiment").value, db.Record)

        subd = crawler.debug_tree[dircheckstr("Directory", "ExperimentalData")]
        assert subd[i]["Project"].name == "project"
        assert isinstance(subd[i]["Project"].get_property("Experiments").value, list)
        assert isinstance(subd[i]["Project"].get_property("Experiments").value[0], db.Record)

        assert isinstance(subd[i]["Project"].get_property("dates").value, list)
        assert subd[i]["Project"].get_property("dates").value[0] == "2022-05-12"