#!/usr/bin/python import unittest import ddOperApi from time import sleep from testServer import forkTestServer, stopTestServer class ddOperApiTest(unittest.TestCase): @classmethod def setUpClass(cls): global forkTestServerPid forkTestServerPid = forkTestServer() @classmethod def tearDownClass(cls): global forkTestServerPid stopTestServer(forkTestServerPid) def setUp(self): self.client = ddOperApi.ddOperApi(url="https://localhost:5000/dd-oper/2.0", checkssl=False) def tearDown(self): pass def test_version(self): self.assertTrue(hasattr(ddOperApi, '__version__')) self.assertEqual(type(ddOperApi.__version__), str) def test_setup(self): self.assertEqual(type(self.client), ddOperApi.ddOperApi) def test_locations(self): locations = self.client.locations() self.assertEqual(type(locations), ddOperApi.ddOperLocation) self.assertTrue(len(locations.data) > 0) self.assertEqual(type(locations.data[0]["properties"]["locationName"]), str) def test_locationNames(self): names = self.client.locations().locationNames() self.assertEqual(type(names), type({}.keys())) self.assertTrue(len(names) > 0) for n in names: self.assertEqual(type(n), str) def test_locationDetail(self): locations = self.client.locations() name = locations.data[0]["properties"]["locationName"] detail = locations.locationDetail(name) self.assertEqual(type(detail), dict) self.assertEqual(detail["properties"]["locationName"], name) def test_quantities(self): quantities = self.client.quantities() self.assertEqual(type(quantities), ddOperApi.ddOperQuantitie) self.assertTrue(len(quantities.data) > 0) self.assertEqual(type(quantities.data[0]), str) for q in quantities.quantities(): self.assertEqual(type(q), str) def test_values(self): values = self.client.values("test1", "null") self.assertEqual(type(values), ddOperApi.ddOperValues) def main(): unittest.main() if __name__ == '__main__': main()