2020-12-24 14:08:31 +01:00
|
|
|
import unittest
|
2020-12-24 14:29:47 +01:00
|
|
|
|
2020-12-24 14:08:31 +01:00
|
|
|
import psycopg2
|
2020-12-24 14:11:58 +01:00
|
|
|
|
2020-12-24 14:29:47 +01:00
|
|
|
from setup import db_database, db_host, db_password, db_user
|
2020-12-24 14:08:31 +01:00
|
|
|
|
|
|
|
|
|
|
|
class DatabaseConnection(unittest.TestCase):
|
|
|
|
def setUp(self):
|
2020-12-24 14:29:47 +01:00
|
|
|
self.conn = psycopg2.connect(
|
|
|
|
host=db_host, user=db_user, password=db_password, database=db_database,
|
|
|
|
)
|
2020-12-24 14:08:31 +01:00
|
|
|
|
|
|
|
def tearDown(self):
|
|
|
|
self.conn.close()
|
|
|
|
|
|
|
|
def test_connection(self):
|
|
|
|
with self.conn.cursor() as cursor:
|
|
|
|
self.assertIsNotNone(cursor)
|