|
| 1 | +#!/usr/bin/env python |
| 2 | + |
| 3 | +import unittest |
| 4 | +from osdf import OSDF |
| 5 | + |
| 6 | +class OsdfTest(unittest.TestCase): |
| 7 | + server = "osdf-devel.igs.umaryland.edu" |
| 8 | + username = "test" |
| 9 | + password = "test" |
| 10 | + |
| 11 | + def testGetInfo(self): |
| 12 | + osdf = OSDF(OsdfTest.server, OsdfTest.username, OsdfTest.password) |
| 13 | + |
| 14 | + info = None |
| 15 | + |
| 16 | + try: |
| 17 | + info = osdf.get_info() |
| 18 | + except: |
| 19 | + pass |
| 20 | + |
| 21 | + self.assertIsNotNone(info, "Information retrieved is not null.") |
| 22 | + |
| 23 | + self.assertTrue(type(info) == dict) |
| 24 | + |
| 25 | + self.assertTrue('description' in info) |
| 26 | + self.assertTrue('title' in info) |
| 27 | + |
| 28 | + self.assertTrue('admin_contact_email1' in info) |
| 29 | + self.assertTrue('admin_contact_email2' in info) |
| 30 | + |
| 31 | + self.assertTrue('technical_contact1' in info) |
| 32 | + self.assertTrue('technical_contact2' in info) |
| 33 | + |
| 34 | + |
| 35 | + def testOqlQuery(self): |
| 36 | + osdf = OSDF(OsdfTest.server, OsdfTest.username, OsdfTest.password) |
| 37 | + |
| 38 | + query = '"project"[node_type]' |
| 39 | + namespace = "test" |
| 40 | + |
| 41 | + results = osdf.oql_query(namespace, query) |
| 42 | + |
| 43 | + self.assertIsNotNone(results, "OQL query result is not None.") |
| 44 | + |
| 45 | + self.assertTrue(type(results) == dict, "OQL query result is a list.") |
| 46 | + |
| 47 | + self.assertTrue('result_count' in results, "Result contains 'result_count' key.") |
| 48 | + self.assertTrue('results' in results, "Result contains 'results' key.") |
| 49 | + self.assertTrue('page' in results, "Result contains 'page' key.") |
| 50 | + |
| 51 | + def testQuery(self): |
| 52 | + osdf = OSDF(OsdfTest.server, OsdfTest.username, OsdfTest.password) |
| 53 | + |
| 54 | + query = '{ "term" : { "node_type" : "project" }}' |
| 55 | + namespace = "test" |
| 56 | + |
| 57 | + results = osdf.query(namespace, query) |
| 58 | + |
| 59 | + self.assertIsNotNone(results, "Query result is not None.") |
| 60 | + |
| 61 | + self.assertTrue(type(results) == dict, "Query result is a list.") |
| 62 | + |
| 63 | + self.assertTrue('result_count' in results, "Result contains 'result_count' key.") |
| 64 | + self.assertTrue('results' in results, "Result contains 'results' key.") |
| 65 | + self.assertTrue('page' in results, "Result contains 'page' key.") |
| 66 | + |
| 67 | + |
| 68 | +if __name__ == "__main__": |
| 69 | + unittest.main() |
0 commit comments