예제 #1
0
def test_fires_callbacks_data_received():
    progress = ProgressListener(upload_id='foo')

    cb = mock.Mock()
    progress.register_callback(cb)
    progress.data_received(42)

    assert len(cb.mock_calls) == 1

    _, call_args, call_kwargs = cb.mock_calls[0]
    state = call_args[0]

    assert not state['files']
    assert state['total'] == [42, None]
예제 #2
0
def test_fires_callbacks_filepart():
    progress = ProgressListener(upload_id='foo')

    cb = mock.Mock()
    progress.register_callback(cb)
    progress.file_part_received('file.txt', 42, 150)

    assert len(cb.mock_calls) == 1

    _, call_args, call_kwargs = cb.mock_calls[0]
    state = call_args[0]

    assert 'file.txt' in state['files']
    assert state['files']['file.txt'] == [42, 150]