RoiTableModelTest.java

/*
 * Copyright (C) 2007 Snorre Gylterud, Stein Magnus Jodal, Johannes Knutsen,
 * Erik Bagge Ottesen, Ralf Bjarne Taraldset, and Iterate AS
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2
 * as published by the Free Software Foundation.
 */

package no.ntnu.mmfplanner.ui.model;

import static org.junit.Assert.*;

import javax.swing.table.TableModel;

import no.ntnu.mmfplanner.ProjectTestFixture;
import no.ntnu.mmfplanner.model.Mmf;
import no.ntnu.mmfplanner.model.MmfException;
import no.ntnu.mmfplanner.model.ProjectRoi;

import org.junit.Before;
import org.junit.Test;

/**
 * Test suite for {@link RoiTableModel}
 */
public class RoiTableModelTest extends ProjectTestFixture {

    TableModel model;

    @Before
    public void setUp() throws Exception {
        model = new RoiTableModel(project, false);
    }

    @Test
    public void testIsCellEditable() {
        for (int i = 0; i < 40; i++) {
            assertFalse(model.isCellEditable(i, i));
        }
    }

    @Test
    public void testGetColumnCount() throws MmfException {
        assertEquals(14, model.getColumnCount());
        project.setPeriods(63);
        assertEquals(65, model.getColumnCount());
        project.setPeriods(1);
        assertEquals(3, model.getColumnCount());
    }

    @Test
    public void testGetColumnNameInt() {
        assertEquals(RoiTableModel.COLUMN_MMF, model.getColumnName(0));

        for (int i = 1; i < 13; i++) {
            assertEquals("" + (i), model.getColumnName(i));
        }
        assertEquals(RoiTableModel.COLUMN_NET, model.getColumnName(13));
    }

    @Test
    public void testGetRowCount() {
        assertEquals(9, model.getRowCount());
        project.add(new Mmf("B", "Test B"));
        assertEquals(10, model.getRowCount());
        project.remove(mmfA);
        project.remove(mmfB);
        project.remove(project.get(0));
        assertEquals(7, model.getRowCount());
    }

    @Test
    public void testGetValueAt() {
        // We only test values and rollingNpv. We already know ProjectRoi is
        // correct, so no need to test the actual values
        ProjectRoi roi = ProjectRoi.getRoiTable(project, project.getInterestRate(), false);
        for (int i = 0; i < roi.rollingNpv.length; i++) {
            assertEquals(roi.rollingNpv[i], model.getValueAt(7, i + 1));
        }

        for (int i = 0; i < roi.values.length; i++) {
            boolean isNull = true;
            for (int j = 0; j < roi.values[i].length; j++) {
                isNull = isNull && (0 == roi.values[i][j]);
                if (isNull) {
                    assertNull(model.getValueAt(i, j + 1));
                } else {
                    assertEquals(roi.values[i][j], model.getValueAt(i, j + 1));
                }
            }
        }
    }
}