tauri-plugin-wdio 1.0.0-next.0

A Tauri plugin for WebDriverIO testing with execute and mocking capabilities
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

use std::collections::HashMap;
use serde_json::Value as JsonValue;
use crate::models::MockConfig;

/// Thread-safe mock registry for storing command mocks
pub struct MockStore {
    mocks: HashMap<String, MockConfig>,
    original_handlers: HashMap<String, JsonValue>, // Store original command handlers if needed
}

impl MockStore {
    pub fn new() -> Self {
        Self {
            mocks: HashMap::new(),
            original_handlers: HashMap::new(),
        }
    }

    pub fn set_mock(&mut self, command: String, config: MockConfig) {
        self.mocks.insert(command, config);
    }

    pub fn get_mock(&self, command: &str) -> Option<&MockConfig> {
        self.mocks.get(command)
    }

    pub fn clear_mocks(&mut self) {
        self.mocks.clear();
    }

    pub fn reset_mocks(&mut self) {
        self.mocks.clear();
        self.original_handlers.clear();
    }

    #[allow(dead_code)]
    pub fn remove_mock(&mut self, command: &str) -> Option<MockConfig> {
        self.mocks.remove(command)
    }

    #[allow(dead_code)]
    pub fn get_all_mocks(&self) -> &HashMap<String, MockConfig> {
        &self.mocks
    }
}