import pytest

from pymba import Vimba, VimbaException


def test_version():
    version = Vimba.version().split('.')
    assert int(version[0]) >= 1
    assert int(version[1]) >= 7
    assert int(version[2]) >= 0


def test_startup_shutdown():
    with pytest.raises(VimbaException) as e:
        Vimba().system().feature_names()
    assert e.value.error_code == VimbaException.ERR_STARTUP_NOT_CALLED

    # manual
    Vimba().startup()
    Vimba().system().feature_names()
    Vimba().shutdown()

    # context manager
    with Vimba() as vmb:
        vmb.system().feature_names()


@pytest.fixture
def vimba() -> Vimba:
    with Vimba() as vimba:
        yield vimba


def test_interface_camera_ids(vimba: Vimba):
    # test id funcs return a list of strings (not bytes)
    for func in (vimba.interface_ids, vimba.camera_ids):
        ids = func()
        assert isinstance(ids, list)
        assert ids
        for x in ids:
            assert isinstance(x, str)


def test_interface(vimba: Vimba):
    interface = vimba.interface(vimba.interface_ids()[0])


def test_camera(vimba: Vimba):
    camera = vimba.camera(vimba.camera_ids()[0])