Source code for tests.test_dgilib_discovery

"""This module holds the automated tests for DGILib Discovery."""

from pydgilib.dgilib import DGILib
from pydgilib.dgilib_exceptions import DeviceReturnError

import pytest


verbosity = (0, 99)


@pytest.mark.parametrize("verbose", verbosity)
def test_discover(verbose):
    """test_discover.

    DGILibDiscovery.discover
    """
    assert DGILib(verbose=verbose).discover() is None


@pytest.mark.parametrize("verbose", verbosity)
def test_get_device_count(verbose):
    """test_get_device_count.

    DGILibDiscovery.get_device_count
    """
    dgilib = DGILib(verbose=verbose)
    dgilib.discover()
    assert isinstance(dgilib.get_device_count(), int)


@pytest.mark.parametrize("verbose", verbosity)
def test_get_device_name(verbose):
    """test_get_device_name.

    DGILibDiscovery.get_device_name
    """
    dgilib = DGILib(verbose=verbose)
    dgilib.discover()
    assert isinstance(dgilib.get_device_name(), bytes)


@pytest.mark.parametrize("verbose", verbosity)
def test_get_device_serial(verbose):
    """test_get_device_serial.

    DGILibDiscovery.get_device_serial
    """
    dgilib = DGILib(verbose=verbose)
    dgilib.discover()
    device_sn = dgilib.get_device_serial()
    assert isinstance(device_sn, bytes)
    assert len(device_sn) == 20


@pytest.mark.parametrize("verbose", verbosity)
def test_is_msd_mode(verbose):
    """test_is_msd_mode.

    DGILibDiscovery.is_msd_mode
    """
    dgilib = DGILib(verbose=verbose)
    dgilib.discover()
    device_sn = dgilib.get_device_serial()
    assert not dgilib.is_msd_mode(device_sn)


@pytest.mark.xfail(raises=DeviceReturnError, reason="It seems this is " +
                   "not supported by my test setup.")
@pytest.mark.xfail(raises=OSError, reason="Using older version of dgilib" +
                   ".dll")
@pytest.mark.parametrize("verbose", verbosity)
def test_set_mode(verbose):
    """test_set_mode.

    DGILibDiscovery.set_mode
    """
    dgilib = DGILib(verbose=verbose)
    dgilib.discover()
    device_sn = dgilib.get_device_serial()
    dgilib.set_mode(device_sn, 0)
    assert dgilib.is_msd_mode(device_sn)
    dgilib.set_mode(device_sn)
    assert not dgilib.is_msd_mode(device_sn)