com.ecfeed.ui.dialogs.AddTestCaseDialog.java Source code

Java tutorial

Introduction

Here is the source code for com.ecfeed.ui.dialogs.AddTestCaseDialog.java

Source

/*******************************************************************************
 *
 * Copyright (c) 2016 ecFeed AS.                                                
 * 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 
 *  
 *******************************************************************************/

package com.ecfeed.ui.dialogs;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.jface.dialogs.IDialogConstants;
import org.eclipse.jface.dialogs.TitleAreaDialog;
import org.eclipse.jface.viewers.ArrayContentProvider;
import org.eclipse.jface.viewers.ColumnLabelProvider;
import org.eclipse.jface.viewers.ComboViewer;
import org.eclipse.jface.viewers.TableViewer;
import org.eclipse.jface.viewers.TableViewerColumn;
import org.eclipse.swt.SWT;
import org.eclipse.swt.events.ModifyEvent;
import org.eclipse.swt.events.ModifyListener;
import org.eclipse.swt.graphics.Color;
import org.eclipse.swt.layout.GridData;
import org.eclipse.swt.layout.GridLayout;
import org.eclipse.swt.widgets.Button;
import org.eclipse.swt.widgets.Combo;
import org.eclipse.swt.widgets.Composite;
import org.eclipse.swt.widgets.Control;
import org.eclipse.swt.widgets.Label;
import org.eclipse.swt.widgets.Shell;
import org.eclipse.swt.widgets.Table;
import org.eclipse.swt.widgets.TableColumn;

import com.ecfeed.core.model.ChoiceNode;
import com.ecfeed.core.model.MethodNode;
import com.ecfeed.core.model.MethodParameterNode;
import com.ecfeed.core.model.TestCaseNode;
import com.ecfeed.ui.common.ColorConstants;
import com.ecfeed.ui.common.ColorManager;
import com.ecfeed.ui.common.Constants;
import com.ecfeed.ui.common.ITestDataEditorListener;
import com.ecfeed.ui.common.Messages;
import com.ecfeed.ui.common.TestDataValueEditingSupport;

public class AddTestCaseDialog extends TitleAreaDialog implements ITestDataEditorListener {

    private MethodNode fMethod;
    private ArrayList<ChoiceNode> fTestData;
    private String fTestSuiteName;
    private Combo fTestSuiteCombo;
    private Button fOkButton;
    private TableViewer fTestDataViewer;

    /**
     * Create the dialog.
     * @param parentShell
     */
    public AddTestCaseDialog(Shell parentShell, MethodNode method) {
        super(parentShell);
        setHelpAvailable(false);
        setShellStyle(SWT.BORDER | SWT.RESIZE | SWT.TITLE | SWT.APPLICATION_MODAL);
        fTestData = new ArrayList<ChoiceNode>();
        List<MethodParameterNode> parameters = method.getMethodParameters();
        for (MethodParameterNode parameter : parameters) {
            if (parameter.isExpected()) {
                fTestData.add(createAnonymuousChoice(parameter));
            } else {

                ChoiceNode testValue = parameter.getChoices().get(0);
                while (testValue.isAbstract()) {
                    testValue = testValue.getChoices().get(0);
                }
                fTestData.add(testValue);
            }
        }
        fMethod = method;
    }

    private ChoiceNode createAnonymuousChoice(MethodParameterNode parent) {
        ChoiceNode choice = new ChoiceNode("@expected", parent.getDefaultValue());
        choice.setParent(parent);
        return choice;
    }

    /**
     * Create contents of the dialog.
     * @param parent
     */
    @Override
    protected Control createDialogArea(Composite parent) {
        setTitle(Messages.DIALOG_ADD_TEST_CASE_TITLE);
        setMessage(Messages.DIALOG_ADD_TEST_CASE_MESSAGE);
        Composite area = (Composite) super.createDialogArea(parent);
        Composite container = new Composite(area, SWT.NONE);
        container.setLayout(new GridLayout(1, false));
        container.setLayoutData(new GridData(GridData.FILL_BOTH));

        createTestSuiteComposite(container);

        createTestDataViewer(container);

        return area;
    }

