Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use dummy_executor method in pytest-play

Best Python code snippet using pytest-play_python

test_engine.py

Source:test_engine.py

1import pytest2import mock3def test_play_engine_constructor(request):4    from pytest_play.engine import PlayEngine5    executor = PlayEngine(request, {'foo': 'bar'})6    assert executor.request is request7    assert executor.variables == {'foo': 'bar'}8def test_get_file_contents(play, data_base_path):9    play.get_file_contents(data_base_path, 'included.yml')10def test_register_teardown(play, data_base_path):11    assert play._teardown == []12    import mock13    callback = mock.MagicMock()14    play.register_teardown_callback(callback)15    play.register_teardown_callback(callback)16    assert callback in play._teardown17    assert len(play._teardown) == 118    assert not callback.called19    play.teardown()20    assert callback.assert_called_once_with() is None21def test_executor_parametrize(dummy_executor):22    assert dummy_executor.parametrize('$foo') == 'bar'23@pytest.mark.parametrize("expr,expected", [24    ('{! variables["foo"].upper() !}', 'BAR')])25def test_executor_parametrize_expression(dummy_executor, expr, expected):26    assert dummy_executor.parametrize(27        expr) == expected28def test_execute(dummy_executor):29    execute_command_mock = mock.MagicMock()30    dummy_executor.execute_command = execute_command_mock31    yml_data = [32        {'provider': 'python', 'type': 'assert', 'expression': '1'},33        {'provider': 'python', 'type': 'assert', 'expression': '2'}34    ]35    dummy_executor.execute(yml_data)36    calls = [37        mock.call(yml_data[0]),38        mock.call(yml_data[1]),39    ]40    assert dummy_executor.execute_command.assert_has_calls(41        calls, any_order=False) is None42def test_execute_extra_vars(dummy_executor):43    execute_command_mock = mock.MagicMock()44    dummy_executor.execute_command = execute_command_mock45    yml_data = [46        {'provider': 'python', 'type': 'assert', 'expression': '1'},47        {'provider': 'python', 'type': 'assert',48         'expression': '2 == $does_not_exist'}49    ]50    assert 'does_not_exist' not in dummy_executor.variables51    dummy_executor.execute(yml_data, extra_variables={'does_not_exist': '2'})52    calls = [53        mock.call(yml_data[0]),54        mock.call(yml_data[1]),55    ]56    assert dummy_executor.execute_command.assert_has_calls(57        calls, any_order=False) is None58def test_execute_bad_type(dummy_executor):59    command = {'provider': 'python', 'typeXX': 'assert', 'expression': '1'}60    with pytest.raises(KeyError):61        dummy_executor.execute_command(command)62def test_execute_bad_command(dummy_executor):63    command = {'provider': 'python', 'type': 'assert', 'expressionXX': '1'}64    with pytest.raises(KeyError):65        dummy_executor.execute_command(command)66def test_execute_not_implemented_command(dummy_executor):67    command = {'provider': 'python', 'type': 'new_command',68               'param': 'http://1'}69    dummy_executor.COMMANDS = ['new_command']70    with pytest.raises(NotImplementedError):71        dummy_executor.execute_command(command)72def test_execute_condition_true(dummy_executor):73    command = {'provider': 'python',74               'type': 'assert',75               'expression': 'False',76               'skip_condition': '1 == 1'}77    dummy_executor.execute_command(command)78def test_execute_condition_false(dummy_executor):79    command = {'provider': 'python',80               'type': 'assert',81               'expression': 'True',82               'skip_condition': '1 == 0'}83    dummy_executor.execute_command(command)84def test_execute_get_basestring(dummy_executor):85    import yaml86    command = yaml.load("""87---88provider: python89type: assert90expression: 1 == 191    """)92    dummy_executor.execute_command(command)93def test_execute_get_basestring_param(dummy_executor):94    import yaml95    command = yaml.load("""96---97provider: python98type: assert99expression: "'$foo' == 'bar'"100""")101    dummy_executor.execute_command(command)102def test_new_provider_custom_command(dummy_executor):103    command = {'type': 'newCommand', 'provider': 'newprovider'}104    dummy_provider = mock.MagicMock()105    with pytest.raises(ValueError):106        dummy_executor.execute_command(command)107    dummy_executor.register_command_provider(108        dummy_provider, 'newprovider')109    # execute new custom command110    dummy_executor.execute_command(command)111    assert dummy_provider.assert_called_once_with(dummy_executor) is None112    assert dummy_provider \113        .return_value \114        .command_newCommand \115        .assert_called_once_with(command) is None116def test_execute_includes(dummy_executor, data_base_path):117    yml_data = [118        {'type': 'include', 'provider': 'include',119         'path': '{0}/{1}'.format(120             data_base_path, 'included.yml')},121        {'type': 'include', 'provider': 'include',122         'path': '{0}/{1}'.format(123             data_base_path, 'assertion.yml')},124    ]125    dummy_executor.execute(yml_data)126    assert dummy_executor.variables['included'] == 1127def test_default_command(play, data_base_path):128    play.variables['include'] = {'comment': 'default comment'}129    play.get_command_provider = mock.MagicMock()130    yml_data = [131        {"provider": "include", "type": "include",132         "path": "{0}/included.yml".format(data_base_path)},133    ]134    from copy import deepcopy135    expected_command = deepcopy(yml_data)[0]136    expected_command['comment'] = 'default comment'137    play.execute(yml_data)138    assert play \139        .get_command_provider \140        .return_value \141        .command_include \142        .assert_called_once_with(143            expected_command) is None144def test_default_command_override(play, data_base_path):145    play.variables['include'] = {'comment': 'default comment'}146    play.get_command_provider = mock.MagicMock()147    yml_data = [148        {"provider": "include", "type": "include",149         "comment": "override",150         "path": "{0}/included.yml".format(data_base_path)},151    ]152    from copy import deepcopy153    expected_command = deepcopy(yml_data)[0]154    expected_command['comment'] = 'override'155    play.execute(yml_data)156    assert play \157        .get_command_provider \158        .return_value \159        .command_include \160        .assert_called_once_with(161            expected_command) is None162def test_default_command_override_dict(play, data_base_path):163    play.variables['include'] = {164        'comment': {'comment': 'default comment'}}165    play.get_command_provider = mock.MagicMock()166    yml_data = [167        {"provider": "include", "type": "include",168         "comment": {"another": "override"},169         "path": "{0}/included.yml".format(data_base_path)},170    ]171    from copy import deepcopy172    expected_command = deepcopy(yml_data)[0]173    expected_command['comment'] = {174        'another': 'override', 'comment': 'default comment'}175    play.execute(yml_data)176    assert play \177        .get_command_provider \178        .return_value \179        .command_include \180        .assert_called_once_with(181            expected_command) is None182def test_default_command_override_dict_2(play, data_base_path):183    play.variables['include'] = {184        'comment': {'comment': 'default comment'}}185    play.get_command_provider = mock.MagicMock()186    yml_data = [187        {"provider": "include", "type": "include",188         "comment": {"another": "override", "comment": "other"},189         "path": "{0}/included.yml".format(data_base_path)},190    ]191    from copy import deepcopy192    expected_command = deepcopy(yml_data)[0]193    expected_command['comment'] = {194        'another': 'override', 'comment': 'other'}195    play.execute(yml_data)196    assert play \197        .get_command_provider \198        .return_value \199        .command_include \200        .assert_called_once_with(201            expected_command) is None202def test_default_command_override_dict_4(203        play, data_base_path):204    play.variables['include'] = {205        'comment': {'comment': 'default comment'}}206    play.get_command_provider = mock.MagicMock()207    yml_data = [208        {"provider": "include", "type": "include",209         "comment": "default comment",210         "path": "{0}/included.yml".format(data_base_path)},211    ]212    from copy import deepcopy213    expected_command = deepcopy(yml_data)[0]214    expected_command['comment'] = 'default comment'215    play.execute(yml_data)216    assert play \217        .get_command_provider \218        .return_value \219        .command_include \220        .assert_called_once_with(221            expected_command) is None222def test_default_command_override_dict_3(223        play, data_base_path):224    play.variables['include'] = {225        'comment': 'default comment'}226    play.get_command_provider = mock.MagicMock()227    yml_data = [228        {"provider": "include", "type": "include",229         "comment": {"another": "override", "comment": "other"},230         "path": "{0}/included.yml".format(data_base_path)},231    ]232    from copy import deepcopy233    expected_command = deepcopy(yml_data)[0]234    expected_command['comment'] = {235        'another': 'override', 'comment': 'other'}236    play.execute(yml_data)237    assert play \238        .get_command_provider \239        .return_value \240        .command_include \241        .assert_called_once_with(242            expected_command) is None243def test_include_string(play, data_base_path):244    play.variables['foo'] = 'bar'245    yml_data = """246---247- provider: include248  type: include249  path: "%s/included.yml"250- provider: python251  type: assert252  expression: "$included == 1"253- provider: python254  type: store_variable255  name: included256  expression: "2"257  comment: update included value from 1 to 2258- provider: python259  type: assert260  expression: "$included == 2"261    """ % data_base_path262    play.execute_raw(yml_data)263    assert play.variables['included'] == 2264def test_teardown(play):265    import mock266    play._teardown = [267        mock.MagicMock(side_effect=AttributeError()),268        mock.MagicMock()]269    play.teardown()270    assert play._teardown[0].assert_called_once_with() is None271    assert play._teardown[1].assert_called_once_with() is None272def test_elapsed_variable(play):273    command = {'provider': 'python',274               'type': 'assert',275               'expression': 'True', }276    assert '_elapsed' not in play.variables277    play.execute_command(command)...

