# GNU MediaGoblin -- federated, autonomous media hosting
-# Copyright (C) 2011 Free Software Foundation, Inc
+# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
-from mediagoblin import mg_globals
+import os
+import shutil
-from mediagoblin.tests.tools import MEDIAGOBLIN_TEST_DB_NAME
+from mediagoblin import mg_globals
+from mediagoblin.tests.tools import (
+ TEST_USER_DEV, suicide_if_bad_celery_environ)
def setup_package():
- pass
+ suicide_if_bad_celery_environ()
+
+ import warnings
+ from sqlalchemy.exc import SAWarning
+ warnings.simplefilter("error", SAWarning)
+
def teardown_package():
- if ((mg_globals.db_connection
- and mg_globals.database.name == MEDIAGOBLIN_TEST_DB_NAME)):
- print "Killing db ..."
- mg_globals.db_connection.drop_database(MEDIAGOBLIN_TEST_DB_NAME)
- print "... done"
+ # Remove and reinstall user_dev directories
+ if os.path.exists(TEST_USER_DEV):
+ shutil.rmtree(TEST_USER_DEV)