Apply suggestions from code review

Co-authored-by: Torsten Kilias <[email protected]>
exasol · Apr 3, 2024 · 6e32d60 · 6e32d60
1 parent 475e18c
commit 6e32d60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/unit/aws/test_codebuild_waiter.py b/test/unit/aws/test_codebuild_waiter.py
@@ -15,7 +15,7 @@ def ms(value: int) -> timedelta:
     return timedelta(milliseconds=value)
 
 
-def codebuild_client(build_duration: timedelta, finish_status: str):
+def codebuild_client(build_duration: timedelta, finish_status: str) -> MagicMock:
     end_time = datetime.now() + build_duration
     def batch_get_builds(ids: List[int]):
         if len(ids) == 1 and ids[0] == BUILD_ID: