Search in sources :

Example 1 with Environment

use of de.zalando.ep.zalenium.util.Environment in project zalenium by zalando.

the class DockerSeleniumRemoteProxyTest method fallbackToDefaultValueWhenEnvVariableIsNotABoolean.

@Test
public void fallbackToDefaultValueWhenEnvVariableIsNotABoolean() {
    try {
        Environment environment = mock(Environment.class, withSettings().useConstructor());
        when(environment.getEnvVariable(DockerSeleniumRemoteProxy.ZALENIUM_VIDEO_RECORDING_ENABLED)).thenReturn("any_nonsense_value");
        when(environment.getBooleanEnvVariable(any(String.class), any(Boolean.class))).thenCallRealMethod();
        DockerSeleniumRemoteProxy.setEnv(environment);
        DockerSeleniumRemoteProxy.readEnvVars();
        Assert.assertEquals(DockerSeleniumRemoteProxy.DEFAULT_VIDEO_RECORDING_ENABLED, proxy.isVideoRecordingEnabled());
    } finally {
        DockerSeleniumRemoteProxy.restoreEnvironment();
    }
}
Also used : Environment(de.zalando.ep.zalenium.util.Environment) Test(org.junit.Test)

Example 2 with Environment

use of de.zalando.ep.zalenium.util.Environment in project zalenium by zalando.

the class DockerSeleniumRemoteProxyTest method videoRecordingIsDisabled.

@Test
public void videoRecordingIsDisabled() throws MalformedObjectNameException, IOException {
    try {
        // Create a docker-selenium container
        RegistrationRequest request = TestUtils.getRegistrationRequestForTesting(30000, DockerSeleniumStarterRemoteProxy.class.getCanonicalName());
        DockerSeleniumStarterRemoteProxy dsProxy = new DockerSeleniumStarterRemoteProxy(request, registry);
        DockerSeleniumStarterRemoteProxy.setMaxDockerSeleniumContainers(1);
        DockerSeleniumStarterRemoteProxy.setConfiguredScreenSize(DockerSeleniumStarterRemoteProxy.DEFAULT_SCREEN_SIZE);
        DockerSeleniumStarterRemoteProxy.setConfiguredTimeZone(DockerSeleniumStarterRemoteProxy.DEFAULT_TZ.getID());
        DockerSeleniumStarterRemoteProxy.setContainerClient(containerClient);
        dsProxy.getNewSession(getCapabilitySupportedByDockerSelenium());
        // Mocking the environment variable to return false for video recording enabled
        Environment environment = mock(Environment.class);
        when(environment.getEnvVariable(DockerSeleniumRemoteProxy.ZALENIUM_VIDEO_RECORDING_ENABLED)).thenReturn("false");
        when(environment.getIntEnvVariable(DockerSeleniumRemoteProxy.ZALENIUM_MAX_TEST_SESSIONS, 1)).thenReturn(1);
        // Creating a spy proxy to verify the invoked methods
        CommonProxyUtilities commonProxyUtilities = TestUtils.mockCommonProxyUtilitiesForDashboardTesting(temporaryFolder);
        TestUtils.ensureRequiredInputFilesExist(temporaryFolder);
        Dashboard.setCommonProxyUtilities(commonProxyUtilities);
        DockerSeleniumRemoteProxy spyProxy = spy(proxy);
        DockerSeleniumRemoteProxy.setEnv(environment);
        DockerSeleniumRemoteProxy.readEnvVars();
        // Start pulling thread
        spyProxy.startPolling();
        // Get a test session
        TestSession newSession = spyProxy.getNewSession(getCapabilitySupportedByDockerSelenium());
        newSession.setExternalKey(new ExternalSessionKey("DockerSeleniumRemoteProxy Test"));
        Assert.assertNotNull(newSession);
        // We start the session, in order to start recording
        WebDriverRequest webDriverRequest = mock(WebDriverRequest.class);
        HttpServletResponse response = mock(HttpServletResponse.class);
        when(webDriverRequest.getMethod()).thenReturn("POST");
        when(webDriverRequest.getRequestType()).thenReturn(RequestType.START_SESSION);
        spyProxy.afterCommand(newSession, webDriverRequest, response);
        // Assert no video recording was started, videoRecording is invoked but processContainerAction should not
        verify(spyProxy, times(1)).videoRecording(DockerSeleniumRemoteProxy.DockerSeleniumContainerAction.START_RECORDING);
        verify(spyProxy, never()).processContainerAction(DockerSeleniumRemoteProxy.DockerSeleniumContainerAction.START_RECORDING, "");
        // We release the sessions, the node should be free
        webDriverRequest = mock(WebDriverRequest.class);
        response = mock(HttpServletResponse.class);
        when(webDriverRequest.getMethod()).thenReturn("DELETE");
        when(webDriverRequest.getRequestType()).thenReturn(RequestType.STOP_SESSION);
        newSession.getSlot().doFinishRelease();
        spyProxy.afterCommand(newSession, webDriverRequest, response);
        spyProxy.afterSession(newSession);
        Assert.assertFalse(spyProxy.isBusy());
        // Now we assert that videoRecording was invoked but processContainerAction not, neither copyVideos
        verify(spyProxy, timeout(40000)).videoRecording(DockerSeleniumRemoteProxy.DockerSeleniumContainerAction.STOP_RECORDING);
        verify(spyProxy, never()).processContainerAction(DockerSeleniumRemoteProxy.DockerSeleniumContainerAction.STOP_RECORDING, "");
        verify(spyProxy, never()).copyVideos("");
    } finally {
        DockerSeleniumRemoteProxy.restoreEnvironment();
        Dashboard.restoreCommonProxyUtilities();
        ObjectName objectName = new ObjectName("org.seleniumhq.grid:type=RemoteProxy,node=\"http://localhost:30000\"");
        new JMXHelper().unregister(objectName);
    }
}
Also used : ExternalSessionKey(org.openqa.grid.internal.ExternalSessionKey) CommonProxyUtilities(de.zalando.ep.zalenium.util.CommonProxyUtilities) JMXHelper(org.openqa.selenium.remote.server.jmx.JMXHelper) TestSession(org.openqa.grid.internal.TestSession) Environment(de.zalando.ep.zalenium.util.Environment) HttpServletResponse(javax.servlet.http.HttpServletResponse) WebDriverRequest(org.openqa.grid.web.servlet.handler.WebDriverRequest) RegistrationRequest(org.openqa.grid.common.RegistrationRequest) ObjectName(javax.management.ObjectName) Test(org.junit.Test)

