package org.keycloak.testsuite.cluster;

import org.junit.Before;
import org.junit.Test;
import org.openqa.selenium.Cookie;

import static org.keycloak.testsuite.util.WaitUtils.pause;

import static org.junit.Assert.assertEquals;

/**
 *
 * @author tkyjovsk
 */
public class SessionFailoverClusterTest extends AbstractFailoverClusterTest {

    @Before
    public void beforeSessionFailover() {
        log.info("Initial node failure");
        failure();
        pause(REBALANCE_WAIT);
    }

    @Test
    public void sessionFailover() {

        boolean expectSuccessfulFailover = SESSION_CACHE_OWNERS >= 2;

        log.info("SESSION FAILOVER TEST: cluster size = " + getClusterSize() + ", session-cache owners = " + SESSION_CACHE_OWNERS
                + " --> Testsing for " + (expectSuccessfulFailover ? "" : "UN") + "SUCCESSFUL session failover.");

        assertEquals(2, getClusterSize());
        
        sessionFailover(expectSuccessfulFailover);
    }

    protected void sessionFailover(boolean expectSuccessfulFailover) {

        // LOGIN
        Cookie sessionCookie = login();

        switchFailedNode();

        // VERIFY
        if (expectSuccessfulFailover) {
            verifyLoggedIn(sessionCookie);
        } else {
            verifyLoggedOut();
            // FIXME test fails if I put re-login here
        }

        switchFailedNode();

        // VERIFY again
        if (expectSuccessfulFailover) {
            verifyLoggedIn(sessionCookie);
        } else {
            verifyLoggedOut();
            login();
        }

        // LOGOUT
        logout();
        verifyLoggedOut();

        switchFailedNode();

        // VERIFY
        verifyLoggedOut();

    }

}
