diff --git a/tests/fixtures/generic/pgpass.json b/tests/fixtures/generic/pgpass.json new file mode 100644 index 00000000..b2b701b7 --- /dev/null +++ b/tests/fixtures/generic/pgpass.json @@ -0,0 +1 @@ +[{"hostname":"dbserver","port":"443","database":"db1","username":"dbuser","password":"pwd123"},{"hostname":"dbserver2","port":"8888","database":"inventory","username":"joeuser","password":"abc123"},{"hostname":"ax1234","port":"1234","database":"tables","username":"dbadmin","password":"password1"},{"hostname":"dbserver3","port":"8888","database":"inventory","username":"joeuser","password":"abc123"},{"hostname":"localhost","port":"5555","database":"table\\1","username":"user:w:colon","password":"password123"},{"hostname":"hostname","port":"*","database":"database","username":"username","password":"password"}] diff --git a/tests/test_pgpass.py b/tests/test_pgpass.py new file mode 100644 index 00000000..44df0f5a --- /dev/null +++ b/tests/test_pgpass.py @@ -0,0 +1,47 @@ +import os +import unittest +import json +from typing import Dict +from jc.parsers.pgpass import parse + +THIS_DIR = os.path.dirname(os.path.abspath(__file__)) + + +class MyTests(unittest.TestCase): + f_in: Dict = {} + f_json: Dict = {} + + @classmethod + def setUpClass(cls): + fixtures = { + 'pgpass': ( + 'fixtures/generic/pgpass.txt', + 'fixtures/generic/pgpass.json') + } + + for file, filepaths in fixtures.items(): + with open(os.path.join(THIS_DIR, filepaths[0]), 'r', encoding='utf-8') as a, \ + open(os.path.join(THIS_DIR, filepaths[1]), 'r', encoding='utf-8') as b: + cls.f_in[file] = a.read() + cls.f_json[file] = json.loads(b.read()) + + + def test_pgpass_nodata(self): + """ + Test 'pgpass' with no data + """ + self.assertEqual(parse('', quiet=True), []) + + + def test_pgpass(self): + """ + Test postgreSQL password file + """ + self.assertEqual( + parse(self.f_in['pgpass'], quiet=True), + self.f_json['pgpass'] + ) + + +if __name__ == '__main__': + unittest.main()