Function test_translate_document_low_level

tests/test_translate_document.py:228–261

(
    translator,
    example_document_path,
    example_document_translation,
    output_document_path,
    server,
)

@needs_mock_server
def test_translate_document_low_level(
    translator,
    example_document_path,
    example_document_translation,
    output_document_path,
    server,
):
    # Set a small document queue time to attempt downloading a queued document
    server.set_doc_queue_time(100)

    with open(example_document_path, "rb") as infile:
        handle = translator.translate_document_upload(
            infile, **default_lang_args
        )
    status = translator.translate_document_get_status(handle)
    assert status.ok and not status.done

    # Test recreating a document handle from id & key
    doc_id, doc_key = handle.document_id, handle.document_key
    del handle

    handle = deepl.DocumentHandle(doc_id, doc_key)
    status = translator.translate_document_get_status(handle)
    assert status.ok

    while status.ok and not status.done:
        status = translator.translate_document_get_status(handle)
        time.sleep(0.2)

    assert status.ok and status.done
    with open(output_document_path, "wb") as outfile:
        translator.translate_document_download(handle, outfile)

    assert output_document_path.read_text() == example_document_translation


def test_translate_document_string(translator, server):