Example #1
0
    def test_clean_up_files_catches_exception(self,
                                              av_get_document_with_exception):
        av_get_document_with_exception.side_effect = NotFoundError('no snap there, friend')

        resp_object = av.clean_up_files('jibba_jabba')

        av_get_document_with_exception.assert_called_once_with('jibba_jabba', document_type='snap')
        assert resp_object.status_code == 404
        assert resp_object.data == '"no snap there, friend"'
Example #2
0
    def test_clean_up_files_catches_exception(self,
                                              av_get_document_with_exception):
        av_get_document_with_exception.side_effect = NotFoundError(
            'no snap there, friend')

        resp_object = av.clean_up_files('jibba_jabba')

        av_get_document_with_exception.assert_called_once_with(
            'jibba_jabba', document_type='snap')
        assert resp_object.status_code == 404
        assert resp_object.data == '"no snap there, friend"'
Example #3
0
    def test_clean_up_files_calls_appropriate_methods(self,
                                                      av_clean_up_files_task,
                                                      av_gather_and_enforce_request_args,
                                                      av_get_document_with_exception):

        av_clean_up_files_task.delay = Mock()
        av_gather_and_enforce_request_args.return_value = {'group_id': 'dooda'}
        av_get_document_with_exception.return_value = {'a': 'b'}

        resp_object = av.clean_up_files('jibba_jabba')

        av_get_document_with_exception.assert_called_once_with('jibba_jabba', document_type='snap')
        av_gather_and_enforce_request_args.assert_called_once_with([{'name': 'group_id', 'default': 'current'}])
        av_clean_up_files_task.delay.assert_called_once_with('jibba_jabba', 'dooda')
Example #4
0
    def test_clean_up_files_calls_appropriate_methods(
            self, av_clean_up_files_task, av_gather_and_enforce_request_args,
            av_get_document_with_exception):

        av_clean_up_files_task.delay = Mock()
        av_gather_and_enforce_request_args.return_value = {'group_id': 'dooda'}
        av_get_document_with_exception.return_value = {'a': 'b'}

        resp_object = av.clean_up_files('jibba_jabba')

        av_get_document_with_exception.assert_called_once_with(
            'jibba_jabba', document_type='snap')
        av_gather_and_enforce_request_args.assert_called_once_with([{
            'name':
            'group_id',
            'default':
            'current'
        }])
        av_clean_up_files_task.delay.assert_called_once_with(
            'jibba_jabba', 'dooda')