Tests

Author: logan-stytch

stytch/src/main/kotlin/com/stytch/java/common/PolicyCache.kt

@@ -7,7 +7,6 @@ import com.stytch.java.b2b.models.rbac.Policy
 import com.stytch.java.b2b.models.rbac.PolicyRequest
 import com.stytch.java.b2b.models.rbac.PolicyRole
 import com.stytch.java.b2b.models.rbacorganizations.GetOrgPolicyRequest
-import com.stytch.java.b2b.models.rbacorganizations.GetOrgPolicyResponse
 import com.stytch.java.b2b.models.sessions.AuthorizationCheck
 import kotlinx.coroutines.CoroutineScope
 import kotlinx.coroutines.Job
@@ -83,7 +82,7 @@ internal class PolicyCache(
     private fun refreshOrgPolicy(orgId: String) {
         val orgs = organizations ?: client.organizations
         when (val result = orgs.getOrgPolicyCompletable(GetOrgPolicyRequest(orgId)).get()) {
-            is StytchResult.Success<GetOrgPolicyResponse> -> {
+            is StytchResult.Success -> {
                 result.value.orgPolicy?.let { orgPolicy ->
                     cachedOrgPolicies[orgId] = CachedOrgPolicy(orgPolicy, Instant.now())
                 }

stytch/src/test/kotlin/com/stytch/java/common/PolicyCacheTest.kt

@@ -255,6 +255,14 @@ internal class PolicyCacheTest {
                             policy = policy,
                         ),
                     )
+                every { organizations.getOrgPolicyCompletable(any()).get() } returns
+                    StytchResult.Success(
+                        GetOrgPolicyResponse(
+                            statusCode = 200,
+                            requestId = "",
+                            orgPolicy = OrgPolicy(roles = emptyList()),
+                        ),
+                    )
             }
 
         val policyCache = PolicyCache(rbacMock, testScope)
@@ -289,6 +297,14 @@ internal class PolicyCacheTest {
                         ),
                     )
                 }
+                every { organizations.getOrgPolicyCompletable(any()).get() } returns
+                    StytchResult.Success(
+                        GetOrgPolicyResponse(
+                            statusCode = 200,
+                            requestId = "",
+                            orgPolicy = OrgPolicy(roles = emptyList()),
+                        ),
+                    )
             }
 
         val policyCache = PolicyCache(rbacMock, testScope)
@@ -333,6 +349,14 @@ internal class PolicyCacheTest {
                             policy = policy,
                         ),
                     )
+                every { organizations.getOrgPolicyCompletable(any()).get() } returns
+                    StytchResult.Success(
+                        GetOrgPolicyResponse(
+                            statusCode = 200,
+                            requestId = "",
+                            orgPolicy = OrgPolicy(roles = emptyList()),
+                        ),
+                    )
             }
 
         val policyCache = PolicyCache(rbacMock, testScope)