Fix bug in deserializing unknown features (#823)

divviup · Feb 22, 2024 · 4ed8096 · 4ed8096
1 parent 3dfccbb
commit 4ed8096
Showing 1 changed file with 68 additions and 0 deletions.
diff --git a/src/entity/aggregator/feature.rs b/src/entity/aggregator/feature.rs
@@ -4,6 +4,7 @@ use std::collections::HashSet;
 #[derive(Deserialize, Serialize, Debug, Clone, PartialEq, Eq, Hash)]
 pub enum Feature {
     TokenHash,
+    #[serde(untagged)]
     Unknown(String),
 }
 
@@ -27,3 +28,70 @@ impl FromIterator<Feature> for Features {
         Self(iter.into_iter().collect())
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use serde_json::json;
+
+    use crate::entity::aggregator::{Feature, Features};
+
+    #[test]
+    fn features_none() {
+        assert_eq!(
+            serde_json::from_value::<Features>(json!([])).unwrap(),
+            Features::from_iter([])
+        )
+    }
+
+    #[test]
+    fn features_token_hash() {
+        assert_eq!(
+            serde_json::from_value::<Features>(json!(["TokenHash"])).unwrap(),
+            Features::from_iter([Feature::TokenHash])
+        );
+        assert_eq!(
+            serde_json::from_value::<Features>(json!(["TokenHash", "TokenHash", "TokenHash"]))
+                .unwrap(),
+            Features::from_iter([Feature::TokenHash])
+        );
+    }
+
+    #[test]
+    fn features_unknown() {
+        assert_eq!(
+            serde_json::from_value::<Features>(json!(["UnspecifiedUnknownFeature"])).unwrap(),
+            Features::from_iter([Feature::Unknown("UnspecifiedUnknownFeature".to_string())])
+        );
+
+        assert_eq!(
+            serde_json::from_value::<Features>(json!([
+                "UnspecifiedUnknownFeature",
+                "UnspecifiedUnknownFeature",
+                "UnspecifiedUnknownFeature"
+            ]))
+            .unwrap(),
+            Features::from_iter([Feature::Unknown("UnspecifiedUnknownFeature".to_string())])
+        );
+
+        assert_eq!(
+            serde_json::from_value::<Features>(json!([
+                "UnspecifiedUnknownFeature",
+                "AnotherUnspecifiedUnknownFeature"
+            ]))
+            .unwrap(),
+            Features::from_iter([
+                Feature::Unknown("UnspecifiedUnknownFeature".to_string()),
+                Feature::Unknown("AnotherUnspecifiedUnknownFeature".to_string())
+            ])
+        );
+
+        assert_eq!(
+            serde_json::from_value::<Features>(json!(["UnspecifiedUnknownFeature", "TokenHash"]))
+                .unwrap(),
+            Features::from_iter([
+                Feature::TokenHash,
+                Feature::Unknown("UnspecifiedUnknownFeature".to_string())
+            ])
+        );
+    }
+}