Example 3 with Environment

use of de.zalando.ep.zalenium.util.Environment in project zalenium by zalando.

the class DockerSeleniumStarterRemoteProxyTest method variablesGrabTheConfiguredEnvironmentVariables.

@Test
public void variablesGrabTheConfiguredEnvironmentVariables() throws MalformedObjectNameException {
    // Mock the environment class that serves as proxy to retrieve env variables
    Environment environment = mock(Environment.class, withSettings().useConstructor());
    int amountOfDesiredContainers = 7;
    int amountOfMaxContainers = 8;
    int screenWidth = 1440;
    int screenHeight = 810;
    String seleniumNodeParams = "-debug";
    TimeZone timeZone = TimeZone.getTimeZone("America/Montreal");
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.ZALENIUM_DESIRED_CONTAINERS)).thenReturn(String.valueOf(amountOfDesiredContainers));
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.ZALENIUM_MAX_DOCKER_SELENIUM_CONTAINERS)).thenReturn(String.valueOf(amountOfMaxContainers));
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.ZALENIUM_SCREEN_HEIGHT)).thenReturn(String.valueOf(screenHeight));
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.ZALENIUM_SCREEN_WIDTH)).thenReturn(String.valueOf(screenWidth));
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.ZALENIUM_TZ)).thenReturn(timeZone.getID());
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.SELENIUM_NODE_PARAMS)).thenReturn(seleniumNodeParams);
    when(environment.getIntEnvVariable(any(String.class), any(Integer.class))).thenCallRealMethod();
    when(environment.getStringEnvVariable(any(String.class), any(String.class))).thenCallRealMethod();
    DockerSeleniumStarterRemoteProxy.setEnv(environment);
    try {
        ObjectName objectName = new ObjectName("org.seleniumhq.grid:type=RemoteProxy,node=\"http://localhost:30000\"");
        new JMXHelper().unregister(objectName);
    } finally {
        DockerSeleniumStarterRemoteProxy.getNewInstance(request, registry);
    }
    Assert.assertEquals(amountOfDesiredContainers, DockerSeleniumStarterRemoteProxy.getDesiredContainersOnStartup());
    Assert.assertEquals(amountOfMaxContainers, DockerSeleniumStarterRemoteProxy.getMaxDockerSeleniumContainers());
    Assert.assertEquals(screenHeight, DockerSeleniumStarterRemoteProxy.getConfiguredScreenSize().getHeight());
    Assert.assertEquals(screenWidth, DockerSeleniumStarterRemoteProxy.getConfiguredScreenSize().getWidth());
    Assert.assertEquals(timeZone.getID(), DockerSeleniumStarterRemoteProxy.getConfiguredTimeZone().getID());
    Assert.assertEquals(seleniumNodeParams, DockerSeleniumStarterRemoteProxy.getSeleniumNodeParameters());
}
Also used : TimeZone(java.util.TimeZone) JMXHelper(org.openqa.selenium.remote.server.jmx.JMXHelper) Environment(de.zalando.ep.zalenium.util.Environment) ObjectName(javax.management.ObjectName) Test(org.junit.Test)

