/**
* Copyright 2012-2017 Gunnar Morling (http://www.gunnarmorling.de/)
* and/or other contributors as indicated by the @authors tag. See the
* copyright.txt file in the distribution for a full listing of all
* contributors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.mapstruct.ap.test.imports.sourcewithextendsbound;
import static org.assertj.core.api.Assertions.assertThat;
import java.util.Collections;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mapstruct.ap.test.imports.sourcewithextendsbound.astronautmapper.AstronautMapper;
import org.mapstruct.ap.test.imports.sourcewithextendsbound.dto.AstronautDto;
import org.mapstruct.ap.test.imports.sourcewithextendsbound.dto.SpaceshipDto;
import org.mapstruct.ap.test.imports.sourcewithextendsbound.entity.Astronaut;
import org.mapstruct.ap.test.imports.sourcewithextendsbound.entity.Spaceship;
import org.mapstruct.ap.testutil.IssueKey;
import org.mapstruct.ap.testutil.WithClasses;
import org.mapstruct.ap.testutil.runner.AnnotationProcessorTestRunner;
import org.mapstruct.ap.testutil.runner.GeneratedSource;
/**
* Test for generating a mapper which references nested types (static inner classes).
*
* @author Gunnar Morling
*/
@WithClasses({
Astronaut.class, Spaceship.class, AstronautDto.class, SpaceshipDto.class, SpaceshipMapper.class,
AstronautMapper.class
})
@RunWith(AnnotationProcessorTestRunner.class)
public class SourceTypeContainsCollectionWithExtendsBoundTest {
private final GeneratedSource generatedSource = new GeneratedSource();
@Rule
public GeneratedSource getGeneratedSource() {
return generatedSource;
}
@Test
@IssueKey("768")
public void generatesImportsForCollectionWithExtendsBoundInSourceType() {
Astronaut astronaut = new Astronaut();
astronaut.setName( "Bob" );
Spaceship spaceship = new Spaceship();
spaceship.setAstronauts( Collections.singleton( astronaut ) );
SpaceshipDto spaceshipDto = SpaceshipMapper.INSTANCE.spaceshipToDto( spaceship );
assertThat( spaceshipDto ).isNotNull();
assertThat( spaceshipDto.getAstronauts() ).extracting( "name" ).containsOnly( "Bob" );
generatedSource.forMapper( SpaceshipMapper.class ).containsImportFor( Astronaut.class );
generatedSource.forMapper( SpaceshipMapper.class ).containsImportFor( Spaceship.class );
generatedSource.forMapper( SpaceshipMapper.class ).containsImportFor( AstronautDto.class );
generatedSource.forMapper( SpaceshipMapper.class ).containsImportFor( SpaceshipDto.class );
}
}