12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061 |
- import os
- import unittest
- import sys
- from pwman.data.drivers.postgresql import PostgresqlDatabase
- from pwman.data.nodes import Node
- from pwman.util.crypto_engine import CryptoEngine
- from .test_crypto_engine import give_key, DummyCallback
- import psycopg2 as pg
- class TestPostGresql(unittest.TestCase):
- @classmethod
- def setUpClass(self):
- secret = open('secret.txt').readline().strip()
- u = "postgresql://oz123:%s@localhost/pwman" % secret
- self.db = PostgresqlDatabase(u)
- self.db._open()
- @classmethod
- def tearDownClass(self):
- self.db._cur.execute("DROP TABLE LOOKUP")
- self.db._cur.execute("DROP TABLE TAG")
- self.db._cur.execute("DROP TABLE NODE")
- self.db._cur.execute("DROP TABLE DBVERSION")
- self.db._cur.execute("DROP TABLE CRYPTO")
- self.db._con.commit()
- def test_1_con(self):
- self.assertIsInstance(self.db._cur, pg._psycopg.cursor)
- def test_2_create_tables(self):
- self.db._create_tables()
-
- self.db._create_tables()
- if __name__ == '__main__':
- ce = CryptoEngine.get()
- ce.callback = DummyCallback()
- ce.changepassword(reader=give_key)
- unittest.main(verbosity=2, failfast=True)
|