View Javadoc
1   /*
2    *    Copyright 2009-2021 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    *       http://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.type;
17  
18  import static org.junit.jupiter.api.Assertions.assertEquals;
19  import static org.junit.jupiter.api.Assertions.assertNull;
20  import static org.mockito.Mockito.never;
21  import static org.mockito.Mockito.verify;
22  import static org.mockito.Mockito.when;
23  
24  import org.junit.jupiter.api.Test;
25  
26  class EnumTypeHandlerTest extends BaseTypeHandlerTest {
27  
28    enum MyEnum {
29      ONE,
30      TWO
31    }
32  
33    private static final TypeHandler<MyEnum> TYPE_HANDLER = new EnumTypeHandler<MyEnum>(MyEnum.class);
34  
35    @Override
36    @Test
37    public void shouldSetParameter() throws Exception {
38      TYPE_HANDLER.setParameter(ps, 1, MyEnum.ONE, null);
39      verify(ps).setString(1, "ONE");
40    }
41  
42    @Test
43    public void shouldSetNullParameter() throws Exception {
44      TYPE_HANDLER.setParameter(ps, 1, null, JdbcType.VARCHAR);
45      verify(ps).setNull(1, JdbcType.VARCHAR.TYPE_CODE);
46    }
47  
48    @Override
49    @Test
50    public void shouldGetResultFromResultSetByName() throws Exception {
51      when(rs.getString("column")).thenReturn("ONE");
52      assertEquals(MyEnum.ONE, TYPE_HANDLER.getResult(rs, "column"));
53      verify(rs, never()).wasNull();
54    }
55  
56    @Override
57    @Test
58    public void shouldGetResultNullFromResultSetByName() throws Exception {
59      when(rs.getString("column")).thenReturn(null);
60      assertNull(TYPE_HANDLER.getResult(rs, "column"));
61      verify(rs, never()).wasNull();
62    }
63  
64    @Override
65    @Test
66    public void shouldGetResultFromResultSetByPosition() throws Exception {
67      when(rs.getString(1)).thenReturn("ONE");
68      assertEquals(MyEnum.ONE, TYPE_HANDLER.getResult(rs, 1));
69      verify(rs, never()).wasNull();
70    }
71  
72    @Override
73    @Test
74    public void shouldGetResultNullFromResultSetByPosition() throws Exception {
75      when(rs.getString(1)).thenReturn(null);
76      assertNull(TYPE_HANDLER.getResult(rs, 1));
77      verify(rs, never()).wasNull();
78    }
79  
80    @Override
81    @Test
82    public void shouldGetResultFromCallableStatement() throws Exception {
83      when(cs.getString(1)).thenReturn("ONE");
84      assertEquals(MyEnum.ONE, TYPE_HANDLER.getResult(cs, 1));
85      verify(cs, never()).wasNull();
86    }
87  
88    @Override
89    @Test
90    public void shouldGetResultNullFromCallableStatement() throws Exception {
91      when(cs.getString(1)).thenReturn(null);
92      assertNull(TYPE_HANDLER.getResult(cs, 1));
93      verify(cs, never()).wasNull();
94    }
95  
96  }