uitest/src/main/java/com/vaadin/tests/widgetset/server/TestWidgetComponent.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

package com.vaadin.tests.widgetset.server;

import com.google.gwt.user.client.ui.Widget;
import com.vaadin.annotations.Widgetset;
import com.vaadin.tests.widgetset.TestingWidgetSet;
import com.vaadin.tests.widgetset.client.TestWidgetConnector;
import com.vaadin.tests.widgetset.client.TestWidgetConnector.TestWidgetState;
import com.vaadin.ui.AbstractComponent;
import com.vaadin.ui.UI;

/**
 * Testing component that shows any widget class inside the
 * com.vaadin.tests.widgetset.client package.
 */
public class TestWidgetComponent extends AbstractComponent {
    private static final String targetPackage = TestWidgetConnector.class
            .getPackage().getName();

    public TestWidgetComponent(Class<? extends Widget> widgetClass) {
        String className = widgetClass.getCanonicalName();
        if (!className.startsWith(targetPackage)) {
            throw new IllegalArgumentException(
                    "Widget class must be inside the " + targetPackage
                            + " package");
        }

        getState().widgetClass = className;
        setSizeFull();
    }

    @Override
    public void attach() {
        super.attach();

        Class<? extends UI> uiClass = getUI().getClass();

        Widgetset widgetset = uiClass.getAnnotation(Widgetset.class);
        if (widgetset == null
                || !widgetset.value().equals(TestingWidgetSet.NAME)) {
            throw new IllegalStateException(
                    "You must add @" + Widgetset.class.getSimpleName() + "("
                            + TestingWidgetSet.class.getSimpleName()
                            + ".NAME) to " + uiClass.getSimpleName());
        }
    }

    @Override
    protected TestWidgetState getState() {
        return (TestWidgetState) super.getState();
    }

}