Unit tests

dmwm · Dec 2, 2022 · f7060c6 · f7060c6
1 parent e98ff86
commit f7060c6
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/python/WMCore_t/MicroService_t/MSRuleCleaner_t/MSRuleCleaner_t.py b/test/python/WMCore_t/MicroService_t/MSRuleCleaner_t/MSRuleCleaner_t.py
@@ -46,6 +46,7 @@ def setUp(self):
                          "logDBUrl": "https://cmsweb-testbed.cern.ch/couchdb/wmstats_logdb",
                          'logDBReporter': 'reqmgr2ms_ruleCleaner',
                          'archiveDelayHours': 8,
+                         'archiveAlarmHours': 24,
                          'enableRealMode': False}
 
         self.creds = {"client_cert": os.getenv("X509_USER_CERT", "Unknown"),
@@ -78,6 +79,15 @@ def setUp(self):
             self.stepChainReq = json.load(fd)
         super(MSRuleCleanerTest, self).setUp()
 
+    def testGetLastStatusTransitionTime(self):
+        wflow = MSRuleCleanerWflow(self.taskChainReq)
+        lastStatusTransition = self.msRuleCleaner._getLastStatusTransitionTime(wflow)
+        self.assertEqual(lastStatusTransition, 1607359514)
+
+    def testIsStatusAdvanceExpired(self):
+        wflow = MSRuleCleanerWflow(self.taskChainReq)
+        self.assertTrue(self.msRuleCleaner._isStatusAdvanceExpired(wflow))
+
     def testPipelineAgentBlock(self):
         # Test plineAgentBlock:
         wflow = MSRuleCleanerWflow(self.taskChainReq)