update db connection to testcase class

Author: cortze

docker-compose.yaml

@@ -10,4 +10,4 @@ services:
       - POSTGRES_PASSWORD=${DB_PASSWORD}
       - POSTGRES_DB=${DB_DATABASE}
     volumes:
-      - ./.github/workflows/init_test_db.sql:/docker-entrypoint-initdb.d/docker_postgres_init.sql
+      - ./psql/docker_init_test_db.sql:/docker-entrypoint-initdb.d/docker_postgres_init.sql

psql/docker_init_test_db.sql

@@ -0,0 +1,20 @@
+-- Create custom tables for the testing
+CREATE TABLE IF NOT EXISTS test_table(
+	id INT,
+	text_col TEXT,
+	attribute_col TEXT,
+	bool_col BOOL,
+
+	PRIMARY KEY(id));
+
+-- Insert dummy data into test_table
+INSERT INTO test_table(
+	id,
+	text_col,
+	attribute_col,
+	bool_col)
+VALUES
+(1, 'first', 'attr_1', false),
+(2, 'second', null,  true),
+(3, 'third', 'attr_1', false),
+(4, 'fourth', 'attr_3', true);

unit_db_test/postgresql.py

@@ -39,10 +39,8 @@ def __init__(
             self.engine = sa.create_engine(url=self.url)
             self.conn = self.engine.connect()
         except pspg.OperationalError as dbError:
-            print(f"Error connecting to the database: {dbError}")
             raise NoConnectionToDBError(dbError)
         except Exception as e:
-            print(f"An unexpected error occurred: {e}")
             raise NoConnectionToDBError(e)
 
     def get_df_from_sql_query(self, sql_query):

unit_db_test/testcase.py

@@ -9,30 +9,36 @@
 
 class DBintegrityTest(unittest.TestCase):
     db_config_file = ".env"
+    db = None
 
     # Initialization and configuration of the database
-    def dbCredentials(self):
+    @classmethod
+    def dbCredentials(cls):
         """ replace this function at each unitt db integrity test with the db credentials """
-        self.db_config_file = ".env"
+        cls.db_config_file = ".env"
 
-    def readCredentialsFromEnv(self):
-        return load_credentials_from(self.db_config_file)
+    @classmethod
+    def readCredentialsFromEnv(cls):
+        return load_credentials_from(cls.db_config_file)
 
-    def connectDB(self, user, password, ip, port, db, url):
-        self.db = Postgres(user=user, password=password, host=ip, port=port, database=db, url=url)
+    @classmethod
+    def connectDB(cls, user, password, ip, port, db):
+        cls.db = Postgres(user=user, password=password, host=ip, port=port, database=db)
 
     # setup main function for the db connection
-    def setUp(self):
+    @classmethod
+    def setUpClass(cls):
         try:
-            u, p, i, port, db, url = self.readCredentialsFromEnv()
-            self.connectDB(u, p, i, port, db, url)
+            u, p, i, port, db = cls.readCredentialsFromEnv()
+            cls.connectDB(u, p, i, port, db)
         except NoEnvVariableError as envError:
             raise envError
         except NoConnectionToDBError as dbError:
             raise dbError
 
-    def setDown(self):
-        self.db.close()
+    @classmethod
+    def setDownClass(cls):
+        cls.db.close()
 
     # Extension of the pre-available assert-methods (Panda Oriented)
     def assertNotNullItemsInColumn(self, df: pd.DataFrame, column: str):