#! /bin/env pytest
"""Tests of the Python bindings for Hamlib

Running this script directly will use the installed bindings.
For an in-tree run use "make check", or set PYTHONPATH to point to
the directories containing Hamlib.py and _Hamlib.so.
"""
import Hamlib

Hamlib.rig_set_debug(Hamlib.RIG_DEBUG_NONE)

AMP_MODEL = Hamlib.AMP_MODEL_DUMMY

class TestClass:
    """Container class for tests"""

    def test_without_open(self):
        """Call all the methods that do not depend on open()"""
        amp = Hamlib.Amp(AMP_MODEL)
        assert amp is not None
        assert amp.do_exception == 0
        assert amp.error_status == Hamlib.RIG_OK

        assert isinstance(amp.caps.model_name, str)
        assert isinstance(amp.caps.mfg_name, str)
        assert isinstance(amp.caps.version, str)
        assert isinstance(amp.caps.copyright, str)

        assert amp.set_conf("", "") is None
        assert amp.get_conf("") == ""
        assert amp.get_conf(0) == ""
        conf = amp.get_conf("mcfg")
        assert isinstance(conf, str)
        assert amp.set_conf("mcfg", "foo") is None
        conf = amp.get_conf("mcfg")
        assert conf == ""  # FIXME: should return "foo"

        assert amp.token_lookup("") is None


    def test_with_open(self):
        """Call all the methods that depend on open()"""
        amp = Hamlib.Amp(AMP_MODEL)
        assert amp is not None

        assert amp.state.comm_state == 0
        assert amp.open() is None
        assert amp.state.comm_state == 1
        info = amp.get_info()
        assert isinstance(info, str)
        assert amp.reset(Hamlib.AMP_RESET_FAULT) is None
        assert amp.set_freq(0) is None
        assert amp.set_freq(123.45) is None
        assert amp.get_freq() == 123.45
        assert amp.get_level(Hamlib.AMP_LEVEL_NONE) is None
        level = amp.get_level(Hamlib.AMP_LEVEL_SWR)
        assert isinstance(level, float)
        level = amp.get_level(Hamlib.AMP_LEVEL_PWR_REFLECTED)
        assert isinstance(level, int)
        level = amp.get_level(Hamlib.AMP_LEVEL_FAULT)
        assert isinstance(level, str)
        level = amp.get_level(123456)
        assert level is None
        assert amp.set_powerstat(Hamlib.RIG_POWER_ON) is None
        assert amp.get_powerstat() == Hamlib.RIG_POWER_ON
        assert amp.close() is None


    def test_all_methods(self):
        """Just call all the methods"""
        amp = Hamlib.Amp(AMP_MODEL)
        assert amp is not None

        # the tests that do not depend on open()
        assert amp.set_conf("", "") is None
        assert amp.get_conf("") == ""
        assert amp.get_conf(0) == ""
        conf = amp.get_conf("mcfg")
        assert isinstance(conf, str)
        assert amp.set_conf("mcfg", "foo") is None
        conf = amp.get_conf("mcfg")
        assert conf == ""  # FIXME: should return "foo"
        assert amp.token_lookup("") is None

        # the tests that depend on open()
        assert amp.state.comm_state == 0
        assert amp.open() is None
        assert amp.state.comm_state == 1
        info = amp.get_info()
        assert isinstance(info, str)
        assert amp.reset(Hamlib.AMP_RESET_FAULT) is None
        assert amp.set_freq(0) is None
        assert amp.set_freq(123.45) is None
        assert amp.get_freq() == 123.45
        assert amp.get_level(Hamlib.AMP_LEVEL_NONE) is None
        level = amp.get_level(Hamlib.AMP_LEVEL_SWR)
        assert isinstance(level, float)
        level = amp.get_level(Hamlib.AMP_LEVEL_PWR_REFLECTED)
        assert isinstance(level, int)
        level = amp.get_level(Hamlib.AMP_LEVEL_FAULT)
        assert isinstance(level, str)
        level = amp.get_level(123456)
        assert level is None
        assert amp.set_powerstat(Hamlib.RIG_POWER_ON) is None
        assert amp.get_powerstat() == Hamlib.RIG_POWER_ON
        assert amp.close() is None
        assert amp.state.comm_state == 0
        info = amp.get_info()
        assert info is None


    def test_object_creation(self):
        """Create all objects available"""
        amp = Hamlib.Rig(AMP_MODEL)
        assert amp is not None

        assert isinstance(amp.caps, Hamlib.rig_caps)
        assert isinstance(amp.state, Hamlib.rig_state)