Introduce a way to update a single policy definition key

Author: michaelklishin

src/responses.rs

@@ -1097,6 +1097,18 @@ impl PolicyDefinition {
     pub fn is_empty(&self) -> bool {
         self.len() == 0
     }
+
+    pub fn insert(&mut self, key: String, value: serde_json::Value) -> Option<serde_json::Value> {
+        match self.0 {
+            Some(ref mut m) => m.insert(key, value),
+            None => {
+                let mut m = Map::new();
+                m.insert(key, value);
+                self.0 = Some(m);
+                None
+            }
+        }
+    }
 }
 
 impl PolicyDefinitionOps for PolicyDefinition {
@@ -1157,6 +1169,10 @@ pub struct Policy {
 }
 
 impl Policy {
+    pub fn insert_definition_key(&mut self, key: String, value: serde_json::Value) -> Option<serde_json::Value> {
+        self.definition.insert(key, value)
+    }
+
     pub fn definition_keys(&self) -> Vec<&str> {
         match &self.definition.0 {
             None => Vec::new(),

tests/unit_policy_tests.rs

@@ -232,6 +232,51 @@ fn test_unit_policy_definition_without_cmq_keys_case1() {
     assert!(m2.contains_key(&k6));
 }
 
+#[test]
+fn test_unit_policy_definition_insert_case1() {
+    let k1 = "max-age".to_owned();
+    let k2 = "max-length-bytes".to_owned();
+
+    let mut m = Map::new();
+    m.insert(k1.clone(), json!("1D"));
+
+    let mut defs = PolicyDefinition(Some(m));
+    assert_eq!(1, defs.len());
+
+    defs.insert(k1.clone(), json!("2D"));
+    assert_eq!(1, defs.len());
+
+    defs.insert(k2.clone(), json!(1000000));
+    assert_eq!(2, defs.len());
+}
+
+#[test]
+fn test_unit_policy_insert_definition_key_case1() {
+    let k1 = "max-age".to_owned();
+    let k2 = "max-length-bytes".to_owned();
+
+    let mut m = Map::new();
+    m.insert(k1.clone(), json!("1D"));
+
+    let defs = PolicyDefinition(Some(m));
+    let mut pol = Policy {
+        name: "test_unit_policy_insert_case1".to_owned(),
+        vhost: "/".to_owned(),
+        pattern: ".*".to_owned(),
+        apply_to: PolicyTarget::Queues,
+        priority: 11,
+        definition: defs.clone(),
+    };
+
+    assert_eq!(1, pol.definition.len());
+
+    pol.insert_definition_key(k1.clone(), json!("2D"));
+    assert_eq!(1, pol.definition.len());
+
+    pol.insert_definition_key(k2.clone(), json!(1000000));
+    assert_eq!(2, pol.definition.len());
+}
+
 #[test]
 fn test_unit_policy_does_match_case1() {
     let mut m = Map::new();