1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.apache.ibatis.submitted.discriminator;
17
18 import static org.junit.jupiter.api.Assertions.*;
19
20 import java.io.Reader;
21 import java.util.List;
22
23 import org.apache.ibatis.BaseDataTest;
24 import org.apache.ibatis.io.Resources;
25 import org.apache.ibatis.session.SqlSession;
26 import org.apache.ibatis.session.SqlSessionFactory;
27 import org.apache.ibatis.session.SqlSessionFactoryBuilder;
28 import org.junit.jupiter.api.BeforeAll;
29 import org.junit.jupiter.api.Test;
30
31 class DiscriminatorTest {
32
33 private static SqlSessionFactory sqlSessionFactory;
34
35 @BeforeAll
36 static void setUp() throws Exception {
37
38 try (Reader reader = Resources.getResourceAsReader("org/apache/ibatis/submitted/discriminator/mybatis-config.xml")) {
39 sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader);
40 }
41
42
43 BaseDataTest.runScript(sqlSessionFactory.getConfiguration().getEnvironment().getDataSource(),
44 "org/apache/ibatis/submitted/discriminator/CreateDB.sql");
45 }
46
47 @Test
48 void shouldSwitchResultType() {
49 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
50 Mapper mapper = sqlSession.getMapper(Mapper.class);
51 List<Vehicle> vehicles = mapper.selectVehicles();
52 assertEquals(Car.class, vehicles.get(0).getClass());
53 assertEquals(Integer.valueOf(5), ((Car)vehicles.get(0)).getDoorCount());
54 assertEquals(Truck.class, vehicles.get(1).getClass());
55 assertEquals(Float.valueOf(1.5f), ((Truck)vehicles.get(1)).getCarryingCapacity());
56 }
57 }
58
59 @Test
60 void shouldInheritResultType() {
61
62 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
63 Mapper mapper = sqlSession.getMapper(Mapper.class);
64 List<Owner> owners = mapper.selectOwnersWithAVehicle();
65 assertEquals(Truck.class, owners.get(0).getVehicle().getClass());
66 assertEquals(Car.class, owners.get(1).getVehicle().getClass());
67 }
68 }
69
70 }