pywatt_sdk 0.5.3

Standardized SDK for building PyWatt modules in Rust
Documentation 
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

#[cfg(test)]
mod tests {
    use pywatt_sdk::secret_client::SecretClient;
    use pywatt_sdk::state::AppState;
    use std::sync::Arc;

    #[test]
    fn test_appstate_basic_functionality() {
        let module_id = "test-module".to_string();
        let orchestrator_api = "http://localhost:8000".to_string();
        let secret_client = Arc::new(SecretClient::new_dummy());
        let user_state = "Test user state".to_string();

        let state = AppState::new(
            module_id.clone(),
            orchestrator_api.clone(),
            secret_client.clone(),
            user_state.clone(),
        );

        // Test getters
        assert_eq!(state.module_id(), module_id);
        assert_eq!(state.orchestrator_api(), orchestrator_api);

        // Can't directly compare Arc<SecretClient> for equality since SecretClient doesn't implement PartialEq
        // Instead, check if they point to the same instance
        assert!(Arc::ptr_eq(state.secret_client(), &secret_client));

        assert_eq!(state.user_state, user_state);
    }

    #[test]
    fn test_appstate_clone() {
        let module_id = "test-module".to_string();
        let orchestrator_api = "http://localhost:8000".to_string();
        let secret_client = Arc::new(SecretClient::new_dummy());
        let user_state = "Test user state".to_string();

        let state = AppState::new(
            module_id.clone(),
            orchestrator_api.clone(),
            secret_client.clone(),
            user_state.clone(),
        );

        let cloned_state = state.clone();

        // Test that both states have the same content
        assert_eq!(cloned_state.module_id(), state.module_id());
        assert_eq!(cloned_state.orchestrator_api(), state.orchestrator_api());
        assert_eq!(cloned_state.user_state, state.user_state);

        // Test that the secret_client is the same instance (Arc comparison)
        assert!(Arc::ptr_eq(
            cloned_state.secret_client(),
            state.secret_client()
        ));
    }

    #[cfg(feature = "builder")]
    #[test]
    fn test_appstate_builder() {
        // Import builder directly
        let builder = pywatt_sdk::builder::AppStateBuilder::<String>::new();

        let state = builder
            .with_module_id("test-module".to_string())
            .with_orchestrator_api("http://localhost:8000".to_string())
            .with_secret_client(Arc::new(SecretClient::new_dummy()))
            .with_custom("Test user state".to_string())
            .build()
            .unwrap();

        assert_eq!(state.module_id(), "test-module");
        assert_eq!(state.orchestrator_api(), "http://localhost:8000");
        assert_eq!(state.user_state, "Test user state");
    }

    // Test custom state types
    #[test]
    fn test_appstate_with_custom_state() {
        #[derive(Clone, Debug, PartialEq)]
        struct MyState {
            counter: i32,
            name: String,
        }

        let module_id = "test-module".to_string();
        let orchestrator_api = "http://localhost:8000".to_string();
        let secret_client = Arc::new(SecretClient::new_dummy());
        let user_state = MyState {
            counter: 42,
            name: "Test".to_string(),
        };

        let state = AppState::new(
            module_id.clone(),
            orchestrator_api.clone(),
            secret_client.clone(),
            user_state.clone(),
        );

        assert_eq!(state.user_state, user_state);
        assert_eq!(state.user_state.counter, 42);
        assert_eq!(state.user_state.name, "Test");
    }
}