coap_lite/
response.rs

1use crate::{
2    header::{MessageClass, MessageType, ResponseType as Status},
3    packet::Packet,
4};
5
6/// The CoAP response.
7#[derive(Clone, Debug, PartialEq)]
8pub struct CoapResponse {
9    pub message: Packet,
10}
11
12impl CoapResponse {
13    /// Creates a new response.
14    pub fn new(request: &Packet) -> Option<CoapResponse> {
15        let mut packet = Packet::new();
16
17        packet.header.set_version(1);
18        let response_type = match request.header.get_type() {
19            MessageType::Confirmable => MessageType::Acknowledgement,
20            MessageType::NonConfirmable => MessageType::NonConfirmable,
21            _ => return None,
22        };
23        packet.header.set_type(response_type);
24        packet.header.code = MessageClass::Response(Status::Content);
25        packet.header.message_id = request.header.message_id;
26        packet.set_token(request.get_token().to_vec());
27
28        Some(CoapResponse { message: packet })
29    }
30
31    /// Sets the status.
32    pub fn set_status(&mut self, status: Status) {
33        self.message.header.code = MessageClass::Response(status);
34    }
35
36    /// Returns the status.
37    pub fn get_status(&self) -> &Status {
38        match self.message.header.code {
39            MessageClass::Response(ref code) => code,
40            _ => &Status::UnKnown,
41        }
42    }
43}
44
45#[cfg(test)]
46mod test {
47    use super::*;
48
49    #[test]
50    fn test_new_response_valid() {
51        for mtyp in [MessageType::Confirmable, MessageType::NonConfirmable] {
52            let mut packet = Packet::new();
53            packet.header.set_type(mtyp);
54            let opt_resp = CoapResponse::new(&packet);
55            assert!(opt_resp.is_some());
56
57            let response = opt_resp.unwrap();
58            assert_eq!(packet.payload, response.message.payload);
59        }
60    }
61
62    #[test]
63    fn test_new_response_invalid() {
64        let mut packet = Packet::new();
65        packet.header.set_type(MessageType::Acknowledgement);
66        assert!(CoapResponse::new(&packet).is_none());
67    }
68}
coap_lite/response.rs

coap_lite/
response.rs
