import sys
import unittest

from artiq.test.hardware_testbench import GenericControllerCase, ControllerCase


class GenericNovatech409BTest:
    def test_parameters_readback(self):
        # write sample data and read it back
        for i in range(4):
            self.driver.set_freq(i, 1e6)
            self.driver.set_phase(i, 0.5)
            self.driver.set_gain(i, 0.25)
        result = self.driver.get_status()

        # check for expected status message; ignore all but first 23 bytes
        # compare with previous result extracted from Novatech
        for i in range(4):
            r = result[i]
            self.assertEqual(r[0:23], "00989680 2000 01F5 0000")


class TestNovatech409B(GenericNovatech409BTest, ControllerCase):
    def setUp(self):
        ControllerCase.setUp(self)
        self.start_controller("novatech409b")
        self.driver = self.device_mgr.get("novatech409b")


class TestNovatech409BSim(GenericNovatech409BTest, GenericControllerCase):
    def get_device_db(self):
        return {
            "novatech409b": {
                "type": "controller",
                "host": "::1",
                "port": 3254,
                "command": (sys.executable.replace("\\", "\\\\")
                            + " -m artiq.frontend.aqctl_novatech409b "
                            + "-p {port} --simulation")
            }
        }

    def setUp(self):
        GenericControllerCase.setUp(self)
        self.start_controller("novatech409b")
        self.driver = self.device_mgr.get("novatech409b")