How to use test_format_request_var_dict method in tavern

Best Python code snippet using tavern

test_core.py

Source:test_core.py

...120        "params",121        "json",122        "headers",123    ))124    def test_format_request_var_dict(self, fulltest, mockargs, includes, request_key):125        """Variables from request should be available to format in response"""126        sent_value = str(uuid.uuid4())127        fulltest["stages"][0]["request"]["method"] = "POST"128        fulltest["stages"][0]["request"][request_key] = {"a_format_key": sent_value}129        if request_key == "json":130            resp_key = "body"131            mockargs[request_key] = lambda: {"returned": sent_value}132        else:133            resp_key = request_key134            mockargs[request_key] = {"returned": sent_value}135        fulltest["stages"][0]["response"][resp_key] = {"returned": "{tavern.request_vars.%s.a_format_key:s}" % request_key}136        mock_response = Mock(**mockargs)137        with patch("tavern.plugins.requests.Session.request", return_value=mock_response) as pmock:138            run_test("heif", fulltest, includes)139        assert pmock.called140    @pytest.mark.parametrize("request_key", (141        "url",142        "method",143    ))144    def test_format_request_var_value(self, fulltest, mockargs, includes, request_key):145        """Variables from request should be available to format in response"""146        sent_value = str(uuid.uuid4())147        fulltest["stages"][0]["request"]["method"] = "POST"148        fulltest["stages"][0]["request"][request_key] = sent_value149        resp_key = request_key150        mockargs[request_key] = {"returned": sent_value}151        fulltest["stages"][0]["response"][resp_key] = {"returned": "{tavern.request_vars.%s:s}" % request_key}152        mock_response = Mock(**mockargs)153        with patch("tavern.plugins.requests.Session.request", return_value=mock_response) as pmock:154            run_test("heif", fulltest, includes)155        assert pmock.called156class TestFormatMQTTVarsJson:157    """Test that formatting request vars from mqtt works as well, with json payload158    """159    @pytest.fixture(name="fulltest")160    def fix_mqtt_publish_test(self):161        spec = {162            "test_name": "An mqtt test with a single stage",163            "mqtt": {164                "connect": "localhost",165            },166            "stages": [167                {168                    "name": "step 1",169                    "mqtt_publish": {170                        "topic": "/abc/123",171                        "json": {172                            "message": str(uuid.uuid4()),173                        }174                    },175                    "mqtt_response": {176                        "topic": "{tavern.request_vars.topic}",177                        "json": {178                            "echo": "{tavern.request_vars.json.message}",179                        },180                    }181                }182            ]183        }184        return spec185    def test_format_request_var_dict(self, fulltest, includes):186        """Variables from request should be available to format in response -187        this is the original keys in the input file, NOT the formatted ones188        where 'json' is converted to 'payload' in the actual MQTT publish"""189        stage = fulltest["stages"][0]190        sent = stage["mqtt_publish"]["json"]191        mockargs = {192            "spec": paho.MQTTMessage,193            "payload": json.dumps({"echo": sent["message"]}).encode("utf8"),194            "topic": stage["mqtt_publish"]["topic"],195        }196        mock_response = Mock(**mockargs)197        fake_client = MagicMock(198            spec=MQTTClient,199            message_received=Mock(return_value=mock_response),...

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.