Produits Entr'ouvert » w.c.s.

def test_backoffice_csv_export_block(pub):

    create_superuser(pub)

    create_environment(pub)

    block = BlockDef()

    block.name = 'foobar'

    block.fields = [

        fields.StringField(id='123', required=True, label='Test', type='string', varname='foo'),

        fields.StringField(id='234', required=True, label='Test2', type='string', varname='bar'),

    ]

    block.digest_template = 'X{{foobar_var_foo}}Y'

    block.store()

    formdef = FormDef.get_by_urlname('form-title')

    formdef.fields = []

    formdef.store()  # make sure sql column is removed

    formdef.fields = [

        fields.BlockField(id='1', label='test', type='block:foobar', max_items=3),

    ]

    formdef.store()

    formdef.data_class().wipe()

    formdata = formdef.data_class()()

    formdata.data = {}

    formdata.data['1'] = {

        'data': [

            {'123': 'foo', '234': 'bar'},

            {'123': 'foo2', '234': 'bar2'},

        ],

        'schema': {'123': 'string', '234': 'string'},

    }

    formdata.just_created()

    formdata.jump_status('new')

    formdata.store()

    app = login(get_app(pub))

    resp = app.get('/backoffice/management/form-title/')

    resp_csv = resp.click('Export as CSV File')

    resp.forms['listing-settings']['1'].checked = True

    resp = resp.forms['listing-settings'].submit()

    resp_csv = resp.click('Export as CSV File')

    assert resp_csv.text.splitlines()[0].split(',')[-3:] == ['test - 1', 'test - 2', 'test - 3']

    assert resp_csv.text.splitlines()[1].split(',')[-3:] == ['XfooY', 'Xfoo2Y', '']

    def get_csv_heading(self):

        nb_items = (self.max_items or 1)

        if nb_items == 1:

            return [self.label]

        headings = ['%s - %s' % (self.label, x + 1) for x in range(nb_items)]

        return headings

    def get_csv_value(self, value, **kwargs):

        nb_items = (self.max_items or 1)

        cells = [''] * nb_items

        if value and value.get('data'):

            for i, subvalue in enumerate(value.get('data')[:nb_items]):

                if subvalue:

                    cells[i] = self.block.get_display_value(subvalue)

        return cells