import importlib
import os
import sys

import pytest

import manim_slides.slide as slide


def assert_import(
    *,
    api_name: str,
    manim: bool,
    manimgl: bool,
) -> None:
    importlib.reload(slide)

    assert slide.API_NAME == api_name
    assert slide.MANIM == manim
    assert slide.MANIMGL == manimgl


def test_force_api() -> None:
    import manim  # noqa: F401

    if "manimlib" in sys.modules:
        del sys.modules["manimlib"]

    os.environ[slide.MANIM_API] = "manimlib"
    os.environ[slide.FORCE_MANIM_API] = "1"

    assert_import(
        api_name="manimlib",
        manim=False,
        manimgl=True,
    )

    del os.environ[slide.MANIM_API]
    del os.environ[slide.FORCE_MANIM_API]


def test_invalid_api() -> None:
    os.environ[slide.MANIM_API] = "manim_slides"

    with pytest.raises(ImportError):
        assert_import(
            api_name="",
            manim=False,
            manimgl=False,
        )

    del os.environ[slide.MANIM_API]


@pytest.mark.filterwarnings("ignore:assert_import")
def test_manim_and_manimgl_imported() -> None:
    import manim  # noqa: F401
    import manimlib  # noqa: F401

    assert_import(
        api_name="manim",
        manim=True,
        manimgl=False,
    )


def test_manim_imported() -> None:
    import manim  # noqa: F401

    if "manimlib" in sys.modules:
        del sys.modules["manimlib"]

    assert_import(
        api_name="manim",
        manim=True,
        manimgl=False,
    )


def test_manimgl_imported() -> None:
    import manimlib  # noqa: F401

    if "manim" in sys.modules:
        del sys.modules["manim"]

    assert_import(
        api_name="manimlib",
        manim=False,
        manimgl=True,
    )


def test_nothing_imported() -> None:
    if "manim" in sys.modules:
        del sys.modules["manim"]

    if "manimlib" in sys.modules:
        del sys.modules["manimlib"]

    assert_import(
        api_name="manim",
        manim=True,
        manimgl=False,
    )