Add unit test to the import_db.py functions

app/import_db.py

@@ -42,7 +42,6 @@ def normalize_string(serial_number, fixed_size=30):
     return f"{all_alpha}{missing_zeros}{all_digit}"
 
 
-
 def get_database_connection():
     """connects to the MySQL database and returns the connection"""
     return MySQLdb.connect(host=config.MYSQL_HOST,
@@ -188,7 +187,7 @@ def import_database_from_excel(filepath):
 
     db.close()
 
-    return
+    return serials_counter, invalid_counter
 
 
 def db_check():
@@ -261,11 +260,11 @@ def separate(input_string):
     db.commit()
 
     db.close()
+    return output
 
 
-filepath = sys.argv[1]
-
-import_database_from_excel(filepath)
-db_check()
-
-os.remove(filepath)
+if __name__ == '__main__':
+    filepath = sys.argv[1]
+    import_database_from_excel(filepath)
+    db_check()
+    os.remove(filepath)

app/main.py

@@ -385,11 +385,12 @@ def process():
     return jsonify(ret), 200
 
 def log_new_sms(status, sender, message, answer, cur):
-	if len(message) > 40:
-		return;
-	now = time.strftime('%Y-%m-%d %H:%M:%S')
-    cur.execute("INSERT INTO PROCESSED_SMS (status, sender, message, answer, date) VALUES (%s, %s, %s, %s, %s)", (status, sender, message, answer, now))
-
+    if len(message) > 40:
+        return
+    now = time.strftime('%Y-%m-%d %H:%M:%S')
+    cur.execute("INSERT INTO PROCESSED_SMS (status, sender, message, answer, date) VALUES (%s, %s, %s, %s, %s)",(status, sender, message, answer, now, ))
+
+
 @app.errorhandler(404)
 def page_not_found(error):
     """ returns 404 page"""

app/test_import_db.py

@@ -0,0 +1,27 @@
+import unittest
+import import_db as idb
+
+
+class TestImportDb(unittest.TestCase):
+    def test_normalize_string(self):
+        self.assertEqual(idb.normalize_string('abc12', 10), 'ABC0000012')
+        self.assertEqual(idb.normalize_string('abc00١٢', 10), 'ABC0000012')
+
+    def test_get_database_connection(self):
+        try:
+            idb.get_database_connection()
+        except Exception as err:
+            self.fail(f"Error in connect to db: {err}")
+
+    def test_import_database_from_excel(self):
+        self.assertEqual(
+            idb.import_database_from_excel('./test_excel.xlsx'),
+            (2, 2))
+
+    def test_db_check(self):
+        self.assertNotEqual(idb.db_check(), [])
+
+
+
+if __name__ == '__main__':
+    unittest.main()