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

How to use fix_mqtt_publish_test method in tavern

Best Python code snippet using tavern

test_core.py

Source:test_core.py

...156class 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}",...

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.