import unittest
from os.path import abspath, dirname, join

from rdf_utils.caching import read_file_and_cache
from rdf_utils.resolver import install_resolver
from rdflib import Dataset

CWD = abspath(dirname(__file__))
ROOT_DIR = dirname(CWD)
EXEC_RQ = join(ROOT_DIR, "queries", "scenario-execution.rq")
SPEC_MODEL_URLS = {
    join(ROOT_DIR, "templates", "pickplace.tmpl.json"): "json-ld",
    join(ROOT_DIR, "variations", "pickplace-secorolab-isaac.var.json"): "json-ld",
}
EXEC_MODEL_URLS = {
    join(ROOT_DIR, "execution", "pickplace-secorolab-ros.obs.json"): "json-ld",
}


class QueriesTest(unittest.TestCase):
    def setUp(self):
        install_resolver()

    def test_scenario_execution_query(self):
        graph = Dataset()
        for url, fmt in {**SPEC_MODEL_URLS, **EXEC_MODEL_URLS}.items():
            graph.parse(url, format=fmt)

        exec_rq_str = read_file_and_cache(EXEC_RQ)
        q_result = graph.query(exec_rq_str)
        assert q_result.type == "CONSTRUCT" and q_result.graph is not None, (
            "ScenarioExecution query did not return a graph"
        )
        assert len(q_result.graph) > 0, "ScenarioExecution query returns an empty graph"