def test_configure_login_method_gets_tasks_setting(self, mock_get, mock_celery): """ Ensure login_method is checked when configure_login_method is called. """ celery_instance.configure_login_method() mock_get.assert_called_once_with('tasks', 'login_method')
def test_configure_broker_login_method_if_settings_not_default(self, mock_get, mock_celery): """ Celery config should have BROKER_LOGIN_METHOD set if login_method is set in settings """ mock_get.return_value = 'EXTERNAL' celery_instance.configure_login_method() mock_celery.conf.update.assert_called_once_with(BROKER_LOGIN_METHOD=mock_get.return_value)
def test_configure_broker_login_method_if_settings_default(self, mock_get, mock_celery): """ Celery config should not have BROKER_LOGIN_METHOD set if login_method is unset in settings """ mock_get.return_value = '' celery_instance.configure_login_method() self.assertFalse(mock_celery.conf.update.called)
def test_configure_broker_login_method_if_settings_default( self, mock_get, mock_celery): """ Celery config should not have BROKER_LOGIN_METHOD set if login_method is unset in settings """ mock_get.return_value = '' celery_instance.configure_login_method() self.assertFalse(mock_celery.conf.update.called)
def test_configure_broker_login_method_if_settings_not_default( self, mock_get, mock_celery): """ Celery config should have BROKER_LOGIN_METHOD set if login_method is set in settings """ mock_get.return_value = 'EXTERNAL' celery_instance.configure_login_method() mock_celery.conf.update.assert_called_once_with( BROKER_LOGIN_METHOD=mock_get.return_value)