summaryrefslogtreecommitdiffstats
path: root/mqtt-tester/src/behaviour/first_packet_from_client_is_connect.rs
blob: 1315a630b9bed8c5638dd86084b05f4968d93a3a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
//
//   This Source Code Form is subject to the terms of the Mozilla Public
//   License, v. 2.0. If a copy of the MPL was not distributed with this
//   file, You can obtain one at http://mozilla.org/MPL/2.0/.
//

use mqtt_format::v3::packet::MPacket;

use crate::{
    behaviour_test::BehaviourTest,
    command::{Input, Output},
    executable::ClientExecutableCommand,
    report::ReportResult,
};

pub struct FirstPacketFromClientIsConnect;

#[async_trait::async_trait]
impl BehaviourTest for FirstPacketFromClientIsConnect {
    fn commands(&self) -> Vec<Box<dyn ClientExecutableCommand>> {
        vec![]
    }

    #[tracing::instrument(skip_all)]
    async fn execute(&self, _input: Input, mut output: Output) -> Result<(), miette::Error> {
        output
            .wait_and_check(
                &(|bytes: &[u8]| -> bool {
                    let packet = match nom::combinator::all_consuming(
                        mqtt_format::v3::packet::mpacket,
                    )(bytes)
                    {
                        Ok((_, packet)) => packet,
                        Err(_e) => return false,
                    };

                    std::matches!(packet, MPacket::Connect { .. })
                }),
            )
            .await?;

        Ok(())
    }

    fn report_name(&self) -> &str {
        "First packet send by client must be CONNECT"
    }

    fn report_desc(&self) -> &str {
        "After a Network Connection is established by a Client to a Server, the first Packet sent from the Client to the Server MUST be a CONNECT Packet."
    }

    fn report_normative(&self) -> &str {
        "[MQTT-3.1.0-1]"
    }

    fn translate_client_exit_code(&self, success: bool) -> ReportResult {
        if success {
            ReportResult::Failure
        } else {
            ReportResult::Success
        }
    }
}