from typing import Type

from tools.docgen.docgen import DocGen
from tools.docgen.entry_variables import EntryVariableDocGen
from tools.docgen.scripting_functions import ScriptingFunctionsDocGen
from ytdl_sub.utils.file_handler import get_file_md5_hash
from ytdl_sub.utils.file_handler import get_md5_hash


def _test_doc_gen(doc_gen: Type[DocGen]) -> None:
    expected_md5_hash = get_md5_hash(doc_gen.generate_and_maybe_write_to_file())
    md5_hash = get_file_md5_hash(doc_gen.LOCATION)

    assert md5_hash == expected_md5_hash


class TestDocGen:
    def test_entry_variables_generated(self):
        _test_doc_gen(EntryVariableDocGen)

    def test_scripting_functions_generated(self):
        _test_doc_gen(ScriptingFunctionsDocGen)