Another test

Signed-off-by: Benjamin Leggett <[email protected]>

Author: bleggett

src/inpod/statemanager.rs

@@ -612,6 +612,47 @@ mod tests {
         state.drain().await;
     }
 
+    #[tokio::test]
+    async fn del_workload_before_snapshot_removes_from_pending() {
+        let fixture = fixture!();
+        let mut state = fixture.state;
+
+        let ns = new_netns();
+
+        // to make the proxy fail, bind to its ports in its netns
+        let _sock = create_proxy_conflict(&ns);
+
+        let data = WorkloadData {
+            netns: ns,
+            workload_uid: uid(0),
+            workload_info: workload_info(),
+        };
+
+        let ret = state
+            .process_msg(WorkloadMessage::AddWorkload(data))
+            .await;
+
+        assert!(state.snapshot_names.len() == 1);
+        assert!(ret.is_err());
+        assert!(state.have_pending());
+
+        state
+            .process_msg(WorkloadMessage::DelWorkload(uid(0)))
+            .await
+            .unwrap();
+
+        assert!(state.snapshot_names.is_empty());
+
+        state
+            .process_msg(WorkloadMessage::WorkloadSnapshotSent)
+            .await
+            .unwrap();
+
+        assert!(state.snapshot_names.is_empty());
+        assert!(!state.have_pending());
+        state.drain().await;
+    }
+
     #[tokio::test]
     async fn add_delete_add_workload_starts_only_one_proxy() {
         let fixture = fixture!();