From c4941aaf4485678d003fa472c837199383b871bd Mon Sep 17 00:00:00 2001
From: Justin Riley <justin.t.riley@gmail.com>
Date: Tue, 2 Aug 2016 18:06:35 -0400
Subject: [PATCH] add tests for one.OneClient

---
 onedns/tests/conftest.py       |  6 ++++++
 onedns/tests/test_oneclient.py | 17 +++++++++++++++++
 2 files changed, 23 insertions(+)
 create mode 100644 onedns/tests/test_oneclient.py

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)