Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 1 addition & 3 deletions d3b_api_client_cli/db/postgres/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,9 +25,7 @@ class DBConnectionParam:

def __post_init__(self):
if any(
value is None
for value in vars(self).values()
if not callable(value)
not value for value in vars(self).values() if not callable(value)
):
display = {}
for k, v in vars(self).items():
Expand Down
52 changes: 52 additions & 0 deletions tests/unit/db/test_postgres.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
"""
Unit tests for interacting with postgres
"""

import pytest

from d3b_api_client_cli.db.postgres import DBConnectionParam


def test_db_connection_params():
"""
Test DBConnectionParam
"""
username = "foo"
password = "bar"
hostname = "postgres.com"
port = "5432"
db_name = "postgres"

params = DBConnectionParam(
username=username,
password=password,
hostname=hostname,
port=port,
db_name=db_name,
)

assert params.username == username
assert params.password == password
assert params.hostname == hostname
assert params.port == port
assert params.db_name == db_name


def test_db_connection_params_error():
"""
Test DBConnectionParam errors
"""
username = ""
password = ""
hostname = None
port = None
db_name = "postgres"

with pytest.raises(ValueError):
DBConnectionParam(
username=username,
password=password,
hostname=hostname,
port=port,
db_name=db_name,
)
Loading