Merge pull request #3 from MycroftAI/dev

minor changes
MycroftAI · Sep 20, 2018 · bf132bf · bf132bf
2 parents 6320171 + 43070d0
commit bf132bf
Showing 1 changed file with 12 additions and 4 deletions.
diff --git a/market/backend/v1/market-api/market_api/config.py b/market/backend/v1/market-api/market_api/config.py
@@ -10,21 +10,29 @@ class BaseConfig:
     """Base configuration."""
     DEBUG = False
     SECRET_KEY = os.environ['JWT_SECRET']
+    SELENE_BASE_URL = os.environ['SELENE_BASE_URL']
+    TARTARUS_BASE_URL = os.environ['TARTARUS_BASE_URL']
 
 
 class DevelopmentConfig(BaseConfig):
     """Development configuration."""
     DEBUG = True
-    SELENE_BASE_URL = os.environ['SELENE_BASE_URL']
-    TARTARUS_BASE_URL = os.environ['TARTARUS_BASE_URL']
+
+
+class TestConfig(BaseConfig):
+    pass
+
+
+class ProdConfig(BaseConfig):
+    pass
 
 
 def get_config_location():
     """Determine which config to load based on environment"""
     environment_configs = dict(
         dev='market_api.config.DevelopmentConfig',
-        # test=TestConfig,
-        # prod=ProdConfig
+        test=TestConfig,
+        prod=ProdConfig
     )
 
     try: