This is an automated email from the ASF dual-hosted git repository.
jackietien pushed a commit to branch ty/SessionPoolTest
in repository https://gitbox.apache.org/repos/asf/iotdb.git
The following commit(s) were added to refs/heads/ty/SessionPoolTest by this
push:
new f5e3e7d4185 change some uts
f5e3e7d4185 is described below
commit f5e3e7d41852408e147a86e4c03b9633ac0fe555
Author: JackieTien97 <[email protected]>
AuthorDate: Tue Jan 2 15:55:14 2024 +0800
change some uts
---
.../java/org/apache/iotdb/session/Session.java | 2 +-
.../iotdb/session/SessionCacheLeaderTest.java | 1 +
.../iotdb/session/SessionConnectionTest.java | 26 +++++++++++++++++++---
.../java/org/apache/iotdb/session/SessionTest.java | 9 +++++++-
.../session/pool/SessionPoolExceptionTest.java | 9 +++++++-
5 files changed, 41 insertions(+), 6 deletions(-)
diff --git
a/iotdb-client/session/src/main/java/org/apache/iotdb/session/Session.java
b/iotdb-client/session/src/main/java/org/apache/iotdb/session/Session.java
index 5ad9e3c3d15..b05514cf821 100644
--- a/iotdb-client/session/src/main/java/org/apache/iotdb/session/Session.java
+++ b/iotdb-client/session/src/main/java/org/apache/iotdb/session/Session.java
@@ -160,7 +160,7 @@ public class Session implements ISession {
protected Version version;
// default enable
- private boolean enableAutoFetch = true;
+ protected boolean enableAutoFetch = true;
private static final String REDIRECT_TWICE = "redirect twice";
diff --git
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionCacheLeaderTest.java
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionCacheLeaderTest.java
index b98bcf47b19..ea148312623 100644
---
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionCacheLeaderTest.java
+++
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionCacheLeaderTest.java
@@ -973,6 +973,7 @@ public class SessionCacheLeaderTest {
SessionConfig.DEFAULT_MAX_FRAME_SIZE,
enableRedirection,
SessionConfig.DEFAULT_VERSION);
+ this.enableAutoFetch = false;
}
@Override
diff --git
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionConnectionTest.java
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionConnectionTest.java
index 6cc28c35a80..75f75bc9243 100644
---
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionConnectionTest.java
+++
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionConnectionTest.java
@@ -86,7 +86,13 @@ public class SessionConnectionTest {
sessionConnection = new SessionConnection();
Whitebox.setInternalState(sessionConnection, "transport", transport);
Whitebox.setInternalState(sessionConnection, "client", client);
- session = new Session(Arrays.asList("127.0.0.1:12"), "root", "root");
+ session =
+ new Session.Builder()
+ .nodeUrls(Collections.singletonList("127.0.0.1:12"))
+ .username("root")
+ .password("root")
+ .enableAutoFetch(false)
+ .build();
Whitebox.setInternalState(sessionConnection, "session", session);
Mockito.when(transport.isOpen()).thenReturn(true);
TSStatus tsStatus = new
TSStatus(TSStatusCode.REDIRECTION_RECOMMEND.getStatusCode());
@@ -151,7 +157,14 @@ public class SessionConnectionTest {
@Test(expected = NumberFormatException.class)
public void testBuildSessionConnection() throws IoTDBConnectionException {
- session = new Session("local", 12, "root", "root");
+ session =
+ new Session.Builder()
+ .host("local")
+ .port(12)
+ .username("root")
+ .password("root")
+ .enableAutoFetch(false)
+ .build();
SessionConnection sessionConnection1 =
new SessionConnection(
session,
@@ -161,7 +174,14 @@ public class SessionConnectionTest {
@Test(expected = IoTDBConnectionException.class)
public void testBuildSessionConnection2() throws IoTDBConnectionException {
- session = new Session("local", 12, "root", "root");
+ session =
+ new Session.Builder()
+ .host("local")
+ .port(12)
+ .username("root")
+ .password("root")
+ .enableAutoFetch(false)
+ .build();
SessionConnection sessionConnection1 =
new SessionConnection(
session,
diff --git
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionTest.java
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionTest.java
index 14085e9fe69..c1c8a76a244 100644
---
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionTest.java
+++
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/SessionTest.java
@@ -67,7 +67,14 @@ public class SessionTest {
@Before
public void setUp() throws IoTDBConnectionException,
StatementExecutionException {
MockitoAnnotations.initMocks(this);
- session = new Session("host", 11, "user", "pwd");
+ session =
+ new Session.Builder()
+ .host("host")
+ .port(11)
+ .username("user")
+ .password("pwd")
+ .enableAutoFetch(false)
+ .build();
Whitebox.setInternalState(session, "defaultSessionConnection",
sessionConnection);
TSQueryTemplateResp resp = new TSQueryTemplateResp();
resp.setMeasurements(Arrays.asList("root.sg1.d1.s1"));
diff --git
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/pool/SessionPoolExceptionTest.java
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/pool/SessionPoolExceptionTest.java
index 8e6a16c12e9..b2a0749ee5f 100644
---
a/iotdb-client/session/src/test/java/org/apache/iotdb/session/pool/SessionPoolExceptionTest.java
+++
b/iotdb-client/session/src/test/java/org/apache/iotdb/session/pool/SessionPoolExceptionTest.java
@@ -69,7 +69,14 @@ public class SessionPoolExceptionTest {
public void setUp() {
MockitoAnnotations.initMocks(this);
- sessionPool = new SessionPool(Arrays.asList("host:11"), "user",
"password", 10);
+ sessionPool =
+ new SessionPool.Builder()
+ .nodeUrls(Collections.singletonList("host:11"))
+ .user("user")
+ .password("password")
+ .maxSize(10)
+ .enableAutoFetch(false)
+ .build();
ConcurrentLinkedDeque<ISession> queue = new ConcurrentLinkedDeque<>();
queue.add(session);
Whitebox.setInternalState(sessionPool, "queue", queue);
