/* * Copyright (C) 2005-2011 Alfresco Software Limited. * * This file is part of Alfresco * * Alfresco 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. * * Alfresco 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 Alfresco. If not, see . */ package org.alfresco.util.schemacomp.model; import static org.mockito.Mockito.verify; import java.util.ArrayList; import java.util.List; import org.alfresco.util.schemacomp.Result.Strength; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mock; import org.mockito.Mockito; import org.mockito.runners.MockitoJUnitRunner; /** * Tests for the Table class. * * @author Matt Ward */ @RunWith(MockitoJUnitRunner.class) public class TableTest extends DbObjectTestBase { private Table table; private Table otherTable; private List columns; private @Mock PrimaryKey primaryKey; private List foreignKeys; private List indexes; @Before public void setUp() throws Exception { columns = listOfMocks(Column.class, 3); foreignKeys = listOfMocks(ForeignKey.class, 1); indexes = listOfMocks(Index.class, 1); table = new Table(null, "the_table", columns, primaryKey, foreignKeys, indexes); otherTable = new Table(null, "the_other_table", columns, primaryKey, foreignKeys, indexes); } private List listOfMocks(Class c, int size) { List list = new ArrayList(size); for (int i = 0; i < size; i++) { list.add((T) Mockito.mock(c)); } return list; } @Override protected List getMocksUsedInDiff() { List mocks = super.getMocksUsedInDiff(); mocks.add(primaryKey); return mocks; } public void doDiffTests() { // Check columns inOrder.verify(comparisonUtils).compareCollections(table.getColumns(), otherTable.getColumns(), ctx); // Check primary key inOrder.verify(primaryKey).diff(otherTable.getPrimaryKey(), ctx, Strength.ERROR); // Check foreign keys inOrder.verify(comparisonUtils).compareCollections( table.getForeignKeys(), otherTable.getForeignKeys(), ctx); // Check indexes inOrder.verify(comparisonUtils).compareCollections( table.getIndexes(), otherTable.getIndexes(), ctx); } @Override protected Table getThisObject() { return table; } @Override protected Table getThatObject() { return otherTable; } @Test public void acceptVisitor() { table.setColumns(columns); table.setForeignKeys(foreignKeys); table.setIndexes(indexes); table.setPrimaryKey(primaryKey); table.accept(visitor); // All the children should be visited List children = new ArrayList(); children.addAll(columns); children.addAll(foreignKeys); children.addAll(indexes); children.add(primaryKey); for (DbObject child : children) { verify(child).accept(visitor); } // The parent itself should be visited verify(visitor).visit(table); } }