Fix Travis build failure.

TeamHG-Memex · Mar 7, 2016 · c7973c2 · c7973c2
1 parent 8fd1452
commit c7973c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/abstract.py b/tests/abstract.py
@@ -91,8 +91,8 @@ def get_credentials_from_env(self):
         '''
 
         try:
-            user = os.environ['POSTGRES_USER']
-            password = os.environ['POSTGRES_PASSWORD']
+            user = os.environ['{}_USER'.format(self.db_type.upper())]
+            password = os.environ['{}_PASSWORD'.format(self.db_type.upper())]
         except KeyError as key:
             message = 'Missing environment variable: {}'
             raise unittest.SkipTest(message.format(key.args[0]))