Fix queue JSON serialisation when flag is none

namib-project · Jul 7, 2023 · 4f42782 · 4f42782
1 parent d08001f
commit 4f42782
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/stmt.rs b/src/stmt.rs
@@ -384,6 +384,7 @@ pub struct Queue {
     /// Queue number.
     pub num: Expression,
 
+    #[serde(skip_serializing_if = "Option::is_none")]
     /// Queue flags.
     pub flags: Option<HashSet<QueueFlag>>,
 }

diff --git a/tests/json_tests.rs b/tests/json_tests.rs
@@ -154,3 +154,14 @@ fn test_parsing_of_queue_without_flags() {
     let parsed: Nftables = serde_json::from_value(json).unwrap();
     assert_eq!(expected, parsed);
 }
+
+#[test]
+fn test_queue_json_serialisation() {
+    let queue = Statement::Queue(Queue {
+        num: Expression::Number(0),
+        flags: None,
+    });
+
+    let expected_json = String::from(r#"{"queue":{"num":0}}"#);
+    assert_eq!(expected_json, serde_json::to_string(&queue).unwrap());
+}