fixed failing test case

Author: sondermanish

app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/McpTokenServiceCEImpl.java

@@ -72,7 +72,7 @@ public McpTokenServiceCEImpl(
     }
 
     // CE: 2 live keys. EE overrides to 10.
-    protected int getMaxActiveKeysPerUser() {
+    public int getMaxActiveKeysPerUser() {
         return 2;
     }
 
@@ -103,6 +103,7 @@ private Mono<Void> validateMCPEnabled() {
 
                     return Mono.just(organization);
                 })
+                .switchIfEmpty(Mono.error(new AppsmithException(AppsmithError.INTERNAL_SERVER_ERROR)))
                 .then();
     }
 

app/server/appsmith-server/src/test/java/com/appsmith/server/services/McpTokenServiceImplTest.java

@@ -198,12 +198,15 @@ void should_rejectCreate_whenKeySpanDaysIsNotAllowed() {
     }
 
     @Test
-    void should_rejectCreate_whenUserAlreadyHasTenLiveKeys() {
-        when(mcpKeyRepository.countByUserIdAndRevokedAtIsNull(user.getId())).thenReturn(Mono.just(10L));
+    void should_rejectCreate_whenUserAlreadyHasMaxActiveKeys() {
+        Long maxPossibleKeysPerUser = Long.valueOf(service.getMaxActiveKeysPerUser());
+        when(mcpKeyRepository.countByUserIdAndRevokedAtIsNull(user.getId()))
+                .thenReturn(Mono.just(maxPossibleKeysPerUser));
 
         StepVerifier.create(service.create(user, "Claude Desktop", 30))
-                .expectErrorSatisfies(error ->
-                        assertThat(error).isInstanceOf(AppsmithException.class).hasMessageContaining("maximum of 10"))
+                .expectErrorSatisfies(error -> assertThat(error)
+                        .isInstanceOf(AppsmithException.class)
+                        .hasMessageContaining("maximum of " + maxPossibleKeysPerUser))
                 .verify();
     }