    private void createTestDataViewer(Composite container) {
        fTestDataViewer = new TableViewer(container, SWT.BORDER | SWT.FULL_SELECTION);
        Table table = fTestDataViewer.getTable();
        table.setLinesVisible(true);
        table.setHeaderVisible(true);
        table.setLayoutData(new GridData(SWT.FILL, SWT.FILL, true, true, 1, 1));

        TableViewerColumn parameterViewerColumn = new TableViewerColumn(fTestDataViewer, SWT.NONE);
        TableColumn parameterColumn = parameterViewerColumn.getColumn();
        parameterColumn.setWidth(200);
        parameterColumn.setText("Parameter");
        parameterViewerColumn.setLabelProvider(new ColumnLabelProvider() {
            @Override
            public String getText(Object element) {
                return ((ChoiceNode) element).getParameter().toString();
            }

            @Override
            public Color getForeground(Object element) {
                return getColor(element);
            }
        });

        TableViewerColumn choiceViewerColumn = new TableViewerColumn(fTestDataViewer, SWT.NONE);
        TableColumn choiceColumn = choiceViewerColumn.getColumn();
        choiceColumn.setWidth(150);
        choiceColumn.setText("Choice");
        choiceViewerColumn.setLabelProvider(new ColumnLabelProvider() {
            @Override
            public String getText(Object element) {
                ChoiceNode testValue = (ChoiceNode) element;
                MethodParameterNode parameter = fMethod.getMethodParameters().get(fTestData.indexOf(testValue));
                if (parameter.isExpected()) {
                    return testValue.getValueString();
                }
                return testValue.toString();
            }

            @Override
            public Color getForeground(Object element) {
                return getColor(element);
            }
        });

        choiceViewerColumn.setEditingSupport(new TestDataValueEditingSupport(fMethod, fTestDataViewer, this));

        fTestDataViewer.setContentProvider(new ArrayContentProvider());
        fTestDataViewer.setInput(fTestData);
    }

    private Color getColor(Object element) {
        ChoiceNode choice = (ChoiceNode) element;
        MethodParameterNode parameter = fMethod.getMethodParameters().get(fTestData.indexOf(choice));
        if (parameter.isExpected()) {
            return ColorManager.getColor(ColorConstants.EXPECTED_VALUE_CATEGORY);
        }
        return null;
    }

    private void createTestSuiteComposite(Composite container) {
        Composite composite = new Composite(container, SWT.NONE);
        composite.setLayout(new GridLayout(2, false));
        composite.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false, 1, 1));

        Label testSuiteLabel = new Label(composite, SWT.NONE);
        testSuiteLabel.setLayoutData(new GridData(SWT.RIGHT, SWT.CENTER, false, false, 1, 1));
        testSuiteLabel.setText("Test suite");

        ComboViewer testSuiteViewer = new ComboViewer(composite, SWT.NONE);
        fTestSuiteCombo = testSuiteViewer.getCombo();
        fTestSuiteCombo.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false, 1, 1));
        fTestSuiteCombo.setItems(fMethod.getTestSuites().toArray(new String[] {}));
        fTestSuiteCombo.addModifyListener(new ModifyListener() {
            @Override
            public void modifyText(ModifyEvent e) {
                validateTestSuiteName();
            }
        });
        fTestSuiteCombo.setText(Constants.DEFAULT_NEW_TEST_SUITE_NAME);
    }

    /**
     * Create contents of the button bar.
     * @param parent
     */
    @Override
    protected void createButtonsForButtonBar(Composite parent) {
        fOkButton = createButton(parent, IDialogConstants.OK_ID, IDialogConstants.OK_LABEL, true);
        createButton(parent, IDialogConstants.CANCEL_ID, IDialogConstants.CANCEL_LABEL, false);
    }

    @Override
    public void okPressed() {
        fTestSuiteName = fTestSuiteCombo.getText();
        super.okPressed();
    }

    private void validateTestSuiteName() {
        if (!TestCaseNode.validateTestSuiteName(fTestSuiteCombo.getText())) {
            setErrorMessage(Messages.DIALOG_TEST_SUITE_NAME_PROBLEM_MESSAGE);
            setOkButton(false);
        } else {
            setErrorMessage(null);
            setOkButton(true);
        }
    }

    private void setOkButton(boolean enabled) {
        if (fOkButton != null && !fOkButton.isDisposed()) {
            fOkButton.setEnabled(enabled);
        }
    }

    @Override
    public void testDataChanged(int index, ChoiceNode newValue) {
        fTestData.set(index, newValue);
        fTestDataViewer.refresh();
    }

    public String getTestSuite() {
        return fTestSuiteName;
    }

    public ArrayList<ChoiceNode> getTestData() {
        return fTestData;
    }
}