ipulse-shared-core-ftredge 2.56__tar.gz → 3.1.1__tar.gz

{ipulse_shared_core_ftredge-2.56/src/ipulse_shared_core_ftredge.egg-info → ipulse_shared_core_ftredge-3.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ipulse_shared_core_ftredge
-Version: 2.56
+Version: 3.1.1
 Summary: Shared Core models and Logger util for the Pulse platform project. Using AI for financial advisory and investment management.
 Home-page: https://github.com/TheFutureEdge/ipulse_shared_core
 Author: Russlan Ramdowar
@@ -8,6 +8,4 @@ License-File: LICENCE
 Requires-Dist: pydantic[email]~=2.5
 Requires-Dist: python-dateutil~=2.8
 Requires-Dist: pytest~=7.1
-Requires-Dist: google-cloud-logging~=3.10.0
-Requires-Dist: google-cloud-error-reporting~=1.11.0
-Requires-Dist: google-cloud-bigquery~=3.24.0
+Requires-Dist: ipulse_shared_base_ftredge>=2.2.2

{ipulse_shared_core_ftredge-2.56 → ipulse_shared_core_ftredge-3.1.1}/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='ipulse_shared_core_ftredge',
-    version='2.56',
+    version='3.1.1',
     package_dir={'': 'src'},  # Specify the source directory
     packages=find_packages(where='src'),  # Look for packages in 'src'
     install_requires=[
@@ -11,9 +11,7 @@ setup(
         'pydantic[email]~=2.5',
         'python-dateutil~=2.8',
         'pytest~=7.1',
-        'google-cloud-logging~=3.10.0',
-        'google-cloud-error-reporting~=1.11.0',
-        'google-cloud-bigquery~=3.24.0'
+        'ipulse_shared_base_ftredge>=2.2.2',
     ],
     author='Russlan Ramdowar',
     description='Shared Core models and Logger util for the Pulse platform project. Using AI for financial advisory and investment management.',

ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge/__init__.py

@@ -0,0 +1,11 @@
+# pylint: disable=missing-module-docstring
+from .models import ( UserAuth, UserProfile,
+                     UserStatus, UserProfileUpdate)
+
+from .utils import (list_as_strings)
+
+from .logging import (get_logger,
+                      log_error,
+                      log_warning,
+                      log_info,
+                      log_debug)

{ipulse_shared_core_ftredge-2.56 → ipulse_shared_core_ftredge-3.1.1}/src/ipulse_shared_core_ftredge/models/__init__.py

@@ -1,7 +1,6 @@
 from .user_profile import UserProfile
 from .user_status import UserStatus
 from .user_profile_update import UserProfileUpdate
-from .organisation import Organisation
 from .user_auth import UserAuth
 
 

ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge/models/organisation.py

@@ -0,0 +1,71 @@
+# pylint: disable=missing-module-docstring
+# pylint: disable=missing-function-docstring
+# pylint: disable=missing-class-docstring
+# pylint: disable=broad-exception-caught
+# pylint: disable=line-too-long
+# pylint: disable=unused-variable
+# pylint: disable=broad-exception-caught
+# from pydantic import BaseModel, validator, ValidationError, Field
+# from typing import Set, Optional
+# import uuid
+# from datetime import datetime
+# import dateutil.parser
+
+# CLASS_VERSION= 1.0
+# MODULE= "core"
+# CLASS_REF="orgn"
+
+
+# class Organisation(BaseModel):
+#     puid: str = Field(default_factory=f"{datetime.utcnow().strftime('%Y%m%d%H%M')}{uuid.uuid4().hex[:8]}_{MODULE}{CLASS_REF}".lower())
+#     name: str
+#     creat_date: datetime = Field(default_factory=datetime.utcnow()) 
+#     updt_date: datetime = Field(default_factory=datetime.utcnow())
+#     creat_by_user: Optional[str] = None 
+#     updt_by_user: Optional[str] = None
+#     relations: Optional[Set[str]]=None
+#     description: Optional[str] = None  # Updated to use Optional
+#     industries: Optional[Set[str]] = None  # Updated to use Optional
+#     website: Optional[str] = None  # Updated to use Optional
+#     org_admin_user_uids: Optional[Set[str]] = None  # Updated to use Optional
+#     class Config:
+#             extra = "forbid"
+
+
+    # @validator('relations', pre=True, always=True)
+    # def validate_relations(cls, relations):
+    #     if not set(relations).issubset(enums.organisation_relations):
+    #         raise ValueError("Invalid relation values provided.")
+    #     return relations
+
+   
+    # @validator('industries', pre=True, always=True)
+    # def validate_industries(cls, industries):
+    #     if industries is not None and not set(industries).issubset(enums.organisation_industries):
+    #         raise ValueError("Invalid industry values provided.")
+    #     return industries
+    
+    # @validator('creat_date', 'updt_date', pre=True)
+    # def parse_date(cls, value):
+    #     if value is None:
+    #         return value
+    #     if isinstance(value, datetime):
+    #         return value
+    #     try:
+    #         # Assuming Firestore returns an ISO 8601 string, adjust if necessary
+    #         print("Putting Updt or Creat date in a valid format in a Validator when creating Organisation object")
+    #         return dateutil.parser.isoparse(value)
+    #     except (TypeError, ValueError):
+    #         raise ValidationError(f"Invalid datetime format inside Organisation: {value}")
+
+
+    # ### Description, Industries, and Website are optional for Retail Customer and mandatory for Non Retail Customer
+    # @validator('description', 'industries', 'website', pre=True, always=True)
+    # def validate_optional_fields(cls, value, values):
+    #     if values.get('name') == 'Retail Customer' and values.get('relations') == {"retail_customer"} or  values.get('relations') == ["retail_customer"]:
+    #         if value is not None:
+    #             raise ValueError("For 'Retail Customer' with only 'retail_customer' relations, description, industries, and website should not be provided.")
+    #     else:
+    #         if value is None:
+    #             raise ValueError("For Non Retail Customer, description, industries, and website are mandatory.")
+    #     return value

ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge/models/resource_catalog_item.py

@@ -0,0 +1,115 @@
+# import uuid
+# from datetime import datetime
+# from pydantic import BaseModel, validator, ValidationError
+# from typing import Dict, Any, Set, Optional
+
+# import dateutil.parser
+
+# CLASS_VERSION = 1.0
+# CLASS_REF = "resdes"
+# MODULE = "core"
+
+# class ResourceCatalogItem(BaseModel):
+
+#     resr_puid_or_name: str #Ex: username
+#     resr_path: str #Ex: ipulse-401013/cloud/firesotre/Users/{user_uid}/username
+#     resr_name: str #Ex: username
+#     resr_pulse_module: str  #Ex: core 
+#     resr_type: str
+#     resr_classifications: Set[str] 
+#     resr_contents:Set[str]
+#     resr_original_or_processed: str
+#     resr_origin: str
+#     resr_origin_organisations_uids: Set[str]
+#     resr_origin_description: str
+#     resr_licences_types: Set[str]
+#     resr_description_details: str 
+#     resr_updtbl_by_non_staff: bool
+#     resr_creat_by_user_uid: str
+#     resr_creat_date: datetime
+#     class_version:float = CLASS_VERSION
+#     resr_columns_count: int
+#     resr_columns: Optional[Dict[Any, Any]] = None #OPTIONAL
+#     resr_structure_version: Optional[str]=None # OPTIONAL
+#     resr_structure_updt_date: Optional[str]=None #OPTIONAL
+#     resr_structure_updt_by_user_uid: Optional[str]=None # OPTIONAL
+#     resr_tags: Optional[Dict[Any, Any]] = None #OPTIONAL
+#     resr_content_updt_date: Optional[str]=None #OPTIONAL
+#     resr_content_updt_by_user_uid: Optional[str]=None # OPTIONAL
+#     puid: Optional[str] = None #TO BE SETUP BY Validator
+#     metadata_version: Optional[float] = None #TO BE SETUP BY  Validator
+#     metadata_creat_date: Optional[datetime] = None #TO BE SETUP BY Management Service
+#     metadata_creat_by: Optional[str] = None #TO BE SETUP BY Management Service
+#     metadata_updt_date: Optional[datetime] = None #TO BE SETUP BY Management Service
+#     metadata_updt_by: Optional[str] = None #TO BE SETUP BY Management Service
+     
+    # @validator('puid', pre=True, always=True)
+    # def set_puid(cls, puid, values):
+    #     if puid is None:
+    #         return f"{datetime.utcnow().strftime('%Y%m%d%H%M')}{uuid.uuid4().hex[:8]}_{MODULE}{CLASS_REF}".lower()
+    #     return puid
+    
+    # @validator('metadata_version', pre=True, always=True)
+    # def set_metadata_version(cls, metadata_version, values):
+    #     if metadata_version is None:
+    #         return 1.0
+    #     else: 
+    #         return metadata_version + 0.1
+
+
+    # @validator('resr_pulse_module', pre=True, always=True)
+    # def validate_resr_pulse_module(cls, resr_pulse_modules):
+    #     if resr_pulse_modules not in enums.pulse_modules:
+    #         raise ValueError("Invalid pulse_modules values provided.")
+    #     return resr_pulse_modules
+    
+    # @validator('resr_type', pre=True, always=True)
+    # def validate_resr_type(cls, resr_type):
+    #     if  resr_type not in enums.resource_types:
+    #         raise ValueError("Invalid resource_types value provided.")
+    #     return resr_type
+    
+    # @validator('resr_classifications', pre=True, always=True)
+    # def validate_resr_classifications(cls, resr_classifications):
+    #     if not resr_classifications.issubset(enums.resource_classifications):
+    #         raise ValueError("Invalid resr_classifications values provided.")
+    #     return resr_classifications
+
+    # @validator('resr_contents', pre=True, always=True)
+    # def validate_resr_contents(cls, resr_contents):
+    #     if not resr_contents.issubset(enums.resource_contents):
+    #         raise ValueError("Invalid resr_contents values provided.")
+    #     return resr_contents
+
+    # @validator('resr_original_or_processed', pre=True, always=True)
+    # def validate_resr_original_or_processed(cls, resr_original_or_processed):
+    #     if  resr_original_or_processed not in enums.resource_original_or_processed:
+    #         raise ValueError("Invalid resr_original_or_processed value provided.")
+    #     return resr_original_or_processed
+    
+    # @validator('resr_origin', pre=True, always=True)
+    # def validate_resr_origin(cls, resr_origin):
+    #     if  resr_origin not in enums.resource_origins:
+    #         raise ValueError("Invalid resource_origins value provided.")
+    #     return resr_origin
+       
+
+    # @validator('metadata_creat_date', 'metadata_updt_date', pre=True)
+    # def parse_date(cls, value):
+    #     if value is None:
+    #         return value
+    #     if isinstance(value, datetime):
+    #         return value
+    #     try:
+    #         # Assuming Firestore returns an ISO 8601 string, adjust if necessary
+    #         return dateutil.parser.isoparse(value)
+    #     except (TypeError, ValueError):
+    #         raise ValidationError(f"Invalid datetime format inside Resource Description: {value}")
+
+
+    
+    # @validator('metadata_updt_date', 'metadata_updt_date', pre=True, always=True)
+    # def set_default_updt_date(cls, date, values):
+    #     if date is None:
+    #         return datetime.utcnow().isoformat()
+    #     return date

{ipulse_shared_core_ftredge-2.56 → ipulse_shared_core_ftredge-3.1.1}/src/ipulse_shared_core_ftredge/models/user_profile.py

@@ -33,9 +33,9 @@ class UserProfile(BaseModel):
     provider_id: str   #User can Read only
     
     username: Optional[str] = None #User can Read and Edit
-    dob: Optional[date] = None #User can Read and Edit 
-    first_name: Optional[str] = None #User can Read and Edit 
-    last_name: Optional[str] = None #User can Read and Edit 
+    dob: Optional[date] = None #User can Read and Edit
+    first_name: Optional[str] = None #User can Read and Edit
+    last_name: Optional[str] = None #User can Read and Edit
     mobile: Optional[str] = None #User can Read and Edit
     class Config:
         extra = "forbid"

ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge/utils/__init__.py

@@ -0,0 +1,3 @@
+# pylint: disable=missing-module-docstring
+
+from .utils_common import (list_as_strings)

ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge/utils/utils_common.py

@@ -0,0 +1,10 @@
+# pylint: disable=missing-module-docstring
+# pylint: disable=missing-function-docstring
+# pylint: disable=logging-fstring-interpolation
+# pylint: disable=line-too-long
+# pylint: disable=missing-class-docstring
+# pylint: disable=broad-exception-caught
+
+def list_as_strings(*enums):
+    """Converts a list of Enum members to their string values."""
+    return [str(enum) for enum in enums]

{ipulse_shared_core_ftredge-2.56 → ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ipulse_shared_core_ftredge
-Version: 2.56
+Version: 3.1.1
 Summary: Shared Core models and Logger util for the Pulse platform project. Using AI for financial advisory and investment management.
 Home-page: https://github.com/TheFutureEdge/ipulse_shared_core
 Author: Russlan Ramdowar
@@ -8,6 +8,4 @@ License-File: LICENCE
 Requires-Dist: pydantic[email]~=2.5
 Requires-Dist: python-dateutil~=2.8
 Requires-Dist: pytest~=7.1
-Requires-Dist: google-cloud-logging~=3.10.0
-Requires-Dist: google-cloud-error-reporting~=1.11.0
-Requires-Dist: google-cloud-bigquery~=3.24.0
+Requires-Dist: ipulse_shared_base_ftredge>=2.2.2

{ipulse_shared_core_ftredge-2.56 → ipulse_shared_core_ftredge-3.1.1}/src/ipulse_shared_core_ftredge.egg-info/SOURCES.txt

@@ -3,27 +3,18 @@ README.md
 pyproject.toml
 setup.py
 src/ipulse_shared_core_ftredge/__init__.py
-src/ipulse_shared_core_ftredge/utils_common.py
-src/ipulse_shared_core_ftredge/utils_gcp.py
-src/ipulse_shared_core_ftredge/utils_templates_and_schemas.py
 src/ipulse_shared_core_ftredge.egg-info/PKG-INFO
 src/ipulse_shared_core_ftredge.egg-info/SOURCES.txt
 src/ipulse_shared_core_ftredge.egg-info/dependency_links.txt
 src/ipulse_shared_core_ftredge.egg-info/requires.txt
 src/ipulse_shared_core_ftredge.egg-info/top_level.txt
-src/ipulse_shared_core_ftredge/enums/__init__.py
-src/ipulse_shared_core_ftredge/enums/enums_common_utils.py
-src/ipulse_shared_core_ftredge/enums/enums_data_eng.py
-src/ipulse_shared_core_ftredge/enums/enums_module_fincore.py
-src/ipulse_shared_core_ftredge/enums/enums_modules.py
 src/ipulse_shared_core_ftredge/models/__init__.py
-src/ipulse_shared_core_ftredge/models/audit_log_firestore.py
 src/ipulse_shared_core_ftredge/models/organisation.py
-src/ipulse_shared_core_ftredge/models/pulse_enums.py
 src/ipulse_shared_core_ftredge/models/resource_catalog_item.py
 src/ipulse_shared_core_ftredge/models/user_auth.py
 src/ipulse_shared_core_ftredge/models/user_profile.py
 src/ipulse_shared_core_ftredge/models/user_profile_update.py
 src/ipulse_shared_core_ftredge/models/user_status.py
-src/ipulse_shared_core_ftredge/tests/__init__.py
-src/ipulse_shared_core_ftredge/tests/test.py
+src/ipulse_shared_core_ftredge/utils/__init__.py
+src/ipulse_shared_core_ftredge/utils/utils_common.py
+tests/test_utils_gcp.py

ipulse_shared_core_ftredge-3.1.1/src/ipulse_shared_core_ftredge.egg-info/requires.txt

@@ -0,0 +1,4 @@
+pydantic[email]~=2.5
+python-dateutil~=2.8
+pytest~=7.1
+ipulse_shared_base_ftredge>=2.2.2

ipulse_shared_core_ftredge-3.1.1/tests/test_utils_gcp.py

@@ -0,0 +1,189 @@
+# pylint: disable=missing-module-docstring
+# pylint: disable=import-error
+# pylint: disable=missing-module-docstring
+# pylint: disable=line-too-long
+# pylint: disable=missing-function-docstring
+
+import os
+import json
+from unittest.mock import MagicMock, patch
+import pytest
+from ipulse_shared_core_ftredge.utils_cloud_gcp import write_json_to_gcs
+
+
+# Mocking Google Cloud Storage components for testing using pytest-mock
+
+@pytest.fixture
+def mock_blob(mocker):
+    mock_blob_class = mocker.patch('google.cloud.storage.Blob', autospec=True)
+    mock_blob = mock_blob_class.return_value
+    mock_blob.exists.return_value = False
+    return mock_blob
+
+
+@pytest.fixture
+def mock_bucket(mocker, mock_blob):
+    mock_bucket_class = mocker.patch('google.cloud.storage.Bucket', autospec=True)
+    mock_bucket = mock_bucket_class.return_value
+    mock_bucket.list_blobs.return_value = []
+    mock_bucket.blob.return_value = mock_blob # this avoids creating a new blob for each test, which will confuse the test results
+    return mock_bucket
+ 
+@pytest.fixture
+def mock_storage_client(mocker, mock_bucket):
+    mock_client_class = mocker.patch('google.cloud.storage.Client', autospec=True)
+    mock_client = mock_client_class.return_value
+    mock_client.bucket.return_value = mock_bucket
+    return mock_client
+
+
+
+# --- Test Cases ---
+
+def test_successful_gcs_upload(mock_storage_client):
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name, test_file_name)
+
+    assert result['gcs_path'] == f"gs://{test_bucket_name}/{test_file_name}"
+    assert result['local_path'] is None
+    assert result['gcs_file_already_exists'] is False
+    assert result['gcs_file_overwritten'] is False
+    assert result['gcs_file_saved_with_increment'] is False
+
+
+def test_invalid_data_type(mock_storage_client):
+    with pytest.raises(ValueError) as exc_info:
+        write_json_to_gcs(mock_storage_client, 12345, "test_bucket", "test_file.json")
+    assert str(exc_info.value) == "Unsupported data type. Data must be a list, dict, or str."
+
+
+def test_overwrite_if_exists(mock_storage_client, mock_blob):
+    mock_blob.exists.return_value = True  # Simulate existing file
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name, test_file_name, overwrite_if_exists=True)
+
+    assert result['gcs_file_overwritten'] is True
+
+
+def test_overwrite_with_substring(mock_storage_client, mock_bucket):
+    mock_bucket.list_blobs.return_value = [MagicMock(name='test_prefix_file1.json'), MagicMock(name='test_prefix_file2.json')]  
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+    test_prefix = 'test_prefix'
+
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name, 
+                            test_file_name, overwrite_if_exists=True, 
+                            file_exists_if_starts_with_prefix=test_prefix)
+    assert result['gcs_file_overwritten'] is True
+    assert result['gcs_file_exists_checked_on_name'] == test_prefix
+
+
+def test_increment_if_exists(mock_storage_client, mock_blob):
+    mock_blob.exists.side_effect = [True, True, False]  # Simulate existing files 
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name, test_file_name, increment_if_exists=True)
+    assert result['gcs_path'] == f"gs://{test_bucket_name}/test_file_v2.json"
+    assert result['gcs_file_saved_with_increment'] is True
+
+
+def test_overwrite_and_increment_raise_value_error(mock_storage_client):
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+    with pytest.raises(ValueError) as exc_info:
+        write_json_to_gcs(mock_storage_client, test_data, test_bucket_name, 
+                        test_file_name, overwrite_if_exists=True,
+                        increment_if_exists=True)
+    assert str(exc_info.value) == "Both 'overwrite_if_exists' and 'increment_if_exists' cannot be True simultaneously."
+
+
+@patch('os.path.exists', return_value=False)  # Assume local file exists for simplicity
+@patch('builtins.open', new_callable=MagicMock)
+def test_local_save_after_gcs_failure(mock_open, mock_exists, mock_storage_client, mock_blob):
+    mock_blob.upload_from_string.side_effect = Exception("GCS Upload Failed")
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    # Expecting an exception because GCS upload fails
+    with pytest.raises(Exception) as exc_info:
+        result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name, test_file_name, save_locally=True)
+
+    assert "GCS Upload Failed" in str(exc_info.value)
+    mock_open.assert_called_once_with(os.path.join("/tmp", test_file_name), 'w', encoding='utf-8')
+
+
+@patch('builtins.open', new_callable=MagicMock)
+def test_local_save_with_custom_path(mock_open, mock_storage_client):
+    local_path = "/my/custom/path"
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name,
+                                test_file_name, local_path=local_path)
+
+    assert result['local_path'] == os.path.join(local_path, test_file_name)
+    mock_open.assert_called_once()
+
+@patch('os.path.exists', side_effect=[True, True, False])
+@patch('builtins.open', new_callable=MagicMock)
+def test_local_save_with_increment(mock_open, mock_exists, mock_storage_client, mock_blob):
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name,
+                                test_file_name, save_locally=True, increment_if_exists=True)
+
+    assert f"/tmp/test_file_v1.json" == result['local_path']
+    mock_open.assert_called_once()
+
+
+@patch('builtins.open', new_callable=MagicMock)
+def test_local_save_overwrite(mock_open, mock_storage_client):
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    # Execute function
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name,
+                               test_file_name, save_locally=True, overwrite_if_exists=True)
+
+    # Check results
+    assert result['local_path'] == os.path.join("/tmp", test_file_name)
+    mock_open.assert_called_once_with(os.path.join("/tmp", test_file_name), 'w', encoding='utf-8')
+    file_handle = mock_open()
+
+
+@patch('os.path.exists', return_value=True)
+@patch('builtins.open', new_callable=MagicMock)
+def test_local_save_skip(mock_open, mock_exists, mock_storage_client):
+    test_data = {"key": "value"}
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    result = write_json_to_gcs(mock_storage_client, test_data, test_bucket_name,
+                            test_file_name, save_locally=True, overwrite_if_exists=False)
+
+    assert result['local_path'] == os.path.join("/tmp", test_file_name)
+    mock_open.assert_not_called()
+
+
+def test_string_data_handling(mock_storage_client, mock_blob):
+    test_string_data = "This is a test string."
+    test_bucket_name = "test_bucket"
+    test_file_name = "test_file.json"
+
+    result = write_json_to_gcs(mock_storage_client, test_string_data, test_bucket_name, test_file_name)
+
+    assert result['gcs_path'] == f"gs://{test_bucket_name}/{test_file_name}" 

