Wrong environment to check the redshift role

data-dot-all · Sep 27, 2024 · 00d92f6 · 00d92f6
1 parent 0b1295a
commit 00d92f6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/dataall/modules/redshift_datasets_shares/services/redshift_table_share_validator.py b/backend/dataall/modules/redshift_datasets_shares/services/redshift_table_share_validator.py
@@ -48,7 +48,7 @@ def validate_share_object_create(
 
     @staticmethod
     def validate_share_object_submit(session, dataset, share) -> bool:
-        environment = EnvironmentService.get_environment_by_uri(session, dataset.environmentUri)
+        environment = EnvironmentService.get_environment_by_uri(session, share.environmentUri)
         RedshiftTableValidator._validate_redshift_role(
             session=session,
             environment=environment,
@@ -60,7 +60,7 @@ def validate_share_object_submit(session, dataset, share) -> bool:
 
     @staticmethod
     def validate_share_object_approve(session, dataset, share) -> bool:
-        environment = EnvironmentService.get_environment_by_uri(session, dataset.environmentUri)
+        environment = EnvironmentService.get_environment_by_uri(session, share.environmentUri)
         RedshiftTableValidator._validate_redshift_role(
             session=session,
             environment=environment,