import os

from peakrdl_busdecoder.cpuif.apb3 import APB3_Cpuif
from ..lib.cpuifs.apb3 import APB3
from ..lib.base_testcase import BaseTestCase


# -------------------------------------------------------------------------------


class ClassOverride_Cpuif(APB3_Cpuif):
    @property
    def port_declaration(self) -> str:
        return "user_apb3_intf.slave s_apb"


class ClassOverride_cpuiftestmode(APB3):
    cpuif_cls = ClassOverride_Cpuif


class Test_class_override(BaseTestCase):
    cpuif = ClassOverride_cpuiftestmode()

    def test_override_success(self):
        output_file = os.path.join(self.get_run_dir(), "busdecoder.sv")
        with open(output_file, "r") as f:
            self.assertIn("user_apb3_intf.slave s_apb", f.read())


# -------------------------------------------------------------------------------


class TemplateOverride_Cpuif(APB3_Cpuif):
    # contains the text "USER TEMPLATE OVERRIDE"
    template_path = "user_apb3_tmpl.sv"


class TemplateOverride_cpuiftestmode(APB3):
    cpuif_cls = TemplateOverride_Cpuif


class Test_template_override(BaseTestCase):
    cpuif = TemplateOverride_cpuiftestmode()

    def test_override_success(self):
        output_file = os.path.join(self.get_run_dir(), "busdecoder.sv")
        with open(output_file, "r") as f:
            self.assertIn("USER TEMPLATE OVERRIDE", f.read())