diff --git a/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php b/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php index 6a56a0b606b8d6a3fb1d051c5b459a680013390f..dda6f4e2557383b9c95dd50d4ee4f7999ef3cc0d 100644 --- a/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php +++ b/src/UnicaenAuthentification/Service/Traits/UserContextServiceAwareTrait.php @@ -18,10 +18,15 @@ trait UserContextServiceAwareTrait * @param UserContext $serviceUserContext * @return self */ - public function setServiceUserContext(UserContext $serviceUserContext) + public function setServiceUserContext(UserContext $serviceUserContext) : UserContext { $this->serviceUserContext = $serviceUserContext; + return $this->serviceUserContext; + } - return $this; + public function getServiceUserContext() : UserContext + { + return $this->serviceUserContext; } -} \ No newline at end of file + +}