Second round of tests for setup_celery_from_config(), with some database information

1 files changed, 33 insertions, 1 deletions

diff --git a/mediagoblin/tests/test_celery_setup.py b/mediagoblin/tests/test_celery_setup.py
index 4c2f1269..da18b0ef 100644
--- a/mediagoblin/tests/test_celery_setup.py
+++ b/mediagoblin/tests/test_celery_setup.py
@@ -27,7 +27,6 @@ def test_setup_celery_from_config():
         for var in vars_to_wipe:
             delattr(module, var)
 
-
     celery_setup.setup_celery_from_config(
         {},
         {'something': {'or': 'other'},
@@ -51,3 +50,36 @@ def test_setup_celery_from_config():
         'database': 'mediagoblin'}
     assert fake_celery_module.CELERY_RESULT_BACKEND == 'mongodb'
     assert fake_celery_module.BROKER_BACKEND == 'mongodb'
+
+    _wipe_testmodule_clean(fake_celery_module)
+
+    celery_setup.setup_celery_from_config(
+        {'db_host': 'mongodb.example.org',
+         'db_port': '8080',
+         'db_name': 'captain_lollerskates',
+         'celery_section': 'vegetable'},
+        {'something': {'or': 'other'},
+         'vegetable': {'some_variable': 'poolf',
+                       'mail_port': '2020',
+                       'CELERYD_ETA_SCHEDULER_PRECISION': '3.1',
+                       'celery_result_persistent': 'false',
+                       'celery_imports': 'baz.bar.foo import.is.a.this'}},
+        'mediagoblin.tests.fake_celery_module', set_environ=False)
+    
+    from mediagoblin.tests import fake_celery_module
+    assert fake_celery_module.SOME_VARIABLE == 'poolf'
+    assert fake_celery_module.MAIL_PORT == 2020
+    assert isinstance(fake_celery_module.MAIL_PORT, int)
+    assert fake_celery_module.CELERYD_ETA_SCHEDULER_PRECISION == 3.1
+    assert isinstance(fake_celery_module.CELERYD_ETA_SCHEDULER_PRECISION, float)
+    assert fake_celery_module.CELERY_RESULT_PERSISTENT is False
+    assert fake_celery_module.CELERY_IMPORTS == [
+        'baz.bar.foo', 'import.is.a.this']
+    assert fake_celery_module.CELERY_MONGODB_BACKEND_SETTINGS == {
+        'database': 'captain_lollerskates',
+        'host': 'mongodb.example.org',
+        'port': 8080}
+    assert fake_celery_module.CELERY_RESULT_BACKEND == 'mongodb'
+    assert fake_celery_module.BROKER_BACKEND == 'mongodb'
+    assert fake_celery_module.BROKER_HOST == 'mongodb.example.org'
+    assert fake_celery_module.BROKER_PORT == 8080