1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.apache.ibatis.submitted.cursor_nested;
17
18 import java.io.Reader;
19 import java.util.Iterator;
20
21 import org.apache.ibatis.BaseDataTest;
22 import org.apache.ibatis.cursor.Cursor;
23 import org.apache.ibatis.io.Resources;
24 import org.apache.ibatis.session.RowBounds;
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.Assertions;
29 import org.junit.jupiter.api.BeforeAll;
30 import org.junit.jupiter.api.Test;
31
32 class CursorNestedTest {
33
34 private static SqlSessionFactory sqlSessionFactory;
35
36 @BeforeAll
37 static void setUp() throws Exception {
38
39 try (Reader reader = Resources.getResourceAsReader("org/apache/ibatis/submitted/cursor_nested/mybatis-config.xml")) {
40 sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader);
41 }
42
43
44 BaseDataTest.runScript(sqlSessionFactory.getConfiguration().getEnvironment().getDataSource(),
45 "org/apache/ibatis/submitted/cursor_nested/CreateDB.sql");
46 }
47
48 @Test
49 void shouldGetAllUser() {
50 Cursor<User> usersCursor;
51 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
52 Mapper mapper = sqlSession.getMapper(Mapper.class);
53 usersCursor = mapper.getAllUsers();
54
55 Assertions.assertFalse(usersCursor.isOpen());
56
57 Iterator<User> iterator = usersCursor.iterator();
58
59
60 Assertions.assertTrue(iterator.hasNext());
61 Assertions.assertTrue(usersCursor.isOpen());
62 Assertions.assertFalse(usersCursor.isConsumed());
63
64 User user = iterator.next();
65 Assertions.assertEquals(2, user.getGroups().size());
66 Assertions.assertEquals(3, user.getRoles().size());
67
68 user = iterator.next();
69 Assertions.assertEquals(1, user.getGroups().size());
70 Assertions.assertEquals(3, user.getRoles().size());
71
72 user = iterator.next();
73 Assertions.assertEquals(3, user.getGroups().size());
74 Assertions.assertEquals(1, user.getRoles().size());
75
76 user = iterator.next();
77 Assertions.assertEquals(2, user.getGroups().size());
78 Assertions.assertEquals(2, user.getRoles().size());
79
80 Assertions.assertTrue(usersCursor.isOpen());
81 Assertions.assertFalse(usersCursor.isConsumed());
82
83
84 Assertions.assertFalse(iterator.hasNext());
85 Assertions.assertFalse(usersCursor.isOpen());
86 Assertions.assertTrue(usersCursor.isConsumed());
87 }
88 Assertions.assertFalse(usersCursor.isOpen());
89 }
90
91 @Test
92 void testCursorWithRowBound() {
93 try (SqlSession sqlSession = sqlSessionFactory.openSession()) {
94 Cursor<User> usersCursor = sqlSession.selectCursor("getAllUsers", null, new RowBounds(2, 1));
95
96 Iterator<User> iterator = usersCursor.iterator();
97
98 Assertions.assertTrue(iterator.hasNext());
99 User user = iterator.next();
100 Assertions.assertEquals("User3", user.getName());
101 Assertions.assertEquals(2, usersCursor.getCurrentIndex());
102
103 Assertions.assertFalse(iterator.hasNext());
104 Assertions.assertFalse(usersCursor.isOpen());
105 Assertions.assertTrue(usersCursor.isConsumed());
106 }
107 }
108 }