Example 4 with Environment

use of de.zalando.ep.zalenium.util.Environment in project zalenium by zalando.

the class DockerSeleniumStarterRemoteProxyTest method amountOfCreatedContainersIsTheConfiguredOne.

@Test
public void amountOfCreatedContainersIsTheConfiguredOne() throws MalformedObjectNameException {
    // Mock the environment class that serves as proxy to retrieve env variables
    Environment environment = mock(Environment.class, withSettings().useConstructor());
    int amountOfDesiredContainers = 7;
    when(environment.getEnvVariable(DockerSeleniumStarterRemoteProxy.ZALENIUM_DESIRED_CONTAINERS)).thenReturn(String.valueOf(amountOfDesiredContainers));
    when(environment.getIntEnvVariable(any(String.class), any(Integer.class))).thenCallRealMethod();
    DockerSeleniumStarterRemoteProxy.setEnv(environment);
    DockerSeleniumStarterRemoteProxy.setSleepIntervalMultiplier(0);
    try {
        ObjectName objectName = new ObjectName("org.seleniumhq.grid:type=RemoteProxy,node=\"http://localhost:30000\"");
        new JMXHelper().unregister(objectName);
    } finally {
        DockerSeleniumStarterRemoteProxy.getNewInstance(request, registry);
    }
    registry.add(spyProxy);
    verify(spyProxy, timeout(5000).times(amountOfDesiredContainers)).startDockerSeleniumContainer(timeZone, screenSize, true);
    Assert.assertEquals(amountOfDesiredContainers, DockerSeleniumStarterRemoteProxy.getDesiredContainersOnStartup());
}
Also used : JMXHelper(org.openqa.selenium.remote.server.jmx.JMXHelper) Environment(de.zalando.ep.zalenium.util.Environment) ObjectName(javax.management.ObjectName) Test(org.junit.Test)

Example 5 with Environment

use of de.zalando.ep.zalenium.util.Environment in project zalenium by zalando.

the class SauceLabsRemoteProxyTest method testEventIsInvoked.

