feat: TT-123 Add new feature to filter by role data

tests/time_tracker_api/time_entries/time_entries_namespace_test.py

@@ -1,5 +1,5 @@
 from datetime import timedelta
-from unittest.mock import ANY, Mock
+from unittest.mock import ANY, Mock, patch
 
 from faker import Faker
 from flask import json
@@ -16,6 +16,9 @@
     datetime_str,
 )
 from utils import worked_time
+from time_tracker_api.time_entries.time_entries_model import (
+    TimeEntryCosmosDBModel,
+)
 
 from werkzeug.exceptions import NotFound, UnprocessableEntity, HTTPException
 
@@ -189,6 +192,81 @@ def test_get_time_entry_should_succeed_with_valid_id(
     dao_get_all_mock.assert_called_once()
 
 
+@patch(
+    'time_tracker_api.time_entries.time_entries_dao.TimeEntriesCosmosDBDao.create_event_context',
+    Mock(),
+)
+@patch(
+    'time_tracker_api.time_entries.time_entries_dao.TimeEntriesCosmosDBDao.build_custom_query',
+    Mock(),
+)
+@patch(
+    'time_tracker_api.time_entries.time_entries_dao.TimeEntriesCosmosDBDao.handle_date_filter_args',
+    Mock(),
+)
+@patch('msal.ConfidentialClientApplication', Mock())
+@patch('utils.azure_users.AzureConnection.get_token', Mock())
+@patch('utils.azure_users.AzureConnection.is_test_user')
+@patch('utils.azure_users.AzureConnection.get_test_user_ids')
+@pytest.mark.parametrize(
+    'current_user_is_tester, expected_user_ids',
+    [
+        (True, ['id1', 'id2']),
+        (False, ['id2']),
+    ],
+)
+def test_get_time_entries_by_type_of_user(
+    get_test_user_ids_mock,
+    is_test_user_mock,
+    client: FlaskClient,
+    valid_header: dict,
+    time_entries_dao,
+    current_user_is_tester,
+    expected_user_ids,
+):
+    test_user_id = "id1"
+    non_test_user_id = "id2"
+    te1 = TimeEntryCosmosDBModel(
+        {
+            "id": '1',
+            "project_id": "1",
+            "owner_id": test_user_id,
+            "tenant_id": '1',
+            "start_date": "",
+        }
+    )
+    te2 = TimeEntryCosmosDBModel(
+        {
+            "id": '2',
+            "project_id": "2",
+            "owner_id": non_test_user_id,
+            "tenant_id": '2',
+            "start_date": "",
+        }
+    )
+
+    find_all_mock = Mock()
+    find_all_mock.return_value = [te1, te2]
+
+    time_entries_dao.repository.find_all = find_all_mock
+
+    is_test_user_mock.return_value = current_user_is_tester
+    get_test_user_ids_mock.return_value = [test_user_id]
+
+    response = client.get(
+        "/time-entries?user_id=*", headers=valid_header, follow_redirects=True
+    )
+
+    is_test_user_mock.assert_called()
+    find_all_mock.assert_called()
+
+    expected_user_ids_in_time_entries = expected_user_ids
+    actual_user_ids_in_time_entries = [
+        time_entry["owner_id"] for time_entry in json.loads(response.data)
+    ]
+    assert expected_user_ids_in_time_entries == actual_user_ids_in_time_entries
+
+
 def test_get_time_entry_should_succeed_with_valid_id(
     client: FlaskClient,
     mocker: MockFixture,
@@ -595,6 +673,11 @@ def test_create_with_valid_uuid_format_should_return_created(
     repository_container_create_item_mock.assert_called()
 
 
+@patch('msal.ConfidentialClientApplication', Mock())
+@patch('utils.azure_users.AzureConnection.get_token', Mock())
+@patch(
+    'utils.azure_users.AzureConnection.is_test_user', Mock(return_value=True)
+)
 @pytest.mark.parametrize(
     'url',
     [
@@ -624,6 +707,11 @@ def test_get_all_passes_date_range_built_from_params_to_find_all(
     assert 'end_date' in kwargs['date_range']
 
 
+@patch('msal.ConfidentialClientApplication', Mock())
+@patch('utils.azure_users.AzureConnection.get_token', Mock())
+@patch(
+    'utils.azure_users.AzureConnection.is_test_user', Mock(return_value=True)
+)
 @pytest.mark.parametrize(
     'url,start_date,end_date',
     [
@@ -660,6 +748,11 @@ def test_get_all_passes_date_range_to_find_all_with_default_tz_offset(
     assert kwargs['date_range']['end_date'] == end_date
 
 
+@patch('msal.ConfidentialClientApplication', Mock())
+@patch('utils.azure_users.AzureConnection.get_token', Mock())
+@patch(
+    'utils.azure_users.AzureConnection.is_test_user', Mock(return_value=True)
+)
 @pytest.mark.parametrize(
     'url,start_date,end_date',
     [

time_tracker_api/time_entries/time_entries_dao.py

@@ -28,6 +28,7 @@
 )
 from time_tracker_api.database import CRUDDao, APICosmosDBDao
 from time_tracker_api.security import current_user_id
+from utils.azure_users import AzureConnection
 
 
 class TimeEntriesDao(CRUDDao):
@@ -93,21 +94,35 @@ def build_custom_query(self, is_admin: bool, conditions: dict = None):
     def get_all(self, conditions: dict = None, **kwargs) -> list:
         event_ctx = self.create_event_context("read-many")
         conditions.update({"owner_id": event_ctx.user_id})
-
+        is_complete_query = conditions.get("user_id") == '*'
         custom_query = self.build_custom_query(
             is_admin=event_ctx.is_admin,
             conditions=conditions,
         )
         date_range = self.handle_date_filter_args(args=conditions)
         limit = conditions.get("limit", None)
         conditions.pop("limit", None)
-        return self.repository.find_all(
+        azure_connection = AzureConnection()
+        current_user_is_tester = azure_connection.is_test_user(
+            event_ctx.user_id
+        )
+        time_entries_list = self.repository.find_all(
             event_ctx,
             conditions=conditions,
             custom_sql_conditions=custom_query,
             date_range=date_range,
             max_count=limit,
         )
+        if not current_user_is_tester and is_complete_query:
+            test_user_ids = azure_connection.get_test_user_ids()
+            time_entries_list = [
+                time_entry
+                for time_entry in time_entries_list
+                if time_entry.owner_id not in test_user_ids
+            ]
+            return time_entries_list
+        else:
+            return time_entries_list
 
     def get_lastest_entries_by_project(
         self, conditions: dict = None, **kwargs