基于 QCADOO的 开源MES，生产制造管理系统Manufacturing Execution System

/** * *************************************************************************** * Copyright (c) 2010 Qcadoo Limited * Project: Qcadoo Framework * Version: 1.4 * * This file is part of Qcadoo. * * Qcadoo is free software; you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published * by the Free Software Foundation; either version 3 of the License, * or (at your option) any later version. * * This program 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA * *************************************************************************** */ package com.qcadoo.model.internal; import static junit.framework.Assert.*; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertSame; import static org.mockito.BDDMockito.given; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; import java.math.BigDecimal; import java.util.Date; import java.util.List; import java.util.Map; import org.junit.Assert; import org.junit.Before; import org.junit.Test; import com.google.common.collect.Lists; import com.google.common.collect.Maps; import com.qcadoo.model.api.DataDefinition; import com.qcadoo.model.api.Entity; import com.qcadoo.model.api.EntityList; import com.qcadoo.model.api.FieldDefinition; import com.qcadoo.model.api.types.BelongsToType; import com.qcadoo.model.api.types.FieldType; import com.qcadoo.model.api.types.HasManyType; import com.qcadoo.model.api.types.ManyToManyType; import com.qcadoo.model.internal.types.DecimalType; import com.qcadoo.model.internal.types.IntegerType; public class DefaultEntityTest { private DefaultEntity defaultEntity; private DataDefinition dataDefinition; private FieldDefinition belongsToFieldDefinition; private DataDefinition belongsToFieldDataDefinition; private FieldDefinition decimalFieldDefinition; private FieldDefinition integerFieldDefinition; private FieldDefinition manyToManyFieldDefinition; private DataDefinition manyToManyDataDefinition; private FieldDefinition booleanFieldDefinition; private FieldDefinition stringFieldDefinition; private FieldDefinition hasManyFieldDefinition; private DataDefinition hasManyDataDefinition; private FieldDefinition secondManyToManyFieldDefinition; private DataDefinition secondManyToManyDataDefinition; private static final String BELONGS_TO_FIELD_NAME = "belongsToField"; private static final String STRING_FIELD_NAME = "stringField"; private static final String BOOLEAN_FIELD_NAME = "booleanField"; private static final String DECIMAL_FIELD_NAME = "decimalField"; private static final String INTEGER_FIELD_NAME = "integerField"; private static final String MANY_TO_MANY_FIELD_NAME = "manyToMany"; private static final String SECOND_MANY_TO_MANY_FIELD_NAME = "secondManyToMany"; private static final String HAS_MANY_FIELD_NAME = "hasMany"; private static final String L_FIRST = "first"; private static final String L_SECOND = "second"; private static final String L_THIRD = "third"; private static final String L_FOURTH = "fourth"; private static final String L_DIFFERENCE = "difference"; private static final String DATE_FIELD_NAME = "dateField"; @Before public final void init() { belongsToFieldDefinition = mock(FieldDefinition.class); final BelongsToType belongsToType = mock(BelongsToType.class); when(belongsToFieldDefinition.getType()).thenReturn(belongsToType); belongsToFieldDataDefinition = mock(DataDefinition.class); when(belongsToFieldDefinition.getDataDefinition()).thenReturn(belongsToFieldDataDefinition); decimalFieldDefinition = mock(FieldDefinition.class); final DecimalType decimalType = new DecimalType(); when(decimalFieldDefinition.getType()).thenReturn(decimalType); integerFieldDefinition = mock(FieldDefinition.class); final IntegerType integerType = new IntegerType(); when(integerFieldDefinition.getType()).thenReturn(integerType); manyToManyFieldDefinition = mock(FieldDefinition.class); final ManyToManyType manyToManyType = mock(ManyToManyType.class); when(manyToManyFieldDefinition.getType()).thenReturn(manyToManyType); manyToManyDataDefinition = mock(DataDefinition.class); when(manyToManyFieldDefinition.getDataDefinition()).thenReturn(manyToManyDataDefinition); secondManyToManyFieldDefinition = mock(FieldDefinition.class); when(secondManyToManyFieldDefinition.getType()).thenReturn(manyToManyType); secondManyToManyDataDefinition = mock(DataDefinition.class); when(secondManyToManyFieldDefinition.getDataDefinition()).thenReturn(secondManyToManyDataDefinition); hasManyFieldDefinition = mock(FieldDefinition.class); final HasManyType hasManyType = mock(HasManyType.class); when(hasManyType.getJoinFieldName()).thenReturn(BELONGS_TO_FIELD_NAME); when(hasManyFieldDefinition.getType()).thenReturn(hasManyType); hasManyDataDefinition = mock(DataDefinition.class); when(hasManyFieldDefinition.getDataDefinition()).thenReturn(hasManyDataDefinition); stringFieldDefinition = mock(FieldDefinition.class); when(stringFieldDefinition.isPersistent()).thenReturn(false); dataDefinition = mock(DataDefinition.class); booleanFieldDefinition = mock(FieldDefinition.class); defaultEntity = new DefaultEntity(dataDefinition); final Map<String, FieldDefinition> fieldsMap = Maps.newHashMap(); fieldsMap.put(BELONGS_TO_FIELD_NAME, belongsToFieldDefinition); fieldsMap.put(STRING_FIELD_NAME, stringFieldDefinition); fieldsMap.put(BOOLEAN_FIELD_NAME, booleanFieldDefinition); fieldsMap.put(DECIMAL_FIELD_NAME, decimalFieldDefinition); fieldsMap.put(INTEGER_FIELD_NAME, integerFieldDefinition); fieldsMap.put(MANY_TO_MANY_FIELD_NAME, manyToManyFieldDefinition); fieldsMap.put(SECOND_MANY_TO_MANY_FIELD_NAME, secondManyToManyFieldDefinition); fieldsMap.put(HAS_MANY_FIELD_NAME, hasManyFieldDefinition); for (Map.Entry<String, FieldDefinition> fieldEntry : fieldsMap.entrySet()) { when(dataDefinition.getField(fieldEntry.getKey())).thenReturn(fieldEntry.getValue()); } when(dataDefinition.getFields()).thenReturn(fieldsMap); } @Test public final void shouldGetBelongsToFieldReturnEntity() throws Exception { // given Entity belongsToEntity = mock(Entity.class); defaultEntity.setField(BELONGS_TO_FIELD_NAME, belongsToEntity); // when Entity returnedEntity = defaultEntity.getBelongsToField(BELONGS_TO_FIELD_NAME); // then assertEquals(belongsToEntity, returnedEntity); } @Test public final void shouldGetBelongsToFieldReturnProxyEntity() throws Exception { // given Long belongsToEntityId = 1L; defaultEntity.setField(BELONGS_TO_FIELD_NAME, belongsToEntityId); // when Entity returnedEntity = defaultEntity.getBelongsToField(BELONGS_TO_FIELD_NAME); // then assertEquals(belongsToEntityId, returnedEntity.getId()); } @Test public final void shouldGetBelongsToFieldReturnProxyEntityUsingIntegerValue() throws Exception { // given Integer belongsToEntityId = 1; defaultEntity.setField(BELONGS_TO_FIELD_NAME, belongsToEntityId); // when Entity returnedEnti