From e853747c900e5bd58d9d3259017bccfc6d63b9c9 Mon Sep 17 00:00:00 2001
From: Lukas Holecek <hluk@email.cz>
Date: May 13 2019 12:04:26 +0000
Subject: tests: Allow overriding only DATABASE_URI for tests


Signed-off-by: Lukas Holecek <hluk@email.cz>

---

diff --git a/waiverdb/app.py b/waiverdb/app.py
index b2181b1..b5df4ac 100644
--- a/waiverdb/app.py
+++ b/waiverdb/app.py
@@ -37,9 +37,17 @@ def load_config(app):
         default_config_obj = 'waiverdb.config.ProductionConfig'
         default_config_file = '/etc/waiverdb/settings.py'
         silent = False
+
     app.config.from_object(default_config_obj)
     config_file = os.environ.get('WAIVERDB_CONFIG', default_config_file)
     app.config.from_pyfile(config_file, silent=silent)
+
+    # Allow overriding only DATABASE_URI for tests.
+    if os.getenv('TEST') == 'true':
+        db_uri = app.config['DATABASE_URI']
+        app.config.from_object(default_config_obj)
+        app.config['DATABASE_URI'] = db_uri
+
     if os.environ.get('SECRET_KEY'):
         app.config['SECRET_KEY'] = os.environ['SECRET_KEY']