스프링부트에서 DbUnit 을 이용하여 DB 테스트 해보기

@Service
public class UserService {
    // ... 주입받는 클래스 생략

    @Transactional
    public User register(UserDto userDto) {

        Company company = companyRepository.findById(userDto.getCompanyId())
                .orElseThrow(EntityNotFoundException::new);

        User user = User.builder()
                .name(userDto.getName())
                .password(userDto.getPassword())
                .email(userDto.getEmail())
                .build();

        user.setCompany(company); // 조회한 Company 설정
        return userRepository.save(user);
    }

}

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
@Transactional
class UserServiceTest {
    // ... 주입받는 클래스 생략

    @Test
    void register() throws Exception {
        Company company1 = companyRepository.save(new Company("company1"));
        Company company2 = companyRepository.save(new Company("company2"));

        UserDto userDto1 = UserDto.builder()
                .name("name!!")
                .password("pass!!")
                .email("email1@email.com")
                .companyId(company1.getId())
                .build();

        UserDto userDto2 = UserDto.builder()
                .name("name!!")
                .password("pass!!")
                .email("email2@email.com")
                .companyId(company2.getId())
                .build();

        User savedUser1 = userService.register(userDto1);
        User savedUser2 = userService.register(userDto2);

        User actual1 = userRepository.findById(savedUser1.getId()).get();
        User actual2 = userRepository.findById(savedUser2.getId()).get();

        assertThat(actual1.getEmail()).isEqualTo(userDto1.getEmail());
        assertThat(actual1.getPassword()).isEqualTo(userDto1.getPassword());
        assertThat(actual1.getName()).isEqualTo(userDto1.getName());
        assertThat(actual1.getCompany()).isEqualTo(company1);

        assertThat(actual2.getEmail()).isEqualTo(userDto2.getEmail());
        assertThat(actual2.getPassword()).isEqualTo(userDto2.getPassword());
        assertThat(actual2.getName()).isEqualTo(userDto2.getName());
        assertThat(actual2.getCompany()).isEqualTo(company2);
    }

}

@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
@Transactional
public class UserServiceTestOnlyDbUnit {
    // ... 주입받는 클래스 생략

    @Autowired
    private DataSource dataSource;

    private Connection connection;
        private IDatabaseConnection iDatabaseConnection;
        private IDataSet flatXmlDataSet;
    
        @BeforeEach
        void setup() throws Exception {
            connection = dataSource.getConnection();
            iDatabaseConnection = new MySqlConnection(connection, "dbunit"); // DbUnit 이 사용할 Connection 설정
    
            InputStream is = this.getClass().getResourceAsStream("company.xml"); // 테스트용 데이터셋 설정
            flatXmlDataSet = new FlatXmlDataSetBuilder().build(is);
            DatabaseOperation.CLEAN_INSERT.execute(iDatabaseConnection, flatXmlDataSet); // 테스트용 데이터셋 DB 에 입력
        }
    
        @AfterEach
        void tearDown() throws Exception {
            // DatabaseOperation.DELETE_ALL.execute(iDatabaseConnection, flatXmlDataSet); // 입력한 데이터셋 삭제
            if (connection != null) {
                connection.close();
            }
            if (iDatabaseConnection != null) {
                iDatabaseConnection.close();
            }
        }

}

public class MySqlConnection extends DatabaseConnection
{
    public MySqlConnection(Connection connection, String schema) throws DatabaseUnitException
    {
        super(connection, schema);
        getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,
                new MySqlDataTypeFactory());
        getConfig().setProperty(DatabaseConfig.PROPERTY_METADATA_HANDLER, 
                new MySqlMetadataHandler());
    }
}

@Configuration
public class TestConfig {

    @Bean
    public DatabaseConfigBean dbUnitDatabaseConfig() {
        DatabaseConfigBean config = new DatabaseConfigBean();
        config.setAllowEmptyFields(true);
        config.setDatatypeFactory(new MySqlDataTypeFactory());
        config.setMetadataHandler(new MySqlMetadataHandler());
        return config;
    }

    @Bean
    public DatabaseDataSourceConnectionFactoryBean dbUnitDatabaseConnection(DataSource dataSource) {
        DatabaseDataSourceConnectionFactoryBean dbUnitDatabaseConnection = new DatabaseDataSourceConnectionFactoryBean();
        dbUnitDatabaseConnection.setDataSource(dataSource);
        dbUnitDatabaseConnection.setDatabaseConfig(dbUnitDatabaseConfig());
        dbUnitDatabaseConnection.setSchema("dbunit");
        return dbUnitDatabaseConnection;
    }

}

package woowa.dbunit.example.dbunit;
// ... 
@Transactional
@Import(TestConfig.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.NONE)
@TestExecutionListeners({
        DbUnitTestExecutionListener.class,
        DependencyInjectionTestExecutionListener.class, // Bean 을 DI 받기 위해 선언해줘야 한다.
//        TransactionDbUnitTestExecutionListener.class // @DatabaseSetup 부터 @DatabaseTearDown 까지 트랜잭션이 적용된다.
                                                    // @DatabaseTearDown 선언시 DbUnitTestExecutionListener 와 함께 사용할 수 없다.
})
@DbUnitConfiguration(databaseConnection = "dbUnitDatabaseConnection")
@DatabaseSetup(value = {"company.xml", "user.xml"}, type = DatabaseOperation.CLEAN_INSERT)
@DatabaseTearDown(value = {"company.xml", "user.xml"}, type = DatabaseOperation.DELETE_ALL)
public class UserServiceTestWithSpringDbUnit {
    // ...
}