#915 Test for GET /client/info and POST /client/add

2 files changed, 56 insertions, 0 deletions

diff --git a/tests/units/test_0021_get_client_info.py b/tests/units/test_0021_get_client_info.py
new file mode 100644
index 0000000..de2126b
--- /dev/null
+++ b/tests/units/test_0021_get_client_info.py
@@ -0,0 +1,16 @@
+import requests
+import unittest
+
+class TestGetClientInfoMethods(unittest.TestCase):
+
+    def setUp(self):
+        self.url = 'http://localhost:8888/client/info'
+        self.headers = {'Authorization' : '07b3bfe728954619b58f0107ad73acc1'}
+        self.json = { "client": ["192.168.56.12"] }
+
+    def test_get(self):
+        returned = requests.get(self.url, headers=self.headers, json=self.json)
+        self.assertEqual(returned.status_code, 200)
+
+if __name__ == '__main__':
+    unittest.main()
\ No newline at end of file
diff --git a/tests/units/test_0022_post_client_add.py b/tests/units/test_0022_post_client_add.py
new file mode 100644
index 0000000..30678ed
--- /dev/null
+++ b/tests/units/test_0022_post_client_add.py
@@ -0,0 +1,40 @@
+import requests
+import unittest
+
+class TestPostClientAddMethods(unittest.TestCase):
+
+    def setUp(self):
+        self.url = 'http://localhost:8888/client/add'
+        self.headers = {'Authorization' : '07b3bfe728954619b58f0107ad73acc1'}
+        self.json = { 'boot' :"19pxeADMIN",
+                      "center": 0,
+                      "hardware_id": 0,
+                       "id": 2,
+                       "ip": "192.168.56.13",
+                       "livedir": "ogLive",
+                       "mac": "0800270E6512",
+                       "maintenance": True,
+                       "name": "pc12",
+                       "netdriver": "generic",
+                       "netiface": "eth1",
+                       "netmask": "255.255.255.0",
+                       "remote": False,
+                       "repo_id": 1,
+                       "room": 1,
+                       "serial_number": "" }
+
+
+    def test_post(self):
+        returned = requests.post(self.url, headers=self.headers, json=self.json)
+        self.assertEqual(returned.status_code, 200)
+
+    #def test_post_no_payload(self):
+        #returned = requests.post(self.url, headers=self.headers, json={})
+        #self.assertEqual(returned.status_code, 400)
+
+    #def test_post_malformed_payload(self):
+        #returned = requests.post(self.url, headers=self.headers, json={'boot' :"19pxeADMIN"})
+        #self.assertEqual(returned.status_code, 400)
+
+if __name__ == '__main__':
+    unittest.main()