Aurélie GABU commited on
Commit
9d1c071
·
1 Parent(s): 009706f

denying CI to be connected to DB

Browse files
Files changed (1) hide show
  1. App/database.py +11 -2
App/database.py CHANGED
@@ -5,6 +5,9 @@ from sqlalchemy.orm import sessionmaker, declarative_base
5
 
6
  load_dotenv()
7
 
 
 
 
8
  DB_USER = os.getenv("DB_USER", "postgres")
9
  DB_PASSWORD = os.getenv("DB_PASSWORD", "password")
10
  DB_HOST = os.getenv("DB_HOST", "localhost")
@@ -13,6 +16,12 @@ DB_NAME = os.getenv("DB_NAME", "test_db")
13
 
14
  DATABASE_URL = (f"postgresql+psycopg2://{DB_USER}:{DB_PASSWORD}"f"@{DB_HOST}:{DB_PORT}/{DB_NAME}")
15
 
16
- engine = create_engine(DATABASE_URL)
17
- SessionLocal = sessionmaker(autocommit = False, autoflush = False, bind = engine)
18
  Base = declarative_base()
 
 
 
 
 
 
 
 
 
5
 
6
  load_dotenv()
7
 
8
+ # Détection si on est en train d'exécuter pytest
9
+ IS_TEST = os.getenv("PYTEST_CURRENT_TEST") is not None
10
+
11
  DB_USER = os.getenv("DB_USER", "postgres")
12
  DB_PASSWORD = os.getenv("DB_PASSWORD", "password")
13
  DB_HOST = os.getenv("DB_HOST", "localhost")
 
16
 
17
  DATABASE_URL = (f"postgresql+psycopg2://{DB_USER}:{DB_PASSWORD}"f"@{DB_HOST}:{DB_PORT}/{DB_NAME}")
18
 
 
 
19
  Base = declarative_base()
20
+
21
+ if not IS_TEST:
22
+ engine = create_engine(DATABASE_URL)
23
+ SessionLocal = sessionmaker(autocommit = False, autoflush = False, bind = engine)
24
+
25
+ else:
26
+ engine = None
27
+ SessionLocal = None