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.assertEquals;
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 (
39 Reader reader = Resources.getResourceAsReader("org/apache/ibatis/submitted/discriminator/mybatis-config.xml")) {
40 sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader);
41 }
42
43
44 BaseDataTest.runScript(sqlSessionFactory.getConfiguration().getEnvironment().getDataSource(),
45 "org/apache/ibatis/submitted/discriminator/CreateDB.sql");
46 }
47
48 @Test
49 void shouldSwitchResultType() {
50 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
51 Mapper mapper = sqlSession.getMapper(Mapper.class);
52 List<Vehicle> vehicles = mapper.selectVehicles();
53 assertEquals(Car.class, vehicles.get(0).getClass());
54 assertEquals(Integer.valueOf(5), ((Car) vehicles.get(0)).getDoorCount());
55 assertEquals(Truck.class, vehicles.get(1).getClass());
56 assertEquals(Float.valueOf(1.5f), ((Truck) vehicles.get(1)).getCarryingCapacity());
57 }
58 }
59
60 @Test
61 void shouldInheritResultType() {
62
63 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
64 Mapper mapper = sqlSession.getMapper(Mapper.class);
65 List<Owner> owners = mapper.selectOwnersWithAVehicle();
66 assertEquals(Truck.class, owners.get(0).getVehicle().getClass());
67 assertEquals(Car.class, owners.get(1).getVehicle().getClass());
68 }
69 }
70
71 @Test
72 void shouldBeAppliedToResultMapInConstructorArg() {
73 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
74 Mapper mapper = sqlSession.getMapper(Mapper.class);
75 List<Owner> owners = mapper.selectOwnersWithAVehicleConstructor();
76 assertEquals(Truck.class, owners.get(0).getVehicle().getClass());
77 assertEquals(Car.class, owners.get(1).getVehicle().getClass());
78 }
79 }
80
81 @Test
82 void shouldBeAppliedToResultMapInConstructorArgNested() {
83 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
84 Mapper mapper = sqlSession.getMapper(Mapper.class);
85 List<Contract> contracts = mapper.selectContracts();
86 assertEquals(2, contracts.size());
87 assertEquals(Truck.class, contracts.get(0).getOwner().getVehicle().getClass());
88 assertEquals(Car.class, contracts.get(1).getOwner().getVehicle().getClass());
89 }
90 }
91
92 }