io.sterodium.extensions.capability.CustomCapabilityMatcher Maven / Gradle / Ivy

Go to download
package io.sterodium.extensions.capability;

import org.openqa.grid.internal.utils.DefaultCapabilityMatcher;

import java.util.Map;
import java.util.logging.Logger;

/**
 * @author Alexey Nikolaenko [email protected]
 *         Date: 20/09/2015
 *
 *         Default capability matcher considers only 4 capabilities:
 *         PLATFORM, BROWSER_NAME, VERSION, applicationName.
 *         There are capabilities like FirefoxDriver.PROFILE which are used to pass profile settings to the node, these need to be ignored.
 *         Custom capability matcher makes extension capability checks in addition to the basic ones.
 */
@SuppressWarnings("unused")
public class CustomCapabilityMatcher extends DefaultCapabilityMatcher {

    private static final Logger LOGGER = Logger.getLogger(CustomCapabilityMatcher.class.getName());

    @Override
    public boolean matches(Map nodeCapability, Map requestedCapability) {
        boolean basicChecks = super.matches(nodeCapability, requestedCapability);

        boolean customChecks = extensionCapabilityCheck(nodeCapability, requestedCapability);

        return basicChecks && customChecks;
    }

    private boolean extensionCapabilityCheck(Map nodeCapability, Map requestedCapability) {
        for (String capability : requestedCapability.keySet()) {
            if (capability.startsWith(Capabilities.EXTENSION_PREFIX)) {
                if (!requestedCapability.get(capability).equals(nodeCapability.get(capability))) {
                    return false;
                }
            }
        }
        return true;
    }
}