implement hooks

src/web/hooks.py

@@ -12,8 +12,9 @@
 from src.web import PinfluencerContext, valid_path_resource_id, ErrorCapsule
 from src.web.constants import AudienceAgeCacheKey, InfluencerDetailsCacheKey, AudienceGenderCacheKey
 from src.web.error_capsules import AudienceDataAlreadyExistsErrorCapsule, BrandNotFoundErrorCapsule, \
-    InfluencerNotFoundErrorCapsule, ListingNotFoundErrorCapsule
-from src.web.views import RawImageRequestDto, ImageRequestDto, ListingResponseDto, NotificationCreateRequestDto
+    InfluencerNotFoundErrorCapsule, ListingNotFoundErrorCapsule, BrandNotAuthorized
+from src.web.views import RawImageRequestDto, ImageRequestDto, ListingResponseDto, NotificationCreateRequestDto, \
+    CollaborationResponseDto
 
 S3_URL = "https://pinfluencer-product-images.s3.eu-west-2.amazonaws.com"
 
@@ -480,11 +481,16 @@ def __init__(self, repository: AudienceGenderRepository):
 
 class CollaborationAfterHooks(SaveableHook):
 
-    def __init__(self, repository: CollaborationRepository):
+    def __init__(self,
+                 repository: CollaborationRepository,
+                 mapper: PinfluencerObjectMapper):
         super().__init__(repository)
+        self.__mapper = mapper
 
     def validate_brand(self, context: PinfluencerContext):
-        ...
+        if context.auth_user_id != self.__mapper.map_from_dict(_from=context.response.body,
+                                                               to=CollaborationResponseDto).brand_auth_user_id:
+            context.error_capsule = [BrandNotAuthorized(auth_user_id=context.auth_user_id)]
 
 
 class AudienceAgeBeforeHooks:

tests/test_hooks.py

@@ -8,7 +8,7 @@
 
 from src._types import AuthUserRepository, BrandRepository, ImageRepository, NotificationRepository, \
     AudienceAgeRepository, InfluencerRepository, AudienceGenderRepository, ListingRepository, CollaborationRepository
-from src.crosscutting import JsonCamelToSnakeCaseDeserializer, AutoFixture
+from src.crosscutting import JsonCamelToSnakeCaseDeserializer, AutoFixture, PinfluencerObjectMapper
 from src.domain.models import User, ValueEnum, CategoryEnum, AudienceAgeSplit, AudienceGenderSplit, \
     AudienceAge, Listing
 from src.domain.validation import InfluencerValidator, BrandValidator, ListingValidator
@@ -1170,7 +1170,9 @@ class TestCollaborationAfterHooks(TestCase):
 
     def setUp(self) -> None:
         self.__repo: CollaborationRepository = Mock()
-        self.__sut = CollaborationAfterHooks(repository=self.__repo)
+        self.__mapper = PinfluencerObjectMapper(logger=Mock())
+        self.__sut = CollaborationAfterHooks(repository=self.__repo,
+                                             mapper=self.__mapper)
 
     def test_validate_brand_when_valid(self):
         # arrange