From 428293aea6eace8848a0bc4487a4706ee6e778af Mon Sep 17 00:00:00 2001
From: Antoine Gaudron-desjardins <antoine.gaudrondesjardins@student-cs.fr>
Date: Fri, 22 Jul 2022 17:14:34 +0200
Subject: [PATCH] fix

---
 backend/test_main.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/backend/test_main.py b/backend/test_main.py
index b36336b..36b5678 100644
--- a/backend/test_main.py
+++ b/backend/test_main.py
@@ -1,8 +1,12 @@
 from fastapi.testclient import TestClient
 from datetime import datetime, timedelta
+from sqlalchemy import create_engine
+from dotenv import load_dotenv
 import pytz
 
+from db import models
 from main import app
+import os
 
 client = TestClient(app)
 
@@ -22,6 +26,20 @@ test = {
 }
 
 
+def init_connection_bdd():
+    load_dotenv()
+    user = os.getenv('MYSQL_USER')
+    password = os.getenv('MYSQL_PASSWORD')
+    host = os.getenv('DB_HOST')
+    port = os.getenv('DB_PORT')
+    database = os.getenv('MYSQL_DATABASE')
+
+    SQLALCHEMY_DATABASE_URL = f"mysql+pymysql://{user}:{password}@{host}:{port}/{database}?charset=utf8"
+
+    engine = create_engine(SQLALCHEMY_DATABASE_URL)
+    models.Base.metadata.create_all(bind=engine)
+
+
 def test_post_opening_hours():
     response = client.post(
         "/api/opening_hours",
-- 
GitLab