de.dentrassi.pm.testing.ChannelTester.java Source code

Introduction

Here is the source code for de.dentrassi.pm.testing.ChannelTester.java

Source

/*******************************************************************************
 * Copyright (c) 2015 IBH SYSTEMS GmbH.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     IBH SYSTEMS GmbH - initial API and implementation
 *******************************************************************************/
package de.dentrassi.pm.testing;

import java.io.File;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

import org.junit.Assert;
import org.openqa.selenium.By;
import org.openqa.selenium.WebElement;

public class ChannelTester {
    private static final String ROW_ID_PREFIX = "row-";

    private final WebContext context;

    private final String id;

    public ChannelTester(final WebContext context, final String id) {
        this.context = context;
        this.id = id;
    }

    /**
     * Get the channel id
     *
     * @return the channel id
     */
    public String getId() {
        return this.id;
    }

    public static ChannelTester create(final WebContext context, final String name) {
        // get before list of channels

        final Set<String> before = getAllChannelIds(context);

        // create channel

        context.getDriver().get(context.resolve("/channel/createWithRecipe"));
        final WebElement element = context.findElement(By.id("name"));
        element.sendKeys(name);
        element.submit();

        // get after list of channels

        final Set<String> after = getAllChannelIds(context);
        after.removeAll(before);

        if (after.isEmpty()) {
            throw new RuntimeException(String.format("Channel '%s' did not get created", name));
        }

        if (after.size() > 1) {
            throw new RuntimeException(
                    String.format("More than one channel was created when adding channel '%s'", name));
        }

        // return new channel

        return new ChannelTester(context, after.iterator().next());
    }

    public static Set<String> getAllChannelIds(final WebContext context) {
        context.getDriver().get(context.resolve("/channel"));

        final Set<String> result = new HashSet<>();

        for (final WebElement ele : context.findElements(By.cssSelector(".channel-id > a"))) {
            result.add(ele.getText());
        }

        return result;
    }

    public Set<String> getAllArtifactIds() {
        get(String.format("/channel/%s/viewPlain", this.id));

        final Set<String> result = new HashSet<>();

        for (final WebElement ele : this.context.findElements(By.tagName("tr"))) {
            final String id = ele.getAttribute("id");
            System.out.println("Entry: " + id);
            if (id == null || !id.startsWith(ROW_ID_PREFIX)) {
                continue;
            }

            result.add(id.substring(ROW_ID_PREFIX.length()));
        }

        return result;
    }

    public void addAspect(final String aspectId) {
        get(String.format("/channel/%s/aspects", this.id));

        // find the div panel

        final WebElement ele = this.context.findElement(By.id(aspectId));

        // ensure the group is selected

        final String groupDivId = ele.findElement(By.xpath("parent::*")).getAttribute("id");
        System.out.println("Group ID: " + groupDivId);
        final String path = String.format("//a[@aria-controls='%s']", groupDivId);
        System.out.println("XPath: " + path);
        this.context.findElement(By.xpath(path)).click();

        // click the add button

        final WebElement btn = ele.findElement(By.className("btn-default"));
        btn.click();

        final WebElement mr = this.context.findElement(By.id("modal-requires"));
        if (mr.isDisplayed()) {
            mr.findElement(By.id("modal-req-with")).click();
        }

        // check if the aspect was really added

        final boolean added = internalGetAspects("aspect-assigned").contains(aspectId);
        Assert.assertTrue(String.format("Aspect %s was not added to the channel", aspectId), added);
    }

    public Set<String> getAssignedAspects() {
        get(String.format("/channel/%s/aspects", this.id));
        return internalGetAspects("aspect-assigned");
    }

    protected Set<String> internalGetAspects(final String className) {
        final List<WebElement> elements = this.context.findElements(By.className(className));

        final Set<String> result = new HashSet<>(elements.size());

        for (final WebElement ele : elements) {
            final String id = ele.getAttribute("id");
            if (id != null) {
                result.add(id);
            }
        }

        return result;
    }

    /**
     * Upload a file to the channel
     *
     * @param localFileName
     *            the file to upload
     * @return a set of artifact ids which got created by this operation. This
     *         list may be empty or contain one or more artifacts
     */
    public Set<String> upload(final String localFileName) {
        final Set<String> before = getAllArtifactIds();

        get(String.format("/channel/%s/add", this.id));

        final WebElement link = this.context.findElement(By.linkText("Upload"));

        Assert.assertTrue(link.findElement(By.xpath("..")).getAttribute("class").contains("active"));
        ;

        // upload file

        final File input = this.context.getTestFile(localFileName);

        {
            final WebElement ele = this.context.findElement(By.id("file"));
            Assert.assertNotNull(ele);

            ele.sendKeys(input.toString());
            ele.submit();
        }

        final Set<String> after = getAllArtifactIds();

        after.removeAll(before);

        return after;
    }

    protected void get(final String url) {
        this.context.getResolved(url);
    }
}