PortalCookiesTest.java

package org.bonitasoft.console.common.server.login;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;

import org.junit.Test;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;

public class PortalCookiesTest {

    @Test
    public void should_add_tenant_cookie_to_response() throws Exception {
        MockHttpServletResponse response = new MockHttpServletResponse();

        PortalCookies.addTenantCookieToResponse(response, 123L);

        assertThat(response.getCookie("bonita.tenant").getValue()).isEqualTo("123");
    }

    @Test
    public void should_get_tenant_cookie_from_request() throws Exception {
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setCookies(new Cookie("bonita.tenant", "123"));

        String tenant = PortalCookies.getTenantCookieFromRequest(request);

        assertThat(tenant).isEqualTo("123");
    }
}