polywrap_tests_utils 0.1.11

Utilities for testing Polywrap Client
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

use std::{fmt::Debug, sync::Arc};

use polywrap_core::{
    invoker::Invoker,
    wrapper::{GetFileOptions, Wrapper},
};

#[derive(Debug)]
pub struct MockWrapper;
#[derive(Debug)]
pub struct DifferentMockWrapper;

impl Wrapper for MockWrapper {
    fn invoke(
        &self,
        method: &str,
        _: Option<&[u8]>,
        _: Option<&[u8]>,
        _: Arc<dyn Invoker>,
    ) -> Result<Vec<u8>, polywrap_core::error::Error> {
        // In Msgpack: True = [195] and False = [194]
        if method == "foo" {
            Ok(vec![195])
        } else {
            Ok(vec![194])
        }
    }

    fn get_file(&self, _: &GetFileOptions) -> Result<Vec<u8>, polywrap_core::error::Error> {
        Ok(vec![1])
    }
}

impl Wrapper for DifferentMockWrapper {
    fn invoke(
        &self,
        method: &str,
        _: Option<&[u8]>,
        _: Option<&[u8]>,
        _: Arc<dyn Invoker>,
    ) -> Result<Vec<u8>, polywrap_core::error::Error> {
        // In Msgpack: True = [195] and False = [194]
        if method == "bar" {
            Ok(vec![195])
        } else {
            Ok(vec![194])
        }
    }

    fn get_file(&self, _: &GetFileOptions) -> Result<Vec<u8>, polywrap_core::error::Error> {
        Ok(vec![1])
    }
}

pub fn get_mock_wrapper() -> Arc<dyn Wrapper> {
    Arc::new(MockWrapper {})
}

pub fn get_different_mock_wrapper() -> Arc<dyn Wrapper> {
    Arc::new(DifferentMockWrapper {})
}