def test_prune(self, mock_flush, mock_remove, mock_expired, mock_datetime, mock_index): self.build_mock_index_and_datetime_now(mock_index, mock_datetime) cache = Cache('test') mock_flush.reset_mock() mock_remove.reset_mock() mock_expired.reset_mock() mock_expired.return_value = True cache.prune() self.assertEqual(1, mock_flush.call_count) self.assertEqual(3, mock_expired.call_count) self.assertEqual(3, mock_remove.call_count)