Reset DATABASE_URL OS environ value properly

Browse Source 
Let's be nice, who knows when this will save the day.
This commit is contained in:
Stefan Wehrmeyer 2013-12-18 11:42:19 +01:00
parent 95717d4584
commit a6095a0506
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_persistence.py
View File

@ -13,6 +13,7 @@ from .sample_data import TEST_DATA, TEST_CITY_1
class DatabaseTestCase(unittest.TestCase): class DatabaseTestCase(unittest.TestCase):
def setUp(self): def setUp(self):
self.old_db_url = os.environ.get('DATABASE_URL')
os.environ['DATABASE_URL'] = 'sqlite:///:memory:' os.environ['DATABASE_URL'] = 'sqlite:///:memory:'
self.db = connect('sqlite:///:memory:') self.db = connect('sqlite:///:memory:')
self.tbl = self.db['weather'] self.tbl = self.db['weather']
@ -21,7 +22,10 @@ class DatabaseTestCase(unittest.TestCase):
def tearDown(self): def tearDown(self):
# ensure env variable was unset # ensure env variable was unset
if self.old_db_url is None:
del os.environ['DATABASE_URL'] del os.environ['DATABASE_URL']
else:
os.environ['DATABASE_URL'] = self.old_db_url
def test_valid_database_url(self): def test_valid_database_url(self):
assert self.db.url, os.environ['DATABASE_URL'] assert self.db.url, os.environ['DATABASE_URL']