ftracker_identifiers/country/
serde.rs1use ::serde::de::{self, Visitor};
8use ::serde::{Deserialize, Deserializer, Serialize, Serializer};
9use core::fmt;
10
11use super::CountryCode;
12
13impl Serialize for CountryCode {
14 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
15 where
16 S: Serializer,
17 {
18 serializer.serialize_str(self.as_str())
19 }
20}
21
22struct CountryCodeVisitor;
23
24impl<'de> Visitor<'de> for CountryCodeVisitor {
25 type Value = CountryCode;
26
27 fn expecting(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
28 f.write_str("a 2-character ISO 3166-1 alpha-2 country code, e.g. US")
29 }
30
31 fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
32 where
33 E: de::Error,
34 {
35 CountryCode::parse(v).map_err(de::Error::custom)
36 }
37}
38
39impl<'de> Deserialize<'de> for CountryCode {
40 fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
41 where
42 D: Deserializer<'de>,
43 {
44 deserializer.deserialize_str(CountryCodeVisitor)
45 }
46}
47
48#[cfg(test)]
49mod tests {
50 use super::super::CountryCode;
51
52 #[test]
53 fn round_trips_through_json() {
54 let code = CountryCode::parse("US").unwrap();
55 let json = serde_json::to_string(&code).unwrap();
56 assert_eq!(json, "\"US\"");
57 let back: CountryCode = serde_json::from_str(&json).unwrap();
58 assert_eq!(code, back);
59 }
60
61 #[test]
62 fn rejects_invalid_json_string() {
63 let err = serde_json::from_str::<CountryCode>("\"ZZ\"").unwrap_err();
64 assert!(err.to_string().contains("ISO 3166-1"));
65 }
66}