@Test public void testFindUniqueAttributes() { final AttributeDescriptorModel attributeDescriptor1 = new AttributeDescriptorModel(); attributeDescriptor1.setUnique(Boolean.TRUE); final Set<String> unique = new HashSet<String>(); unique.add("unique1"); final ItemModelConverter modelConverter = mock(ItemModelConverter.class); when(modelConverter.getUniqueAttributes()).thenReturn(unique); when(converterRegistry.getModelConverterBySourceType("model1")).thenReturn(modelConverter); final Set<String> actualUnique = typeService.getUniqueAttributes("model1"); Assert.assertEquals("One attribute expected", 1, actualUnique.size()); Assert.assertEquals("Unexpected attribute got", "unique1", actualUnique.iterator().next()); }
@Test public void testGetUniqueAttributesWithoutModelConverter() { final AttributeDescriptorModel attributeDescriptor1 = new AttributeDescriptorModel(); attributeDescriptor1.setUnique(Boolean.TRUE); final ModelConverter nonItemModelConverter = mock(ModelConverter.class); final ComposedTypeModel nonItemModel = mock(ComposedTypeModel.class); when(converterRegistry.getModelConverterBySourceType("nonItemModel")) .thenReturn(nonItemModelConverter); when((nonItemModel).getDeclaredattributedescriptors()) .thenReturn(Collections.singletonList(attributeDescriptor1)); when(typeDao.findComposedTypeByCode("nonItemModel")).thenReturn(nonItemModel); final Set<String> actualUnique = typeService.getUniqueAttributes("nonItemModel"); Assert.assertEquals("One attribute expected", 1, actualUnique.size()); }