com.qspin.qtaste.javaguifx.server.TableRowCounter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javagui-fx Show documentation
The newest version!
/*
    Copyright 2007-2012 QSpin - www.qspin.be

    This file is part of QTaste framework.

    QTaste is free software: you can redistribute it and/or modify
    it under the terms of the GNU Lesser General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    QTaste is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Lesser General Public License for more details.

    You should have received a copy of the GNU Lesser General Public License
    along with QTaste. If not, see .
*/

package com.qspin.qtaste.javaguifx.server;

import java.awt.Component;
import java.awt.Label;

import javax.swing.JLabel;
import javax.swing.JTable;
import javax.swing.SwingUtilities;
import javax.swing.table.TableCellRenderer;
import javax.swing.table.TableModel;

import javafx.scene.Node;

import com.qspin.qtaste.testsuite.QTasteException;
import com.qspin.qtaste.testsuite.QTasteTestFailException;

final class TableRowCounter extends ComponentCommander {

    private int lastCounter = 0;

    @Override
    Integer executeCommand(int timeout, String componentName, Object... data) throws QTasteException {
        Node c = getComponentByName(componentName);
        //		if ( c == null || !(c instanceof JTable) )
        //		{
        //			throw new QTasteTestFailException("The component " + componentName + " is not a JTable.");
        //		}
        //		JTable table = (JTable)c;
        //		TableModel model = table.getModel();
        //		int columnIndex = getColumnIndex(data[0].toString(), model);
        //		return countRows(data[1].toString(), columnIndex, table);
        return 0;
    }

    private int countRows(final String pValue, final int pColumnIndex, final JTable pTable) throws QTasteException {
        try {
            SwingUtilities.invokeAndWait(() -> {
                for (int rowIndex = 0; rowIndex < pTable.getModel().getRowCount(); ++rowIndex) {
                    //format model value
                    Object cellValue = pTable.getModel().getValueAt(rowIndex, pColumnIndex);
                    TableCellRenderer renderer = pTable.getCellRenderer(rowIndex, pColumnIndex);
                    Component c = renderer.getTableCellRendererComponent(pTable, cellValue, false, false, rowIndex, pColumnIndex);
                    String valueRepresentation;
                    if (c instanceof Label) {
                        LOGGER.debug("cell is represented by a Label");
                        valueRepresentation = ((Label) c).getText();
                    } else if (c instanceof JLabel) {
                        LOGGER.debug("cell is represented by a JLabel");
                        valueRepresentation = ((JLabel) c).getText();
                    } else {
                        LOGGER.debug("cell is represented by a " + c.getClass());
                        valueRepresentation = c.toString();
                    }

                    //check with searched value
                    LOGGER.debug("compare value (" + pValue + ") with the cell value (" + valueRepresentation + ")");
                    if (pValue.equals(valueRepresentation)) {
                        lastCounter++;
                    }
                }
            });
        } catch (Exception ex) {
            //LOGGER.fatal(ex.getMessage(), ex);
            throw new QTasteTestFailException("Error counting the rows " + ex.getMessage());
        }
        return lastCounter;
    }

    private int getColumnIndex(String pColumnName, TableModel pModel) throws QTasteTestFailException {
        for (int columnIndex = 0; columnIndex < pModel.getColumnCount(); columnIndex++) {
            if (pColumnName.equals(pModel.getColumnName(columnIndex))) {
                return columnIndex;
            }
        }
        throw new QTasteTestFailException("The column " + pColumnName + " is not found.");
    }

}