RawResultMapTest.java

package tk.mybatis.mapper.rawresultmap;

import org.apache.ibatis.session.SqlSession;
import org.junit.Assert;
import org.junit.Test;
import tk.mybatis.mapper.base.BaseTest;
import tk.mybatis.mapper.code.Style;
import tk.mybatis.mapper.entity.Config;

import java.io.IOException;
import java.io.Reader;
import java.net.URL;
import java.util.Arrays;
import java.util.List;

/**
 * @author liuzh
 */
public class RawResultMapTest extends BaseTest {

    @Override
    protected Config getConfig() {
        Config config = super.getConfig();
        config.setStyle(Style.normal);
        return config;
    }

    @Override
    protected Reader getConfigFileAsReader() throws IOException {
        return toReader(RawResultMapTest.class.getResource("mybatis-config-rawresultmap.xml"));
    }

    @Override
    protected Reader getSqlFileAsReader() throws IOException {
        URL url = RawResultMapTest.class.getResource("CreateDB.sql");
        return toReader(url);
    }

    @Test
    public void testSelect() {
        SqlSession sqlSession = getSqlSession();
        try {
            UserMapper mapper = sqlSession.getMapper(UserMapper.class);

            List<User> users;

            System.out.println("------selectAll------");
            users = mapper.selectAll();
            userAssert(users);
            System.out.println("###################");

            System.out.println("------selectRawAnnotation------");
            users = mapper.selectRawAnnotation();
            userAssert(users);
            System.out.println("###################");

            System.out.println("------selectRawAnnotationResultMap------");
            users = mapper.selectRawAnnotationResultMap();
            userAssert(users);
            System.out.println("###################");

            System.out.println("------fetchRawResultMap------");
            users = mapper.fetchRawResultMap();
            userAssert(users);
            System.out.println("###################");

            System.out.println("------fetchDynamicResultMap------");
            users = mapper.fetchDynamicResultMap();
            userAssert(users);
            System.out.println("###################");

            System.out.println("------fetchRawResultType------");
            users = mapper.fetchRawResultType();
            userAssert(users);
            System.out.println("###################");

            System.out.println("------fetchDynamicSqlType------");
            users = mapper.fetchDynamicSqlType(Arrays.asList(1, 2, 3));
            userAssert(users);
            System.out.println("###################");

            System.out.println("------getMapUser------");
            System.out.println(mapper.getMapUser());
            System.out.println("###################");

            System.out.println(mapper.selectCount2());

        } finally {
            sqlSession.close();
        }
    }

    private static void userAssert(List<User> users) {
        users.forEach(u -> {
            System.out.println(u);
            Assert.assertNotNull(u.getUname());
            Assert.assertNotNull(u.getAge());
            Assert.assertNotNull(u.getCreateTime());
        });
    }
}