ipulse_shared_core_ftredge-2.56/src/ipulse_shared_core_ftredge/__init__.py

@@ -1,15 +0,0 @@
-from .models import (Organisation, UserAuth, UserProfile,
-                     UserStatus, UserProfileUpdate, pulse_enums)
-from .utils_gcp import (setup_gcp_logger_and_error_report, 
-                        read_csv_from_gcs, read_json_from_gcs, 
-                        write_csv_to_gcs, write_json_to_gcs)
-from .utils_templates_and_schemas import (create_bigquery_schema_from_json,
-                                          check_format_against_schema_template)
-from .utils_common import (ContextLog,  Pipelinemon)
-
-from .enums import (TargetLogs, LogLevel, Unit, Frequency,
-                    Module, SubModule, BaseDataCategory,
-                    FinCoreCategory, FincCoreSubCategory,
-                    FinCoreRecordsCategory, ExchangeOrPublisher,
-                    SourcingPipelineType, SourcingTriggerType,
-                    DWEvent, DWEventTriggerType)

ipulse_shared_core_ftredge-2.56/src/ipulse_shared_core_ftredge/enums/__init__.py

@@ -1,28 +0,0 @@
-
-# pylint: disable=missing-module-docstring
-# pylint: disable=missing-function-docstring
-# pylint: disable=missing-class-docstring
-
-from .enums_common_utils import (LogLevel,
-                                SystemsImpacted,
-                                TargetLogs,
-                                Unit,
-                                Frequency)
-
-
-from .enums_modules import(Module,
-                           SubModule,
-                           BaseDataCategory)
-
-
-from .enums_module_fincore import (FinCoreCategory,
-                                    FincCoreSubCategory,
-                                    FinCoreRecordsCategory,
-                                    ExchangeOrPublisher)
-
-
-
-from .enums_data_eng import (SourcingTriggerType,
-                             SourcingPipelineType,
-                             DWEvent,
-                             DWEventTriggerType)