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
65
66
67
68
69

/* 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 serde::{Deserialize, Serialize};

use crate::common::BoolValue;

#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
#[allow(non_camel_case_types)]
pub enum AppStatus {
    eAttemptQuit,
    eConsiderQuit,
    eForceQuit,
    eRestart,
}

#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub enum Command {
    #[serde(rename = "Marionette:AcceptConnections")]
    AcceptConnections(BoolValue),
    #[serde(rename = "Marionette:Quit")]
    DeleteSession { flags: Vec<AppStatus> },
    #[serde(rename = "Marionette:GetContext")]
    GetContext,
    #[serde(rename = "Marionette:GetScreenOrientation")]
    GetScreenOrientation,
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::test::assert_ser_de;
    use serde_json::json;

    #[test]
    fn test_json_command_accept_connections() {
        assert_ser_de(
            &Command::AcceptConnections(BoolValue::new(false)),
            json!({"Marionette:AcceptConnections": {"value": false }}),
        );
    }

    #[test]
    fn test_json_command_delete_session() {
        let data = &Command::DeleteSession {
            flags: vec![AppStatus::eForceQuit],
        };
        assert_ser_de(data, json!({"Marionette:Quit": {"flags": ["eForceQuit"]}}));
    }

    #[test]
    fn test_json_command_get_context() {
        assert_ser_de(&Command::GetContext, json!("Marionette:GetContext"));
    }

    #[test]
    fn test_json_command_get_screen_orientation() {
        assert_ser_de(
            &Command::GetScreenOrientation,
            json!("Marionette:GetScreenOrientation"),
        );
    }

    #[test]
    fn test_json_command_invalid() {
        assert!(serde_json::from_value::<Command>(json!("foo")).is_err());
    }
}