def test_construct_engine_from_params_with_passed_values(self):
engine = PG.construct_engine_from_params(user='******',
password='******',
host='testhost',
port='1111',
dbname='testname')
assert engine.url.drivername == 'postgresql'
assert engine.url.username == 'testuser'
assert engine.url.password == 'testpassword'
assert engine.url.host == 'testhost'
assert engine.url.port == 1111
assert engine.url.database == 'testname'
def test_construct_engine_from_env(self, monkeypatch):
monkeypatch.setenv('PGUSER', 'envuser')
monkeypatch.setenv('PGPASSWORD', 'envpass')
monkeypatch.setenv('PGHOST', 'envhost')
monkeypatch.setenv('PGPORT', '1234')
monkeypatch.setenv('PGDATABASE', 'envdb')
engine = PG.construct_engine_from_params()
assert engine.url.drivername == 'postgresql'
assert engine.url.username == 'envuser'
assert engine.url.password == 'envpass'
assert engine.url.host == 'envhost'
assert engine.url.port == 1234
assert engine.url.database == 'envdb'
def test_construct_engine_from_default(self, monkeypatch):
with monkeypatch.context() as m:
m.delenv('PGUSER', raising=False)
m.delenv('PGPASSWORD', raising=False)
m.delenv('PGHOST', raising=False)
m.delenv('PGPORT', raising=False)
m.delenv('PGDATABASE', raising=False)
engine = PG.construct_engine_from_params()
assert engine.url.drivername == 'postgresql'
assert engine.url.username == getpass.getuser()
assert engine.url.password == ''
assert engine.url.host == 'localhost'
assert engine.url.port == 5432
assert engine.url.database == engine.url.username