[shardingsphere] branch master updated: test added for JDBCContext (#25920)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhonghongsheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/shardingsphere.git


The following commit(s) were added to refs/heads/master by this push:
     new 4c587437a83 test added for JDBCContext (#25920)
4c587437a83 is described below

commit 4c587437a839288ac8c74e4ba7ef90ef2537c65a
Author: Salman <sa...@gmail.com>
AuthorDate: Thu Jun 1 04:03:28 2023 +0200

    test added for JDBCContext (#25920)
    
    * test added for JDBCContext
    
    * fails fixed and remarks changed
    
    * review remarks fixed
---
 .../driver/jdbc/context/JDBCContextTest.java       | 65 ++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/jdbc/core/src/test/java/org/apache/shardingsphere/driver/jdbc/context/JDBCContextTest.java b/jdbc/core/src/test/java/org/apache/shardingsphere/driver/jdbc/context/JDBCContextTest.java
new file mode 100644
index 00000000000..137624f78d0
--- /dev/null
+++ b/jdbc/core/src/test/java/org/apache/shardingsphere/driver/jdbc/context/JDBCContextTest.java
@@ -0,0 +1,65 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.shardingsphere.driver.jdbc.context;
+
+import org.apache.shardingsphere.driver.state.circuit.datasource.CircuitBreakerDataSource;
+import org.apache.shardingsphere.mode.manager.cluster.coordinator.registry.config.event.datasource.DataSourceChangedEvent;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import javax.sql.DataSource;
+import java.sql.SQLException;
+import java.util.HashMap;
+import java.util.Map;
+
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
+import static org.mockito.Mockito.mock;
+
+@ExtendWith(MockitoExtension.class)
+class JDBCContextTest {
+
+    @Test
+    void assertNullCachedDbMetadataWithEmptyDatasource() throws Exception {
+        JDBCContext actual = new JDBCContext(new HashMap<>());
+        assertNull(actual.getCachedDatabaseMetaData());
+    }
+
+    @Test
+    void assertNotNullCashedDbMetadataWith() throws SQLException {
+        Map<String, DataSource> dataSourceMap = getStringDataSourceMap();
+        JDBCContext jdbcContext = new JDBCContext(dataSourceMap);
+        assertNotNull(jdbcContext.getCachedDatabaseMetaData());
+    }
+
+    @Test
+    void assetNullMetadataAfterRefreshingExisting() throws SQLException {
+        Map<String, DataSource> stringDataSourceMap = getStringDataSourceMap();
+        JDBCContext jdbcContext = new JDBCContext(stringDataSourceMap);
+        DataSourceChangedEvent event = mock();
+        jdbcContext.refreshCachedDatabaseMetaData(event);
+        assertNull(jdbcContext.getCachedDatabaseMetaData());
+    }
+
+    private static Map<String, DataSource> getStringDataSourceMap() {
+        Map<String, DataSource> result = new HashMap<>();
+        result.put("test_db", new CircuitBreakerDataSource());
+        return result;
+    }
+}