mirror of
https://github.com/bvanroll/python-vault-db.git
synced 2025-08-29 03:52:40 +00:00
52 lines
1.3 KiB
Python
52 lines
1.3 KiB
Python
import unittest
|
|
import sys
|
|
import os
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
import src
|
|
|
|
class DatabaseClassTests(unittest.TestCase):
|
|
|
|
def test_vault_connection_error(self):
|
|
#TODO give some "valid" url
|
|
self.fail()
|
|
|
|
def test_vault_non_existent_database(self):
|
|
#TODO test non existent_database
|
|
self.fail()
|
|
|
|
def test_psql_connection(self):
|
|
#TODO exec psql connect
|
|
print("hi")
|
|
test = src.Database()
|
|
test.check_valid()
|
|
self.assertEqual(test.valid, False)
|
|
|
|
def test_psql_invalid_vault_url(self):
|
|
try:
|
|
src.Database(dbname="sdfk", vault_url="localhost", vault_port=8200, token="bla")
|
|
except:
|
|
#TODO check the err msg
|
|
return
|
|
else:
|
|
self.fail()
|
|
|
|
|
|
def test_psql_invalid_vault_port(self):
|
|
try:
|
|
src.Database(dbname="sdfk", vault_url="localhost", vault_port="a", token="bla")
|
|
except:
|
|
#TODO check the error message
|
|
return
|
|
else:
|
|
self.fail()
|
|
|
|
|
|
def test_psql_valid(self):
|
|
print(src.Database(dbname="psql", vault_url="http://localhost", token="s.GoR2nisHPeKU1vOaw9hZ5L7h").get_creds())
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|