View Javadoc
1   /*
2    *    Copyright 2009-2023 the original author or authors.
3    *
4    *    Licensed under the Apache License, Version 2.0 (the "License");
5    *    you may not use this file except in compliance with the License.
6    *    You may obtain a copy of the License at
7    *
8    *       https://www.apache.org/licenses/LICENSE-2.0
9    *
10   *    Unless required by applicable law or agreed to in writing, software
11   *    distributed under the License is distributed on an "AS IS" BASIS,
12   *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   *    See the License for the specific language governing permissions and
14   *    limitations under the License.
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      // create an SqlSessionFactory
38      try (
39          Reader reader = Resources.getResourceAsReader("org/apache/ibatis/submitted/discriminator/mybatis-config.xml")) {
40        sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader);
41      }
42  
43      // populate in-memory database
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      // #486
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  }