Support enum unit variants as map keys (#37)

2 files changed, 43 insertions, 0 deletions

diff --git a/tests/test_deserialize.rs b/tests/test_deserialize.rs
index 8d518ee..f0ccee9 100644
--- a/tests/test_deserialize.rs
+++ b/tests/test_deserialize.rs
@@ -640,5 +640,23 @@ fn deserialize_plus() {
 
 #[test]
 fn deserialize_vec_of_structs() {
+}
+
+#[test]
+fn deserialize_map_with_unit_enum_keys() {
+    #[derive(Deserialize, Eq, PartialEq, Hash)]
+    enum Operator {
+        Lt,
+        Gt,
+    }
+
+    #[derive(Deserialize)]
+    struct Filter {
+        point: HashMap<Operator, u64>,
+    }
+
+    let test: Filter = serde_qs::from_str("point[Gt]=123&point[Lt]=321").unwrap();
 
+    assert_eq!(test.point[&Operator::Gt], 123);
+    assert_eq!(test.point[&Operator::Lt], 321);
 }
diff --git a/tests/test_serialize.rs b/tests/test_serialize.rs
index f650beb..8aecfe3 100644
--- a/tests/test_serialize.rs
+++ b/tests/test_serialize.rs
@@ -131,3 +131,28 @@ fn serialize_flatten() {
     let rec_params = qs::to_string(&query).unwrap();
     assert_eq!(rec_params, params);
 }
+
+#[test]
+fn serialize_map_with_unit_enum_keys() {
+    use std::collections::HashMap;
+
+    #[derive(Serialize, Eq, PartialEq, Hash)]
+    enum Operator {
+        Lt,
+        Gt,
+    }
+
+    #[derive(Serialize)]
+    struct Filter {
+        point: HashMap<Operator, u64>,
+    }
+
+    let mut map = HashMap::new();
+    map.insert(Operator::Gt, 123);
+    map.insert(Operator::Lt, 321);
+    let test = Filter { point: map };
+
+    let query = qs::to_string(&test).unwrap();
+
+    assert!(query == "point[Lt]=321&point[Gt]=123" || query == "point[Gt]=123&point[Lt]=321");
+}