@Test
public void testEventIsInvoked() throws IOException {
    try {
        // Capability which should result in a created session
        Map<String, Object> requestedCapability = new HashMap<>();
        requestedCapability.put(CapabilityType.BROWSER_NAME, BrowserType.SAFARI);
        requestedCapability.put(CapabilityType.PLATFORM_NAME, Platform.MAC);
        // Getting a test session in the sauce labs node
        TestSession testSession = sauceLabsProxy.getNewSession(requestedCapability);
        Assert.assertNotNull(testSession);
        // We release the sessions and invoke the afterCommand with a mocked object
        Environment env = mock(Environment.class);
        when(env.getBooleanEnvVariable("ZALENIUM_SEND_ANONYMOUS_USAGE_INFO", false)).thenReturn(true);
        when(env.getStringEnvVariable("ZALENIUM_GA_API_VERSION", "")).thenReturn("1");
        when(env.getStringEnvVariable("ZALENIUM_GA_TRACKING_ID", "")).thenReturn("UA-88441352");
        when(env.getStringEnvVariable("ZALENIUM_GA_ENDPOINT", "")).thenReturn("https://www.google-analytics.com/collect");
        when(env.getStringEnvVariable("ZALENIUM_GA_ANONYMOUS_CLIENT_ID", "")).thenReturn("RANDOM_STRING");
        HttpClient client = mock(HttpClient.class);
        HttpResponse httpResponse = mock(HttpResponse.class);
        when(client.execute(any(HttpPost.class))).thenReturn(httpResponse);
        GoogleAnalyticsApi ga = new GoogleAnalyticsApi();
        GoogleAnalyticsApi gaSpy = spy(ga);
        gaSpy.setEnv(env);
        gaSpy.setHttpClient(client);
        SauceLabsRemoteProxy.setGa(gaSpy);
        WebDriverRequest webDriverRequest = mock(WebDriverRequest.class);
        HttpServletResponse response = mock(HttpServletResponse.class);
        when(webDriverRequest.getMethod()).thenReturn("DELETE");
        when(webDriverRequest.getRequestType()).thenReturn(RequestType.STOP_SESSION);
        testSession.getSlot().doFinishRelease();
        testSession.setExternalKey(new ExternalSessionKey("testKey"));
        sauceLabsProxy.afterCommand(testSession, webDriverRequest, response);
        verify(gaSpy, times(1)).testEvent(anyString(), anyString(), anyLong());
    } finally {
        SauceLabsRemoteProxy.restoreGa();
    }
}
Also used : HttpPost(org.apache.http.client.methods.HttpPost) GoogleAnalyticsApi(de.zalando.ep.zalenium.util.GoogleAnalyticsApi) ExternalSessionKey(org.openqa.grid.internal.ExternalSessionKey) HashMap(java.util.HashMap) HttpResponse(org.apache.http.HttpResponse) HttpServletResponse(javax.servlet.http.HttpServletResponse) TestSession(org.openqa.grid.internal.TestSession) HttpClient(org.apache.http.client.HttpClient) Environment(de.zalando.ep.zalenium.util.Environment) WebDriverRequest(org.openqa.grid.web.servlet.handler.WebDriverRequest) Test(org.junit.Test)

Aggregations

Environment (de.zalando.ep.zalenium.util.Environment)13 Test (org.junit.Test)11 HashMap (java.util.HashMap)6 HttpServletResponse (javax.servlet.http.HttpServletResponse)6 ObjectName (javax.management.ObjectName)5 TestSession (org.openqa.grid.internal.TestSession)5 WebDriverRequest (org.openqa.grid.web.servlet.handler.WebDriverRequest)5 JMXHelper (org.openqa.selenium.remote.server.jmx.JMXHelper)4 JsonObject (com.google.gson.JsonObject)3 CommonProxyUtilities (de.zalando.ep.zalenium.util.CommonProxyUtilities)3 ServletOutputStream (javax.servlet.ServletOutputStream)3 ExternalSessionKey (org.openqa.grid.internal.ExternalSessionKey)3 JsonElement (com.google.gson.JsonElement)2 Mockito.anyString (org.mockito.Mockito.anyString)2 RegistrationRequest (org.openqa.grid.common.RegistrationRequest)2 ContainerClient (de.zalando.ep.zalenium.container.ContainerClient)1 ContainerClientRegistration (de.zalando.ep.zalenium.container.ContainerClientRegistration)1 ContainerCreationStatus (de.zalando.ep.zalenium.container.ContainerCreationStatus)1 TestInformation (de.zalando.ep.zalenium.dashboard.TestInformation)1 GoogleAnalyticsApi (de.zalando.ep.zalenium.util.GoogleAnalyticsApi)1