org.openqa.grid.e2e.misc.GridDistributionTest.java Source code

Introduction

Here is the source code for org.openqa.grid.e2e.misc.GridDistributionTest.java

Source

// Licensed to the Software Freedom Conservancy (SFC) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The SFC licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

package org.openqa.grid.e2e.misc;

import junit.framework.Assert;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.openqa.grid.common.GridRole;
import org.openqa.grid.e2e.utils.GridTestHelper;
import org.openqa.grid.e2e.utils.RegistryTestHelper;
import org.openqa.grid.internal.ProxySet;
import org.openqa.grid.internal.RemoteProxy;
import org.openqa.grid.internal.TestSlot;
import org.openqa.grid.internal.utils.SelfRegisteringRemote;
import org.openqa.grid.web.Hub;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.server.SeleniumServer;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class GridDistributionTest {

    private static Hub hub;
    private static List<WebDriver> drivers = new ArrayList<>();

    @BeforeClass
    public static void prepare() throws Exception {

        hub = GridTestHelper.getHub();

        for (int i = 0; i < 8; i++) {
            SelfRegisteringRemote remote = GridTestHelper.getRemoteWithoutCapabilities(hub, GridRole.NODE);

            remote.addBrowser(DesiredCapabilities.chrome(), 3);
            remote.setRemoteServer(new SeleniumServer(remote.getConfiguration()));
            remote.startRemoteServer();
            remote.sendRegistrationRequest();
            RegistryTestHelper.waitForNode(hub.getRegistry(), i + 1);
        }
    }

    @Test
    public void testLoadIsDistributedEvenly() throws Throwable {

        for (int i = 0; i < 8; i++) {
            drivers.add(GridTestHelper.getRemoteWebDriver(DesiredCapabilities.chrome(), hub));
        }

        ProxySet ps = hub.getRegistry().getAllProxies();

        for (RemoteProxy p : ps) {
            int freeslots = 0;
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() == null) {
                    freeslots++;
                }
            }
            Assert.assertEquals("checking proxy free slots, all should have only one session running", freeslots,
                    2);
        }

        for (int i = 0; i < 8; i++) {
            drivers.add(GridTestHelper.getRemoteWebDriver(DesiredCapabilities.chrome(), hub));
        }

        for (RemoteProxy p : ps) {
            int freeslots = 0;
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() == null) {
                    freeslots++;
                }
            }
            Assert.assertEquals("checking proxy free slots, all should have two sessions running", freeslots, 1);
        }

        drivers.add(GridTestHelper.getRemoteWebDriver(DesiredCapabilities.chrome(), hub));

        Boolean foundOneFull = false;
        for (RemoteProxy p : ps) {
            int freeslots = 0;
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() == null) {
                    freeslots++;
                }
            }
            if (freeslots == 0) {
                if (!foundOneFull) {
                    foundOneFull = true;
                } else {
                    throw new RuntimeException("Found more than one node with all test slots running sessions");
                }
            }
        }
    }

    @Test
    public void testLeastRecentlyUsedNodesPickedFirst() throws Throwable {
        ProxySet ps = hub.getRegistry().getAllProxies();

        for (int i = 0; i < 4; i++) {
            drivers.add(GridTestHelper.getRemoteWebDriver(DesiredCapabilities.chrome(), hub));
        }

        Set<String> chosenNodes = new HashSet<>();

        for (RemoteProxy p : ps) {
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() != null) {
                    chosenNodes.add(p.getRemoteHost().toString());
                    break;
                }
            }
        }

        stopDrivers(drivers);

        for (int i = 0; i < 4; i++) {
            drivers.add(GridTestHelper.getRemoteWebDriver(DesiredCapabilities.chrome(), hub));
        }

        for (RemoteProxy p : ps) {
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() != null) {
                    Assert.assertFalse(
                            "Should not be immediately reused: " + p.getRemoteHost().toString()
                                    + " previously used nodes: " + chosenNodes,
                            chosenNodes.contains(p.getRemoteHost().toString()));
                    break;
                }
            }
        }

        chosenNodes.clear();

        for (RemoteProxy p : ps) {
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() != null) {
                    chosenNodes.add(p.getRemoteHost().toString());
                    break;
                }
            }
        }

        stopDrivers(drivers);

        for (int i = 0; i < 4; i++) {
            drivers.add(GridTestHelper.getRemoteWebDriver(DesiredCapabilities.chrome(), hub));
        }

        for (RemoteProxy p : ps) {
            for (TestSlot ts : p.getTestSlots()) {
                if (ts.getSession() != null) {
                    Assert.assertFalse(
                            "Should not be immediately reused: " + p.getRemoteHost().toString()
                                    + " previously used nodes: " + chosenNodes,
                            chosenNodes.contains(p.getRemoteHost().toString()));
                    break;
                }
            }
        }
    }

    private static void stopDrivers(List<WebDriver> drivers) {
        for (WebDriver driver : drivers) {
            try {
                driver.quit();
            } catch (Exception e) {
                System.out.println(e.toString());
            }
        }
        drivers.clear();
    }

    @AfterClass
    public static void stop() throws Exception {
        stopDrivers(drivers);
        hub.stop();
    }
}