Produits Entr'ouvert » w.c.s.

def test_formdata_generated_document_ods_download(pub):

    create_user(pub)

    wf = Workflow(name='status')

    st1 = wf.add_status('Status1', 'st1')

    export_to = st1.add_action('export_to_model', id='_export_to')

    export_to.convert_to_pdf = False

    export_to.label = 'create doc'

    template_filename = os.path.join(os.path.dirname(__file__), '..', 'template.ods')

    with open(template_filename, 'rb') as fd:

        template = fd.read()

    upload = QuixoteUpload('/foo/template.ods', content_type='application/octet-stream')

    upload.fp = io.BytesIO()

    upload.fp.write(template)

    upload.fp.seek(0)

    export_to.model_file = UploadedFile(pub.app_dir, None, upload)

    export_to.by = ['_submitter']

    wf.store()

    FormDef.wipe()

    formdef = FormDef()

    formdef.name = 'test_formdata_generated_document_ods_download'

    formdef.url_name = 'test'

    formdef.workflow_id = wf.id

    formdef.fields = []

    formdef.store()

    formdef.data_class().wipe()

    resp = login(get_app(pub), username='foo', password='foo').get('/test/')

    resp = resp.forms[0].submit('submit')  # -> validation

    resp = resp.forms[0].submit('submit')  # -> submit

    resp = resp.follow()

    resp = resp.form.submit('button_export_to')

    resp = resp.follow()  # $form/$id/create_doc

    resp = resp.follow()  # $form/$id/create_doc/

    with io.BytesIO(resp.body) as generated_ods:

        with zipfile.ZipFile(generated_ods) as z_ods:

            for name in z_ods.namelist():

                if name != 'content.xml':

                    continue

                content_bytes = z_ods.read(name)

                assert b' xmlns:of="urn:oasis:names:tc:opendocument:xmlns:of:1.2" ' in content_bytes

                assert b'>test_formdata_generated_document_ods_download<' in content_bytes

                # check it's valid XML

                ET.tostring(ET.XML(content_bytes))

            if (

                root.find(f'{{{OO_OFFICE_NS}}}body/{{{OO_OFFICE_NS}}}spreadsheet')

                and b'xmlns:of=' not in content

            ):

                # force xmlns:of namespace inclusion in spreadsheet files, as it may be

                # required for proper handling of table:formula attributes.

                # (there is no easy way to have ElementTree include namespace declarations

                # if there are no elements of that namespace)

                content = content.replace(

                    b':document-content ',

                    b':document-content xmlns:of="urn:oasis:names:tc:opendocument:xmlns:of:1.2" ',

                    1,

                )