1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.apache.ibatis.submitted.enum_with_method;
17
18 import java.io.Reader;
19
20 import org.apache.ibatis.BaseDataTest;
21 import org.apache.ibatis.io.Resources;
22 import org.apache.ibatis.session.SqlSession;
23 import org.apache.ibatis.session.SqlSessionFactory;
24 import org.apache.ibatis.session.SqlSessionFactoryBuilder;
25 import org.junit.jupiter.api.Assertions;
26 import org.junit.jupiter.api.BeforeAll;
27 import org.junit.jupiter.api.Test;
28
29 class EnumWithMethodTest {
30
31 private static SqlSessionFactory sqlSessionFactory;
32
33 @BeforeAll
34 static void setUp() throws Exception {
35
36 try (Reader reader = Resources
37 .getResourceAsReader("org/apache/ibatis/submitted/enum_with_method/mybatis-config.xml")) {
38 sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader);
39 }
40
41
42 BaseDataTest.runScript(sqlSessionFactory.getConfiguration().getEnvironment().getDataSource(),
43 "org/apache/ibatis/submitted/enum_with_method/CreateDB.sql");
44 }
45
46 @Test
47 void shouldGetAUser() {
48 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
49 Mapper mapper = sqlSession.getMapper(Mapper.class);
50 User user = mapper.getUser(1);
51 Assertions.assertEquals("User1", user.getName());
52 Assertions.assertEquals(Mood.GOOD, user.getMood());
53 }
54 }
55
56 @Test
57 void shouldInsertAUser() {
58 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
59 Mapper mapper = sqlSession.getMapper(Mapper.class);
60 User user = new User();
61 user.setId(2);
62 user.setName("User2");
63 user.setCur(Currency.Dollar);
64 user.setMood(Mood.BAD);
65 mapper.insertUser(user);
66 sqlSession.commit();
67 }
68 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
69 Mapper mapper = sqlSession.getMapper(Mapper.class);
70 User user = mapper.getUser(2);
71 Assertions.assertEquals("User2", user.getName());
72 Assertions.assertEquals(Mood.BAD, user.getMood());
73 }
74 }
75
76 }