-# GNU Mediagoblin -- federated, autonomous media hosting
-# Copyright (C) 2011 Free Software Foundation, Inc
+# GNU MediaGoblin -- federated, autonomous media hosting
+# Copyright (C) 2011 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
+
+from mediagoblin.tests.tools import (
+ MEDIAGOBLIN_TEST_DB_NAME, suicide_if_bad_celery_environ)
+
+
+def setup_package():
+ suicide_if_bad_celery_environ()
+
+
+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"