From 31360816e98cd84fca33be10a1b21c373de3f9e6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Martin=20Ad=C3=A1mek?= <banan23@gmail.com>
Date: Wed, 27 Jul 2022 16:10:38 +0200
Subject: [PATCH] chore: use protected property directly

---
 packages/core/src/session_pool/session.ts      | 3 ++-
 packages/core/src/session_pool/session_pool.ts | 5 -----
 test/core/session_pool/session.test.ts         | 3 ++-
 3 files changed, 4 insertions(+), 7 deletions(-)

diff --git a/packages/core/src/session_pool/session.ts b/packages/core/src/session_pool/session.ts
index ecdbca763ce3..99c208de7d4c 100644
--- a/packages/core/src/session_pool/session.ts
+++ b/packages/core/src/session_pool/session.ts
@@ -264,7 +264,8 @@ export class Session {
      * @returns Whether the session was retired.
      */
     retireOnBlockedStatusCodes(statusCode: number, blockedStatusCodes: number[] = []): boolean {
-        const isBlocked = this.sessionPool.getBlockedStatusCodes(blockedStatusCodes).includes(statusCode);
+        // @ts-expect-error
+        const isBlocked = this.sessionPool.blockedStatusCodes.concat(blockedStatusCodes).includes(statusCode);
         if (isBlocked) {
             this.retire();
         }
diff --git a/packages/core/src/session_pool/session_pool.ts b/packages/core/src/session_pool/session_pool.ts
index c6825ef26bee..e3e02cfee849 100644
--- a/packages/core/src/session_pool/session_pool.ts
+++ b/packages/core/src/session_pool/session_pool.ts
@@ -314,11 +314,6 @@ export class SessionPool extends EventEmitter {
         await this.persistState();
     }
 
-    /** @internal */
-    getBlockedStatusCodes(additionalCodes: number[] = []): number[] {
-        return this.blockedStatusCodes.concat(additionalCodes);
-    }
-
     /**
      * SessionPool should not work before initialization.
      */
diff --git a/test/core/session_pool/session.test.ts b/test/core/session_pool/session.test.ts
index 486e9a2056de..e2be4069b704 100644
--- a/test/core/session_pool/session.test.ts
+++ b/test/core/session_pool/session.test.ts
@@ -178,7 +178,8 @@ describe('Session - testing session behaviour ', () => {
         expect(session.retireOnBlockedStatusCodes(200)).toBeFalsy();
         expect(session.retireOnBlockedStatusCodes(400)).toBeFalsy();
         expect(session.retireOnBlockedStatusCodes(500)).toBeFalsy();
-        sessionPool.getBlockedStatusCodes().forEach((status) => {
+        // @ts-expect-error
+        sessionPool.blockedStatusCodes.forEach((status) => {
             const sess = new Session({ sessionPool });
             let isCalled;
             const call = () => { isCalled = true; };