import inspect

import pytest

from helpers import GetNextId
from server.authentication import User

DEFAULT_ONTOLOGY_NAME = "current_test_"


@pytest.fixture(scope="session")
def sheerka():
    from core.Sheerka import Sheerka

    sheerka = Sheerka()
    sheerka.initialize("mem://")
    return sheerka


@pytest.fixture(scope="module", autouse=True)
def on_new_module(sheerka, request):
    """
    For each new module, make sure to create a new ontology
    Remove it at the end of the module
    :param sheerka:
    :type sheerka:
    :param request:
    :type request:
    :return:
    :rtype:
    """
    from core.Event import Event
    from core.ExecutionContext import ExecutionContext, ContextActions
    module_name = request.module.__name__.split(".")[-1]
    context = ExecutionContext("test",
                               Event(message=f"Executing module {module_name}"),
                               sheerka,
                               ContextActions.TESTING,
                               None)

    ontology = sheerka.om.push_ontology(module_name)
    yield
    sheerka.om.revert_ontology(context, ontology)


@pytest.fixture(scope="function")
def context(sheerka):
    from core.Event import Event
    from core.ExecutionContext import ExecutionContext, ContextActions

    return ExecutionContext("test",
                            Event(message=""),
                            sheerka,
                            ContextActions.TESTING,
                            None)


@pytest.fixture()
def next_id():
    return GetNextId()


@pytest.fixture()
def user():
    return User(username="johan doe", email="johan.doe@sheerka.com", firstname="johan", lastname="doe")


class TestUsingFileBasedSheerka:
    @pytest.fixture(scope="class")
    def sheerka(self):
        sheerka = Sheerka()
        sheerka.initialize()
        return sheerka


class NewOntology:
    """
    For some test who may need to declare the same concepts across the tests
    """
    from core.ExecutionContext import ExecutionContext

    def __init__(self, context: ExecutionContext, name=None):
        self.sheerka = context.sheerka
        self.context = context
        self.name = name
        self.ontology = None

        if self.name is None:
            self.name = inspect.stack()[1][3]

    def __enter__(self):
        self.ontology = self.sheerka.om.push_ontology(self.name)
        return self.ontology

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.sheerka.om.revert_ontology(self.context, self.ontology)
        return False