icydb-base 0.0.12

IcyDB — A type-safe, embedded ORM and schema system for the Internet Computer
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

use crate::{core::traits::Sanitizer, prelude::*};

///
/// MimeType
/// Lowercases and trims whitespace.
///

#[sanitizer]
pub struct MimeType;

impl Sanitizer<String> for MimeType {
    fn sanitize(&self, value: String) -> String {
        value.trim().to_ascii_lowercase()
    }
}

///
/// Url
/// Trims whitespace and ensures a valid scheme (adds `https://` if missing).
///

#[sanitizer]
pub struct Url;

impl Sanitizer<String> for Url {
    fn sanitize(&self, value: String) -> String {
        let trimmed = value.trim();

        if trimmed.starts_with("http://") || trimmed.starts_with("https://") {
            trimmed.to_string()
        } else {
            format!("https://{trimmed}")
        }
    }
}

///
/// TESTS
///

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_mime_type_sanitize_trims_and_lowercases() {
        let sanitizer = MimeType;

        assert_eq!(sanitizer.sanitize("  Text/HTML  ".to_string()), "text/html");
        assert_eq!(
            sanitizer.sanitize("APPLICATION/JSON".to_string()),
            "application/json"
        );
        assert_eq!(sanitizer.sanitize(" image/JPEG ".to_string()), "image/jpeg");
    }

    #[test]
    fn test_url_sanitize_adds_https_when_missing() {
        let sanitizer = Url;

        assert_eq!(
            sanitizer.sanitize("example.com".to_string()),
            "https://example.com"
        );
        assert_eq!(
            sanitizer.sanitize(" www.example.com ".to_string()),
            "https://www.example.com"
        );
    }

    #[test]
    fn test_url_sanitize_keeps_existing_scheme() {
        let sanitizer = Url;

        assert_eq!(
            sanitizer.sanitize("https://example.com".to_string()),
            "https://example.com"
        );
        assert_eq!(
            sanitizer.sanitize("http://example.com".to_string()),
            "http://example.com"
        );
    }

    #[test]
    fn test_url_sanitize_trims_whitespace() {
        let sanitizer = Url;

        assert_eq!(
            sanitizer.sanitize("   https://example.com   ".to_string()),
            "https://example.com"
        );
        assert_eq!(
            sanitizer.sanitize("   example.com   ".to_string()),
            "https://example.com"
        );
    }
}