@@ -76,6 +76,8 @@ public static function createSignedTokenFromData($privateKey, $tokenData) {
7676 ->set (SSOToken::CLAIM_THEME_BACKGROUND_COLOR , $ tokenData [SSOToken::CLAIM_THEME_BACKGROUND_COLOR ])
7777 ->set (SSOToken::CLAIM_USER_LOCALE , $ tokenData [SSOToken::CLAIM_USER_LOCALE ])
7878 ->set (SSOToken::CLAIM_USER_TAGS , $ tokenData [SSOToken::CLAIM_USER_TAGS ])
79+ ->set (SSOToken::CLAIM_BRANCH_ID , $ tokenData [SSOToken::CLAIM_BRANCH_ID ])
80+ ->set (SSOToken::CLAIM_BRANCH_SLUG , $ tokenData [SSOToken::CLAIM_BRANCH_SLUG ])
7981 ->sign ($ signer , $ keychain ->getPrivateKey ($ privateKey ))
8082 ->getToken ();
8183
@@ -110,6 +112,8 @@ private static function createUnsignedTokenFromData($tokenData) {
110112 ->set (SSOToken::CLAIM_THEME_BACKGROUND_COLOR , $ tokenData [SSOToken::CLAIM_THEME_BACKGROUND_COLOR ])
111113 ->set (SSOToken::CLAIM_USER_LOCALE , $ tokenData [SSOToken::CLAIM_USER_LOCALE ])
112114 ->set (SSOToken::CLAIM_USER_TAGS , $ tokenData [SSOToken::CLAIM_USER_TAGS ])
115+ ->set (SSOToken::CLAIM_BRANCH_ID , $ tokenData [SSOToken::CLAIM_BRANCH_ID ])
116+ ->set (SSOToken::CLAIM_BRANCH_SLUG , $ tokenData [SSOToken::CLAIM_BRANCH_SLUG ])
113117 ->getToken ();
114118
115119 return $ token ;
@@ -336,6 +340,8 @@ public function testConstructorToFailOnUnsignedToken() {
336340 * @covers \Staffbase\plugins\sdk\SSOToken::getTags()
337341 * @covers \Staffbase\plugins\sdk\SSOToken::hasClaim()
338342 * @covers \Staffbase\plugins\sdk\SSOToken::getClaim()
343+ * @covers \Staffbase\plugins\sdk\SSOToken::getBranchId()
344+ * @covers \Staffbase\plugins\sdk\SSOToken::getBranchSlug()
339345 */
340346 public function testAccessorsGiveCorrectValues () {
341347
0 commit comments