vaadin-framework/uitest/src/main/java/com/vaadin/tests/components/abstractcomponent/ChangeHierarchyBeforeResponse.java

/*
 * Copyright 2000-2016 Vaadin Ltd.
 *
 * Licensed 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 com.vaadin.tests.components.abstractcomponent;

import com.vaadin.server.VaadinRequest;
import com.vaadin.tests.components.AbstractReindeerTestUI;
import com.vaadin.ui.Button;
import com.vaadin.ui.CssLayout;
import com.vaadin.ui.Label;

public class ChangeHierarchyBeforeResponse extends AbstractReindeerTestUI {
    private CssLayout layout = new CssLayout() {
        @Override
        public void beforeClientResponse(boolean initial) {
            super.beforeClientResponse(initial);
            if (initial) {
                addComponent(buttonToAdd);
                removeComponent(labelToRemove);
            }
        }
    };

    private Label labelToRemove = new Label("Label to remove") {
        int count = 0;

        @Override
        public void beforeClientResponse(boolean initial) {
            super.beforeClientResponse(initial);
            if (initial) {
                count++;
                setValue("Initial count: " + count);
            }
        }
    };

    private Button buttonToAdd = new Button("Added from beforeClientResponse",
            event -> layout.addComponent(labelToRemove)) {
        @Override
        public void beforeClientResponse(boolean initial) {
            super.beforeClientResponse(initial);
            setCaption("Add label to layout");
        }
    };

    @Override
    protected void setup(VaadinRequest request) {
        layout.addComponent(labelToRemove);

        addComponent(layout);
    }

}