Skip to content

Commit 19c48f8

Browse files
authored
fix: TT-242 refactory functions find all with id in list and review test functions (#294)
1 parent 1ded20b commit 19c48f8

File tree

2 files changed

+9
-11
lines changed

2 files changed

+9
-11
lines changed

tests/time_tracker_api/activities/activities_model_test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@
1616
)
1717
def test_find_all_with_id_in_list(
1818
find_partition_key_value_mock,
19-
create_sql_condition_for_visibility_mock,
2019
event_context: EventContext,
2120
activity_repository: ActivityCosmosDBRepository,
2221
):
@@ -31,9 +30,10 @@ def test_find_all_with_id_in_list(
3130
activity_repository.container = Mock()
3231
activity_repository.container.query_items = query_items_mock
3332

34-
result = activity_repository.find_all_with_id_in_list(event_context, [expected_item])
33+
result = activity_repository.find_all_with_id_in_list(
34+
event_context, [expected_item]
35+
)
3536

36-
create_sql_condition_for_visibility_mock.assert_called_once()
3737
find_partition_key_value_mock.assert_called_once()
3838
query_items_mock.assert_called_once()
3939

time_tracker_api/activities/activities_model.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -67,15 +67,13 @@ def find_all_with_id_in_list(
6767
visible_only=True,
6868
mapper: Callable = None,
6969
):
70-
visibility = self.create_sql_condition_for_visibility(visible_only)
71-
query_str = """
72-
SELECT * FROM c
73-
WHERE {condition}
74-
{visibility_condition}
75-
""".format(
76-
condition=create_sql_in_condition("id", activity_ids),
77-
visibility_condition=visibility,
70+
query_builder = (
71+
CosmosDBQueryBuilder()
72+
.add_sql_in_condition('id', activity_ids)
73+
.add_sql_visibility_condition(visible_only)
74+
.build()
7875
)
76+
query_str = query_builder.get_query()
7977

8078
tenant_id_value = self.find_partition_key_value(event_context)
8179
result = self.container.query_items(

0 commit comments

Comments
 (0)