test_executor.py

Source:test_executor.py

1import concurrent.futures2from unittest import mock3import pytest4from app import executor5from app.data import PageContent6@pytest.mark.parametrize(7    'articles,call_count',8    [9        [25, 1],10        [26, 2],11        [1, 1],12    ],13)14@mock.patch('app.executor.get_articles_links')15def test_submit_execution_main_pages(16    func_mock: mock.Mock,17    articles: int,18    call_count: int,19    dummy_executor: concurrent.futures.Executor,20) -> None:21    func_mock.return_value = func_mock22    for future in executor.submit_execution_main_pages(dummy_executor, articles):23        assert future.result() == func_mock24    assert func_mock.call_count == call_count25@mock.patch('app.executor.parse_article')26def test_submit_execution_articles(27    func_mock: mock.Mock, dummy_executor: concurrent.futures.Executor28):29    func_mock.return_value = func_mock30    futures = executor.submit_execution_articles(31        dummy_executor,32        [33            'https://example.com/link1',34            'https://example.com/link2',35        ],36    )37    assert func_mock.call_count == 238    assert func_mock.call_args_list == [39        mock.call('https://example.com/link1'),40        mock.call('https://example.com/link2'),41    ]42    for future in futures:43        assert future.result() == func_mock44@pytest.mark.parametrize(45    'articles,links',46    [47        [1, 1],48        [2, 1],49        [3, 4],50    ],51)52@mock.patch('app.executor.submit_execution_articles')53@mock.patch('app.executor.submit_execution_main_pages')54def test_scrape_concurrent(55    main_func_mock: mock.Mock,56    articles_func_mock: mock.Mock,57    dummy_executor: concurrent.futures.Executor,58    articles: int,59    links: int,60):61    future = concurrent.futures.Future()  # type: ignore[var-annotated]62    future.set_result([f'https://example.org/link{link+1}' for link in range(links)])63    main_func_mock.return_value = [future, future]64    future = concurrent.futures.Future()65    result = PageContent('title', 'text', ('image1', 'image2'))66    future.set_result(result)67    articles_func_mock.return_value = {future: 'link'}...

terminal_graph_node.py

Source:terminal_graph_node.py

1__author__ = 'Bohdan Mushkevych'2from flow.core.flow_graph_node import FlowGraphNode3from flow.core.simple_actions import IdentityAction4from flow.core.step_executor import StepExecutor, ACTIONSET_COMPLETE5class TerminalGraphNode(FlowGraphNode):6    """ represents a Terminal Node (start, finish) in the FlowGraph """7    def __init__(self, step_name):8        dummy_executor = StepExecutor(step_name, IdentityAction())9        dummy_executor.pre_actionset.state = ACTIONSET_COMPLETE10        dummy_executor.main_actionset.state = ACTIONSET_COMPLETE11        dummy_executor.post_actionset.state = ACTIONSET_COMPLETE...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.