From 57a7c58f63b44dca298ae8399a6ff7191eae131d Mon Sep 17 00:00:00 2001 From: mattack108 Date: Sun, 14 Jul 2013 21:51:35 +0100 Subject: [PATCH] Test case when `url` comes as env variable. --- test/test_persistence.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/test/test_persistence.py b/test/test_persistence.py index a29056c..4afa10c 100644 --- a/test/test_persistence.py +++ b/test/test_persistence.py @@ -1,3 +1,4 @@ +import os import unittest from datetime import datetime @@ -9,11 +10,19 @@ from sample_data import TEST_DATA class DatabaseTestCase(unittest.TestCase): def setUp(self): - self.db = connect('sqlite:///:memory:') + os.environ['DATABASE_URL'] = 'sqlite:///:memory:' + self.db = connect() self.tbl = self.db['weather'] for row in TEST_DATA: self.tbl.insert(row) + def tearDown(self): + # ensure env variable was unset + del os.environ['DATABASE_URL'] + + def test_valid_database_url(self): + assert self.db.url, os.environ['DATABASE_URL'] + def test_tables(self): assert self.db.tables == ['weather'], self.db.tables @@ -35,7 +44,7 @@ class DatabaseTestCase(unittest.TestCase): class TableTestCase(unittest.TestCase): def setUp(self): - self.db = connect('sqlite:///:memory:') + self.db = connect(url='sqlite:///:memory:') self.tbl = self.db['weather'] for row in TEST_DATA: self.tbl.insert(row)