Add travis config to test on sqlite, postgres, mysql

This commit is contained in:
Stefan Wehrmeyer
2014-01-31 18:10:49 +01:00
parent 9035edceeb
commit 2865e7c50e
2 changed files with 11 additions and 11 deletions
+2 -10
View File
@@ -13,20 +13,12 @@ from .sample_data import TEST_DATA, TEST_CITY_1
class DatabaseTestCase(unittest.TestCase):
def setUp(self):
self.old_db_url = os.environ.get('DATABASE_URL')
os.environ['DATABASE_URL'] = 'sqlite:///:memory:'
self.db = connect('sqlite:///:memory:')
os.environ.setdefault('DATABASE_URL', 'sqlite:///:memory:')
self.db = connect(os.environ['DATABASE_URL'])
self.tbl = self.db['weather']
for row in TEST_DATA:
self.tbl.insert(row)
def tearDown(self):
# ensure env variable was unset
if self.old_db_url is None:
del os.environ['DATABASE_URL']
else:
os.environ['DATABASE_URL'] = self.old_db_url
def test_valid_database_url(self):
assert self.db.url, os.environ['DATABASE_URL']