from diceplayer.shared.config.gaussian_config import GaussianDTO
from diceplayer.shared.config.player_config import PlayerConfig
from diceplayer.shared.config.dice_config import DiceConfig

import unittest


def get_config_dict():
    return {
        'opt': True,
        'mem': 12,
        'maxcyc': 100,
        'nprocs': 4,
        'ncores': 4,
        'dice': {
            'ljname': 'test',
            'outname': 'test',
            'dens': 1.0,
            'nmol': [1],
            'nstep': [1, 1],
        },
        'gaussian': {
            'level': 'test',
            'qmprog': 'g16',
            'keywords': 'test',
        }
    }


class TestPlayerDTO(unittest.TestCase):
    def setUp(self) -> None:
        self.dice_dto = DiceConfig(
            ljname='test',
            outname='test',
            dens=1.0,
            nmol=[1],
            nstep=[1, 1],
        )
        self.gaussian_dto = GaussianDTO(
            level='test',
            qmprog='g16',
            keywords='test',
        )

    def test_class_instantiation(self):
        player_dto = PlayerConfig(
            opt=True,
            mem=12,
            maxcyc=100,
            nprocs=4,
            ncores=4,
            dice=self.dice_dto,
            gaussian=self.gaussian_dto
        )

        self.assertIsInstance(player_dto, PlayerConfig)
        self.assertIsInstance(player_dto.dice, DiceConfig)
        self.assertIsInstance(player_dto.gaussian, GaussianDTO)

    def test_min_altsteps(self):
        player_dto = PlayerConfig(
            opt=True,
            mem=12,
            maxcyc=100,
            nprocs=4,
            ncores=4,
            altsteps=100,
            dice=self.dice_dto,
            gaussian=self.gaussian_dto
        )

        self.assertEqual(player_dto.altsteps, 20000)

    def test_from_dict(self):
        player_dto = PlayerConfig.from_dict(
            get_config_dict()
        )

        self.assertIsInstance(player_dto, PlayerConfig)
        self.assertIsInstance(player_dto.dice, DiceConfig)
        self.assertIsInstance(player_dto.gaussian, GaussianDTO)


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