def test_exit_no_post(self, mock_time, mock_print): mock_time.return_value = 456 stats = ht.events.stats.HoudiniEventStats(None) stats._post_report = False stats._last_run_time = 0 stats._total_time = 0 stats._last_started = 123 stats._run_count = 0 exc_type = MagicMock() exc_val = MagicMock() exc_tb = MagicMock() stats.__exit__(exc_type, exc_val, exc_tb) run_time = 456-123 self.assertEqual(stats.last_run_time, run_time) self.assertEqual(stats.total_time, run_time) self.assertEqual(stats.run_count, 1) mock_print.assert_not_called()
def test___exit___no_report(self, init_stats, mocker): mock_time = mocker.patch("time.time") mock_print = mocker.patch.object(ht.events.stats.HoudiniEventStats, "print_report") stats = init_stats() stats._post_report = False mock_run_time = mocker.MagicMock(spec=int) mock_time.return_value = mock_run_time mock_total_time = mocker.MagicMock(spec=int) mock_last_run_time = mocker.MagicMock(spec=int) mock_last_started = mocker.MagicMock(spec=int) mock_run_count = mocker.MagicMock(spec=int) stats._last_run_time = mock_last_run_time stats._total_time = mock_total_time stats._last_started = mock_last_started stats._run_count = mock_run_count exc_type = mocker.MagicMock() exc_val = mocker.MagicMock() exc_tb = mocker.MagicMock() stats.__exit__(exc_type, exc_val, exc_tb) run_time = mock_run_time - mock_last_started assert stats.last_run_time == run_time assert stats.total_time == mock_total_time + run_time assert stats.run_count == mock_run_count + 1 mock_print.assert_not_called()
def test_exit_no_post(self, mock_time, mock_print): mock_time.return_value = 456 stats = ht.events.stats.HoudiniEventStats(None) stats._post_report = False stats._last_run_time = 0 stats._total_time = 0 stats._last_started = 123 stats._run_count = 0 exc_type = MagicMock() exc_val = MagicMock() exc_tb = MagicMock() stats.__exit__(exc_type, exc_val, exc_tb) run_time = 456-123 self.assertEqual(stats.last_run_time, run_time) self.assertEqual(stats.total_time, run_time) self.assertEqual(stats.run_count, 1) mock_print.assert_not_called()