diff --git a/onedns/tests/conftest.py b/onedns/tests/conftest.py index 40b77ef..3ed5afb 100644 --- a/onedns/tests/conftest.py +++ b/onedns/tests/conftest.py @@ -1,6 +1,7 @@ import pytest from onedns import resolver +from onedns.clients import one DOMAIN = 'onedns.test' @@ -14,3 +15,8 @@ def dns(request): dns.start(dns_address=INTERFACE, dns_port=PORT, tcp=True) request.addfinalizer(dns.close) return dns + + +@pytest.fixture(scope="module") +def oneclient(request): + return one.OneClient() diff --git a/onedns/tests/test_oneclient.py b/onedns/tests/test_oneclient.py new file mode 100644 index 0000000..a1745a8 --- /dev/null +++ b/onedns/tests/test_oneclient.py @@ -0,0 +1,17 @@ +import pytest + +import oca +from oca import vm + + +def test_get_vms(oneclient): + vms = oneclient.vms() + assert isinstance(vms, vm.VirtualMachinePool) + assert len(vms) > 0 + + +def test_get_vm_by_id(oneclient): + with pytest.raises(TypeError): + oneclient.get_vm_by_id('asdf') + vm = oneclient.get_vm_by_id(0) + assert isinstance(vm, oca.VirtualMachine)