from dataprofiler.profilers.profiler_options import BooleanOption
from dataprofiler.tests.profilers.profiler_options.test_base_option import (
    TestBaseOption,
)


class TestBooleanOption(TestBaseOption):

    option_class = BooleanOption
    keys = []

    def test_init(self):
        option = self.get_options()
        self.assertDictEqual({"is_enabled": True}, option.properties)
        option = self.get_options(is_enabled=False)
        self.assertDictEqual({"is_enabled": False}, option.properties)

    def test_set_helper(self):
        super().test_set_helper()

        # Enable and Disable Option
        for is_enabled in [True, False]:
            option = self.get_options(is_enabled=not is_enabled)
            option._set_helper({"is_enabled": is_enabled}, "")
            self.assertEqual(is_enabled, option.properties["is_enabled"])

        # Treat is_enabled as a BooleanOption
        expected_error = "type object 'is_enabled' has no attribute " "'is_enabled'"
        with self.assertRaisesRegex(AttributeError, expected_error):
            option._set_helper({"is_enabled.is_enabled": True}, "")

    def test_set(self):
        super().test_set()

        # Enable and Disable Options
        for is_enabled in [True, False]:
            option = self.get_options(is_enabled=not is_enabled)
            option.set({"is_enabled": is_enabled})
            self.assertEqual(is_enabled, option.properties["is_enabled"])

        # Treat is_enabled as a BooleanOption
        expected_error = "type object 'is_enabled' has no attribute " "'is_enabled'"
        with self.assertRaisesRegex(AttributeError, expected_error):
            option.set({"is_enabled.is_enabled": True})

    def test_validate_helper(self):
        option = self.get_options(is_enabled=True)
        optpth = self.get_options_path()

        # Default Configuration Is Valid
        self.assertEqual([], option._validate_helper())

        # Variable Path Is Not A String
        expected_error = "The variable path must be a string."
        with self.assertRaisesRegex(ValueError, expected_error):
            option._validate_helper(1)

        # Option is_enabled is not a boolean
        option = self.get_options(is_enabled="Hello World")
        expected_error = [f"{optpth}.is_enabled must be a Boolean."]
        self.assertSetEqual(set(expected_error), set(option._validate_helper()))

    def test_validate(self):
        option = self.get_options(is_enabled=True)
        optpth = self.get_options_path()

        # Default Configuration Is Valid
        self.assertEqual(None, option.validate())

        # Option is_enabled is not a boolean
        option = self.get_options(is_enabled="Hello World")
        expected_error = f"{optpth}.is_enabled must be a Boolean."
        with self.assertRaisesRegex(ValueError, expected_error):
            option.validate(raise_error=True)

        expected_error = [expected_error]
        self.assertSetEqual(
            set(expected_error), set(option.validate(raise_error=False))
        )

    def test_eq(self):
        super().test_eq()

        options = self.get_options()
        options2 = self.get_options()
        options.is_enabled = False
        self.assertNotEqual(options, options2)
        options2.is_enabled = False
        self.assertEqual(options, options2)
