From e8b5f93401144af9773c371d846b16dd0c9277b5 Mon Sep 17 00:00:00 2001 From: junos Date: Thu, 24 Dec 2020 14:08:31 +0100 Subject: [PATCH] Reformat code. --- test/database.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 test/database.py diff --git a/test/database.py b/test/database.py new file mode 100644 index 0000000..a9d4a86 --- /dev/null +++ b/test/database.py @@ -0,0 +1,21 @@ +import unittest +import psycopg2 +import os + +db_host = "212.235.208.113" +db_database = "staw" +db_user = "staw_db" + +class DatabaseConnection(unittest.TestCase): + def setUp(self): + self.conn = psycopg2.connect(host=db_host, + user=db_user, + password=os.getenv("DB_PASSWORD"), + database=db_database) + + def tearDown(self): + self.conn.close() + + def test_connection(self): + with self.conn.cursor() as cursor: + self.assertIsNotNone(cursor)