# ------------------------------------------------------------------------------
#  Copyright 2022 Upstream Data Inc                                            -
#                                                                              -
#  Licensed under the Apache License, Version 2.0 (the "License");             -
#  you may not use this file except in compliance with the License.            -
#  You may obtain a copy of the License at                                     -
#                                                                              -
#      http://www.apache.org/licenses/LICENSE-2.0                              -
#                                                                              -
#  Unless required by applicable law or agreed to in writing, software         -
#  distributed under the License is distributed on an "AS IS" BASIS,           -
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.    -
#  See the License for the specific language governing permissions and         -
#  limitations under the License.                                              -
# ------------------------------------------------------------------------------
import inspect
import unittest
import warnings
from dataclasses import asdict

from pyasic.miners.miner_factory import MINER_CLASSES


class MinersTest(unittest.TestCase):
    def test_miner_type_creation(self):
        warnings.filterwarnings("ignore")
        for miner_type in MINER_CLASSES.keys():
            for miner_model in MINER_CLASSES[miner_type].keys():
                with self.subTest(
                    msg=f"Test creation of miner",
                    miner_type=miner_type,
                    miner_model=miner_model,
                ):
                    miner = MINER_CLASSES[miner_type][miner_model]("127.0.0.1")
                    self.assertTrue(
                        isinstance(miner, MINER_CLASSES[miner_type][miner_model])
                    )

    def test_miner_data_map_keys(self):
        keys = sorted(
            [
                "api_ver",
                "config",
                "env_temp",
                "errors",
                "fan_psu",
                "fans",
                "fault_light",
                "fw_ver",
                "hashboards",
                "hashrate",
                "hostname",
                "is_mining",
                "mac",
                "expected_hashrate",
                "uptime",
                "wattage",
                "wattage_limit",
            ]
        )
        warnings.filterwarnings("ignore")
        for miner_type in MINER_CLASSES.keys():
            for miner_model in MINER_CLASSES[miner_type].keys():
                with self.subTest(
                    msg=f"Data map key check",
                    miner_type=miner_type,
                    miner_model=miner_model,
                ):
                    miner = MINER_CLASSES[miner_type][miner_model]("127.0.0.1")
                    miner_keys = sorted(
                        [str(k) for k in asdict(miner.data_locations).keys()]
                    )
                    self.assertEqual(miner_keys, keys)

    def test_data_locations_match_signatures_command(self):
        warnings.filterwarnings("ignore")
        for miner_type in MINER_CLASSES.keys():
            for miner_model in MINER_CLASSES[miner_type].keys():
                miner = MINER_CLASSES[miner_type][miner_model]("127.0.0.1")
                for data_point in asdict(miner.data_locations).values():
                    with self.subTest(
                        msg=f"Test {data_point['cmd']} signature matches",
                        miner_type=miner_type,
                        miner_model=miner_model,
                    ):
                        func = getattr(miner, data_point["cmd"])
                        signature = inspect.signature(func)
                        parameters = signature.parameters
                        param_names = list(parameters.keys())
                        for arg in ["kwargs", "args"]:
                            try:
                                param_names.remove(arg)
                            except ValueError:
                                pass
                        self.assertEqual(
                            set(param_names),
                            set([k["name"] for k in data_point["kwargs"]]),
                        )


if __name__ == "__main__":
    unittest.main()