1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16 package org.apache.ibatis.datasource.unpooled;
17
18 import static org.junit.jupiter.api.Assertions.assertEquals;
19
20 import java.sql.Connection;
21
22 import org.apache.ibatis.datasource.pooled.PooledDataSource;
23 import org.apache.ibatis.testcontainers.PgContainer;
24 import org.junit.jupiter.api.Tag;
25 import org.junit.jupiter.api.Test;
26
27 @Tag("TestcontainersTests")
28 class NetworkTimeoutTest {
29
30 @Test
31 void testNetworkTimeout_UnpooledDataSource() throws Exception {
32 UnpooledDataSource dataSource = (UnpooledDataSource) PgContainer.getUnpooledDataSource();
33 dataSource.setDefaultNetworkTimeout(5000);
34 try (Connection connection = dataSource.getConnection()) {
35 assertEquals(5000, connection.getNetworkTimeout());
36 }
37 }
38
39 @Test
40 void testNetworkTimeout_PooledDataSource() throws Exception {
41 UnpooledDataSource unpooledDataSource = (UnpooledDataSource) PgContainer.getUnpooledDataSource();
42 PooledDataSource dataSource = new PooledDataSource(unpooledDataSource);
43 dataSource.setDefaultNetworkTimeout(5000);
44 try (Connection connection = dataSource.getConnection()) {
45 assertEquals(5000, connection.getNetworkTimeout());
46 }
47 }
48
49 }