})) {
            KerberosCredentials credentials = new KerberosCredentials(LOGIN_CONTEXT_NAME);
            KerberosKey[] keys = credentials.getKeys();
            assertEquals(2, keys.length);
        }
    }

    /**
     * Test getKeys method when subject has no Kerberos keys.
     *
     * @throws LoginException if login fails.
     */
    @Test

        assertArgumentNotNull("bundle", bundle);

        final Map<String, String> ret = newHashMap();
        for (final Enumeration<String> e = bundle.getKeys(); e.hasMoreElements();) {
            final String key = e.nextElement();
            final String value = bundle.getString(key);
            ret.put(key, value);
        }
        return ret;
    }

        this.subject = lc.getSubject();
    }

    /**
     * Retrieves all Kerberos keys from the authenticated subject.
     *
     * @return array of KerberosKey objects
     */
    public KerberosKey[] getKeys() {
        List<Key> serverKeys = new ArrayList<>();

        Set<Object> serverPrivateCredentials = this.subject.getPrivateCredentials();
        for (Object credential : serverPrivateCredentials) {

                byte[] crypt = encOctets.getOctets();

                if (keys == null) {
                    try {
                        keys = new KerberosCredentials().getKeys();
                    } catch (LoginException e) {
                        throw new PACDecodingException("Login failure", e);
                    }
                }

        Smb2LeaseKey key1 = new Smb2LeaseKey();
        Smb2LeaseKey key2 = new Smb2LeaseKey();
        
        assertNotNull(key1.getKey());
        assertEquals(16, key1.getKey().length);
        assertFalse(Arrays.equals(key1.getKey(), key2.getKey()));
    }
    
    @Test
    public void testLeaseStateFlags() {
        int state = Smb2LeaseState.SMB2_LEASE_READ_WRITE;
        

    private String serverAddress;
    private int flags;
    
    // Getters and setters...
}

class WitnessRegisterResponse {
    private String registrationId;
    private int returnCode;
    
    public boolean isSuccess() { return returnCode == 0; }
    public String getError() { return "Error code: " + returnCode; }
    // Getters and setters...
}

class WitnessUnregisterRequest {

        decodedContext.decode(buffer, 0, buffer.length);

        // Verify decoded values
        assertArrayEquals(originalKey.getKey(), decodedContext.getLeaseKey().getKey());
        assertEquals(originalLeaseState, decodedContext.getLeaseState());
        assertEquals(originalScope, decodedContext.getCacheScope());
        assertEquals(60000L, decodedContext.getMaxCacheAge());

        assertTrue(result);
        assertEquals(1, chain.changePasswordCalls.size());
        assertEquals("testuser", chain.changePasswordCalls.get(0).getKey());
        assertEquals("newpassword123", chain.changePasswordCalls.get(0).getValue());
    }

    // Test changePassword with failure
    public void test_changePassword_failure() {

            this.uniqueEndpoints = uniqueEndpoints;
            this.healthCheckingEnabled = healthCheckingEnabled;
            this.proactiveHealthCheckEnabled = proactiveHealthCheckEnabled;
        }

        // Getters
        public int getActiveConnections() {
            return activeConnections;
        }

        public int getNonPooledConnections() {
            return nonPooledConnections;
        }

            when(mockReferralData.getKey()).thenReturn(key);
            doNothing().when(mockReferralData).setKey(key);

            mockReferralData.setKey(key);
            String retrievedKey = mockReferralData.getKey();

            assertEquals(key, retrievedKey);
            verify(mockReferralData, times(1)).setKey(key);
            verify(mockReferralData, times(1)).getKey();