#!/usr/bin/env python3
import unittest

from pyfroniusreg import gen24_registers

from pymodbus.client.tcp import ModbusTcpClient

fronius1 = ModbusTcpClient("172.19.107.211", port=502, timeout=10)
fronius1.connect()


class TestRead(unittest.TestCase):
    def test_read_scaled(self):
        soc = gen24_registers.scaledStateOfCharge.getValue(fronius1)
        assert isinstance(soc, float)

    def test_read_direct(self):
        dr = gen24_registers.OutWRte.getValue(fronius1)
        assert isinstance(dr, int)

    def test_read_string16(self):
        Mn = gen24_registers.Mn.getValue(fronius1)
        assert Mn[0:7] == "Fronius"

    def test_read_string8(self):
        # this isn't a good test, this value changes regularly
        Vr = gen24_registers.Vr.getValue(fronius1)
        assert Vr == "1.33.7-1"

    def test_read_model(self):
        Md = gen24_registers.Md.getValue(fronius1)
        assert Md == "Primo GEN24 5.0\x00"

    # def test_read_sn(self):
    # This doesn't seem to return anything useful
    # SN = gen24_registers.SN.getValue(fronius1)
    # assert SN == "12345567"

    def test_write_direct(self):
        current = gen24_registers.OutWRte.getValue(fronius1)
        retval = gen24_registers.OutWRte.setValue(fronius1, current)
        assert retval is not None

    def test_write_scaled(self):
        current = gen24_registers.scaledInWRte.getValue(fronius1)
        retval = gen24_registers.scaledInWRte.setValue(fronius1, current)
        assert retval is not None


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