diff --git a/tests/time_tracker_api/time_entries/time_entries_namespace_test.py b/tests/time_tracker_api/time_entries/time_entries_namespace_test.py index 40e3f4ea..214f130b 100644 --- a/tests/time_tracker_api/time_entries/time_entries_namespace_test.py +++ b/tests/time_tracker_api/time_entries/time_entries_namespace_test.py @@ -624,7 +624,7 @@ def test_summary_is_called_with_date_range_from_worked_time_module( worked_time.date_range = Mock(return_value=worked_time.date_range()) repository_find_all_mock = mocker.patch.object( - time_entries_dao.repository, 'find_all', return_value=[] + time_entries_dao.repository, 'find_all_entries', return_value=[] ) response = client.get( diff --git a/time_tracker_api/time_entries/time_entries_model.py b/time_tracker_api/time_entries/time_entries_model.py index 98b524cf..d52435a9 100644 --- a/time_tracker_api/time_entries/time_entries_model.py +++ b/time_tracker_api/time_entries/time_entries_model.py @@ -157,6 +157,27 @@ def create_sql_date_range_filter(date_range: dict) -> str: else: return '' + def find_all_entries( + self, + event_context: EventContext, + conditions: dict = {}, + custom_sql_conditions: List[str] = [], + date_range: dict = {}, + ): + custom_sql_conditions.append( + self.create_sql_date_range_filter(date_range) + ) + + custom_params = self.generate_params(date_range) + time_entries = CosmosDBRepository.find_all( + self, + event_context=event_context, + conditions=conditions, + custom_sql_conditions=custom_sql_conditions, + custom_params=custom_params, + ) + return time_entries + def find_all( self, event_context: EventContext, @@ -461,7 +482,7 @@ def get_worked_time(self, args: dict = {}): ) conditions = {"owner_id": event_ctx.user_id} - time_entries = self.repository.find_all( + time_entries = self.repository.find_all_entries( event_ctx, conditions=conditions, date_range=worked_time.date_range(),