| #!/usr/bin/env python |
| # coding: UTF-8 |
| |
| # testPsqlWrapper |
| # |
| # Copyright(c) 2015 Uptime Technologies, LLC. |
| |
| import unittest |
| import os |
| import PsqlWrapper |
| |
| class TestPsqlWrapper(unittest.TestCase): |
| def _setUp(self): |
| print("setup") |
| |
| def testPsqlWrapper001(self): |
| p = PsqlWrapper.PsqlWrapper(None, None, None, None) |
| |
| self.assertTrue(p is not None) |
| self.assertEqual(p.host, 'localhost') |
| self.assertEqual(p.port, 5432) |
| self.assertEqual(p.username, os.environ['USER']) |
| self.assertEqual(p.dbname, p.username) |
| |
| def testPsqlWrapper002(self): |
| p = PsqlWrapper.PsqlWrapper('localhost', '5432', 'postgres', 'postgres') |
| |
| self.assertTrue(p is not None) |
| self.assertEqual(p.host, 'localhost') |
| self.assertEqual(p.port, 5432) |
| self.assertEqual(p.username, 'postgres') |
| self.assertEqual(p.dbname, 'postgres') |
| |
| def testPsqlWrapper003(self): |
| os.environ['PGHOST'] = '127.0.0.1' |
| os.environ['PGPORT'] = '5433' |
| os.environ['PGUSER'] = 'username' |
| os.environ['PGDATABASE'] = 'dbname' |
| |
| p = PsqlWrapper.PsqlWrapper(None, None, None, None) |
| |
| self.assertTrue(p is not None) |
| self.assertEqual(p.host, '127.0.0.1') |
| self.assertEqual(p.port, 5433) |
| self.assertEqual(p.username, 'username') |
| self.assertEqual(p.dbname, 'dbname') |
| |
| def testPsqlWrapper004(self): |
| os.environ['PGHOST'] = '127.0.0.1' |
| os.environ['PGPORT'] = '5433' |
| os.environ['PGUSER'] = 'username' |
| os.environ['PGDATABASE'] = 'dbname' |
| |
| p = PsqlWrapper.PsqlWrapper('localhost', '5432', 'postgres', 'postgres') |
| |
| self.assertTrue(p is not None) |
| self.assertEqual(p.host, 'localhost') |
| self.assertEqual(p.port, 5432) |
| self.assertEqual(p.username, 'postgres') |
| self.assertEqual(p.dbname, 'postgres') |
| |
| def _tearDown(self): |
| print("teardown") |
| |
| if __name__ == '__main__': |
| unittest.main() |
