Celery wasn't really being properly connected during tests.
[mediagoblin.git] / mediagoblin / celery_setup / from_tests.py
1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011 Free Software Foundation, Inc
3 #
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
8 #
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
13 #
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16
17 import os
18
19 from mediagoblin.tests.tools import TEST_APP_CONFIG
20 from mediagoblin.celery_setup.from_celery import setup_self
21
22
23 OUR_MODULENAME = __name__
24 CELERY_SETUP = False
25
26
27 if os.environ.get('CELERY_CONFIG_MODULE') == OUR_MODULENAME:
28 setup_self(check_environ_for_conf=False, module_name=OUR_MODULENAME,
29 default_conf_file=TEST_APP_CONFIG)
30 CELERY_SETUP = True