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

How to use test_format_request_var_value method in tavern

Best Python code snippet using tavern

test_core.py

Source:test_core.py

...140    @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),200        )201        with patch("tavern.mqtt.paho.Client", fake_client), \202        patch("tavern.core.get_extra_sessions", return_value={"mqtt": fake_client}) as pmock:203            run_test("heif", fulltest, includes)204        assert pmock.called205class TestFormatMQTTVarsPlain:206    """Test that formatting request vars from mqtt works as well, with normal payload207    """208    @pytest.fixture(name="fulltest")209    def fix_mqtt_publish_test(self):210        spec = {211            "test_name": "An mqtt test with a single stage",212            "mqtt": {213                "connect": "localhost",214            },215            "stages": [216                {217                    "name": "step 1",218                    "mqtt_publish": {219                        "topic": "/abc/123",220                        "payload": "hello",221                    },222                    "mqtt_response": {223                        "topic": "{tavern.request_vars.topic}",224                        "payload": "{tavern.request_vars.payload}",225                    }226                }227            ]228        }229        return spec230    def test_format_request_var_value(self, fulltest, includes):231        """Same as above but with plain keys"""232        stage = fulltest["stages"][0]233        sent = stage["mqtt_publish"]["payload"]234        mockargs = {235            "spec": paho.MQTTMessage,236            "payload": sent.encode("utf8"),237            "topic": stage["mqtt_publish"]["topic"],238        }239        mock_response = Mock(**mockargs)240        fake_client = MagicMock(241            spec=MQTTClient,242            message_received=Mock(return_value=mock_response),243        )244        with patch("tavern.mqtt.paho.Client", fake_client), \...

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.