mirror of
https://github.com/kellyjonbrazil/jc.git
synced 2025-10-08 23:22:21 +02:00
add pgpass tests
This commit is contained in:
1
tests/fixtures/generic/pgpass.json
vendored
Normal file
1
tests/fixtures/generic/pgpass.json
vendored
Normal file
@@ -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"}]
|
47
tests/test_pgpass.py
Normal file
47
tests/test_pgpass.py
Normal file
@@ -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()
|
Reference in New Issue
Block a user