def test_not_enabled(self): t = TaskTimerStats() t.enabled = False self.assertFalse(t.publish(isnot="enabled")) self.assertFalse(getattr(t, "time_start", None)) t.run("foo", "bar", [], {}) t.stop()
def assertTimeElapsed(self, time_sleep, max_appx, min_appx, appx): t = TaskTimerStats() t.enabled = True t.run("foo", "bar", [], {}) self.assertTrue(t.time_start) time.sleep(time_sleep) time_stop = t.stop() self.assertTrue(time_stop) self.assertFalse(time_stop > max_appx) self.assertFalse(time_stop <= min_appx) strstop = str(time_stop)[0:3] # Time elapsed is approximately 0.1 seconds. self.assertTrue(strstop == appx)
def test_collect_report_dump(self): timer1 = TaskTimerStats() timer1.enabled = True timer1.run("foo", "bar", [], {}) timer2 = TaskTimerStats() timer2.enabled = True timer2.run("foo", "bar", [], {}) timer3 = TaskTimerStats() timer3.enabled = True timer3.run("foo", "bar", [], {}) for timer in (timer1, timer2, timer3): timer.stop() # Collect self.s.collect() self.assertEquals(self.s.total_tasks_processed, 3) # Report with OverrideStdout() as outs: stdout, stderr = outs self.s.report() self.assertTrue( "Total processing time by task type:" in stdout.getvalue()) # Dump to cache self.s.dump_to_cache()