使用flask和pytest测试不同的提交按钮

Question

我的烧瓶应用程序中有一个表单，我想在该表单上测试工作流程。

我有一个同时具有Add和Done按钮的表单。

我遇到的问题是，我不知道在使用时是否可以选择“按下”哪个按钮。

res = client.post('/', data ={** my data **})

我被迫使用硒之类的东西还是可以在pytest / pytest-flask中完成此操作？

forms.py

class MyForm(FlaskForm):
    # Fields
    ...

    # submit
    add = SubmitField('Add')
    done = SubmitField('Done & Continue')

测试

import pytest

class TestApp:
    def test_me(self, client):
        '''
        Test
        '''
        res = client.post('/', data = {
            'field': 'field_data'
        }, follow_redirects=True)
        assert 'On add page' in res.get_data(as_text=True) <-- want to be able to switch between these
        assert 'On done page' in res.get_data(as_text=True) <--