multiio 0.2.3

A unified I/O orchestration library for CLI/server applications
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

//! Custom format tests.

use crate::format::{CustomFormat, FormatError, FormatKind, FormatRegistry};
use serde::{Deserialize, Serialize};

#[derive(Debug, Serialize, Deserialize, PartialEq)]
struct TestData {
    name: String,
    value: i32,
}

#[test]
fn test_custom_format_registration() {
    let mut registry = FormatRegistry::new();

    // Register a custom format
    let custom = CustomFormat::new("test-format", &["tf"])
        .with_deserialize(|bytes| {
            serde_json::from_slice(bytes).map_err(|e| FormatError::Serde(Box::new(e)))
        })
        .with_serialize(|value| {
            serde_json::to_vec(value).map_err(|e| FormatError::Serde(Box::new(e)))
        });

    registry.register_custom(custom);

    // Check registration
    assert!(registry.has_format(&FormatKind::Custom("test-format")));
    assert!(registry.get_custom("test-format").is_some());
}

#[test]
fn test_custom_format_extension_lookup() {
    let mut registry = FormatRegistry::new();

    let custom = CustomFormat::new("my-format", &["myf", "myfmt"])
        .with_deserialize(|bytes| {
            serde_json::from_slice(bytes).map_err(|e| FormatError::Serde(Box::new(e)))
        })
        .with_serialize(|value| {
            serde_json::to_vec(value).map_err(|e| FormatError::Serde(Box::new(e)))
        });

    registry.register_custom(custom);

    // Test extension lookup
    let kind = registry.kind_for_extension("myf");
    assert_eq!(kind, Some(FormatKind::Custom("my-format")));

    let kind = registry.kind_for_extension("myfmt");
    assert_eq!(kind, Some(FormatKind::Custom("my-format")));
}

#[test]
fn test_custom_format_serialize_deserialize() {
    let mut registry = FormatRegistry::new();

    // Create a custom format that wraps data in brackets
    let bracket_format = CustomFormat::new("bracket", &["brk"])
        .with_deserialize(|bytes| {
            // Remove brackets and parse as JSON
            let s = String::from_utf8_lossy(bytes);
            let inner = s.trim_start_matches('[').trim_end_matches(']');
            serde_json::from_str(inner).map_err(|e| FormatError::Serde(Box::new(e)))
        })
        .with_serialize(|value| {
            let json = serde_json::to_string(value).map_err(|e| FormatError::Serde(Box::new(e)))?;
            Ok(format!("[{}]", json).into_bytes())
        });

    registry.register_custom(bracket_format);

    // Test serialization
    let data = TestData {
        name: "test".to_string(),
        value: 42,
    };

    let bytes = registry
        .serialize_value(Some(&FormatKind::Custom("bracket")), &[], &data)
        .unwrap();

    let output = String::from_utf8(bytes).unwrap();
    assert!(output.starts_with('['));
    assert!(output.ends_with(']'));

    // Test deserialization
    let result: TestData = registry
        .deserialize_value(Some(&FormatKind::Custom("bracket")), &[], output.as_bytes())
        .unwrap();

    assert_eq!(result, data);
}

#[test]
fn custom_format_without_deserialize_errors() {
    let fmt = CustomFormat::new("no-deser", &["and"]).with_serialize(|value| {
        serde_json::to_vec(value).map_err(|e| FormatError::Serde(Box::new(e)))
    });

    let err = fmt
        .deserialize::<serde_json::Value>(b"{}")
        .expect_err("expected error when deserializing without handler");

    match err {
        FormatError::Other(inner) => {
            let msg = inner.to_string();
            assert!(
                msg.contains("does not support deserialization"),
                "unexpected message: {}",
                msg
            );
        }
        other => panic!("expected FormatError::Other, got: {other:?}"),
    }
}

#[test]
fn custom_format_without_serialize_errors() {
    let fmt = CustomFormat::new("no-serial", &["ns"]).with_deserialize(|bytes| {
        serde_json::from_slice(bytes).map_err(|e| FormatError::Serde(Box::new(e)))
    });

    let value = TestData {
        name: "x".to_string(),
        value: 1,
    };

    let err = fmt
        .serialize(&value)
        .expect_err("expected error when serializing without handler");

    match err {
        FormatError::Other(inner) => {
            let msg = inner.to_string();
            assert!(
                msg.contains("does not support serialization"),
                "unexpected message: {}",
                msg
            );
        }
        other => panic!("expected FormatError::Other, got: {other:?}"),
    }
}