Skip to content
Closed
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
feat: TT-181 change projects_model and projects_model_test functions …
…name
  • Loading branch information
eduardisrael committed Mar 18, 2021
commit 79ace4b69fdec378299d9f10bfdec4908c606cda
8 changes: 4 additions & 4 deletions tests/time_tracker_api/projects/projects_model_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,22 +21,22 @@ def test_create_sql_in_condition(
customer_ids_list,
expected_result,
):
result = project_repository.create_sql_in_condition(customer_ids_list)
result = project_repository.create_sql_customer_id_in_condition(customer_ids_list)
assert expected_result == result


@patch(
'time_tracker_api.projects.projects_model.ProjectCosmosDBRepository.create_sql_condition_for_visibility'
)
@patch(
'time_tracker_api.projects.projects_model.ProjectCosmosDBRepository.create_sql_in_condition'
'time_tracker_api.projects.projects_model.ProjectCosmosDBRepository.create_sql_customer_id_in_condition'
)
@patch(
'time_tracker_api.projects.projects_model.ProjectCosmosDBRepository.find_partition_key_value'
)
def test_find_all_with_customer_id_in_list(
find_partition_key_value_mock,
create_sql_in_condition_mock,
create_sql_customer_id_in_condition_mock,
create_sql_condition_for_visibility_mock,
event_context: EventContext,
project_repository: ProjectCosmosDBRepository,
Expand All @@ -56,7 +56,7 @@ def test_find_all_with_customer_id_in_list(
result = project_repository.find_all_with_customer_id_in_list(event_context, [])

create_sql_condition_for_visibility_mock.assert_called_once()
create_sql_in_condition_mock.assert_called_once()
create_sql_customer_id_in_condition_mock.assert_called_once()
find_partition_key_value_mock.assert_called_once()
query_items_mock.assert_called_once()

Expand Down
4 changes: 2 additions & 2 deletions time_tracker_api/projects/projects_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def __init__(self):
mapper=ProjectCosmosDBModel,
)

def create_sql_in_condition(self, customer_ids_list):
def create_sql_customer_id_in_condition(self, customer_ids_list):
id_values = convert_list_to_tuple_string(customer_ids_list)

return "c.customer_id IN {value_condition}".format(value_condition=id_values)
Expand All @@ -84,7 +84,7 @@ def find_all_with_customer_id_in_list(
WHERE {condition}
{visibility_condition}
""".format(
condition=self.create_sql_in_condition(customer_ids_list),
condition=self.create_sql_customer_id_in_condition(customer_ids_list),
visibility_condition=visibility,
)

Expand Down