From 7e747e87f81f3c3d421c83f67a0723532e4a9c66 Mon Sep 17 00:00:00 2001
From: Valentin Kuznetsov <vkuznet@gmail.com>
Date: Mon, 20 Mar 2023 11:40:14 -0400
Subject: [PATCH] Unit test adjustment for MSPileup clean-up task

---
 .../WMCore_t/MicroService_t/MSPileup_t/MSPileupTasks_t.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/test/python/WMCore_t/MicroService_t/MSPileup_t/MSPileupTasks_t.py b/test/python/WMCore_t/MicroService_t/MSPileup_t/MSPileupTasks_t.py
index 227762d4a4..8cc0889f18 100644
--- a/test/python/WMCore_t/MicroService_t/MSPileup_t/MSPileupTasks_t.py
+++ b/test/python/WMCore_t/MicroService_t/MSPileup_t/MSPileupTasks_t.py
@@ -227,6 +227,14 @@ def testMSPileupTasksWithMockApi(self):
                 found = True
         self.assertEqual(found, True)
 
+        # update doc in MSPileup and call cleanup task to delete it
+        data['active'] = False
+        data['rulesIds'] = []
+        data['currentRSEs'] = []
+        data['deactivatedOn'] = 0
+        self.mgr.updatePileup(data)
+        obj.cleanupTask(0)
+
 
 if __name__ == '__main__':
     unittest.main()