diff --git a/cl/tests/api/utils.py b/cl/tests/api/utils.py index 629f61d8a0..3e7452009c 100644 --- a/cl/tests/api/utils.py +++ b/cl/tests/api/utils.py @@ -1,6 +1,7 @@ import unittest from unittest.mock import MagicMock, patch + class TestApiUsage(unittest.TestCase): def setUp(self): self.start_date = '2023-01-01' @@ -17,7 +18,7 @@ def test_recent_api_usage(self, mock_get_redis_interface): mock_redis.pipeline.return_value = mock_pipeline mock_pipeline.execute.return_value = [ - self.v3_results] * len(self.expected_dates) + + self.v3_results] * len(self.expected_dates) + [self.v4_results] * len(self.expected_dates) results = get_recent_api_usage(self.start_date, self.end_date)