mirror of
https://github.com/kellyjonbrazil/jc.git
synced 2025-07-17 01:32:37 +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