org.sitoolkit.wt.domain.pageload.PageContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sit-wt-runtime Show documentation
There is a newer version: 2.3
package org.sitoolkit.wt.domain.pageload;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Comparator;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.SortedMap;
import java.util.TreeMap;

import org.sitoolkit.wt.domain.evidence.ElementPosition;
import org.sitoolkit.wt.domain.testscript.TestStep;

public class PageContext {

    private String title;

    private String url;

    private SortedMap map = new TreeMap<>(new PointComparator());

    private Map> formIdMap = new HashMap<>();

    private class PointComparator implements Comparator {

        @Override
        public int compare(ElementPosition o1, ElementPosition o2) {
            if (o1 == null | o2 == null) {
                return 0;
            }
            int dy = o1.getY() - o2.getY();

            if (dy != 0) {
                return dy;
            }

            int dx = o1.getX() - o2.getX();

            if (dx != 0) {
                return dx;
            }

            return o1.hashCode() - o2.hashCode();
        }
    }

    public void add(ElementPosition pos, TestStep step, ElementId formId) {
        map.put(pos, step);

        if (formId != null) {
            List testSteps = formIdMap.get(formId);
            if (testSteps == null) {
                testSteps = new ArrayList<>();
                formIdMap.put(formId, testSteps);
            }
            testSteps.add(step);
        }

    }

    public List asList() {
        List list = new ArrayList<>();

        int stepNo = 1;
        for (Entry entry : map.entrySet()) {
            TestStep step = entry.getValue();
            step.setNo(Integer.toString(stepNo++));
            list.add(step);
        }

        return list;
    }

    public boolean containsNameInForm(ElementId formId, String name) {
        List testSteps = formIdMap.get(formId);

        if (testSteps == null) {
            return false;
        }

        for (TestStep testStep : testSteps) {
            if (testStep.getLocator().equalsByName(name)) {
                return true;
            }

        }
        return false;
    }

    public boolean containsName(String name) {
        for (Collection testSteps : formIdMap.values()) {
            for (TestStep testStep : testSteps) {

                if (testStep.getLocator() != null && testStep.getLocator().equalsByName(name)) {
                    return true;
                }
            }
        }
        return false;
    }

    public TestStep create() {
        TestStep step = new TestStep();
        step.setCurrentCaseNo("001");
        step.setTestData("001", "y");

        return step;
    }

    public TestStep registTestStep(ElementPosition pos, ElementId formId) {
        TestStep step = create();
        add(pos, step, formId);

        return step;
    }

    public String getTitle() {
        return title;
    }

    public void setTitle(String title) {
        this.title = title;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }
}