[shardingsphere] branch master updated: Add unit test for WeightReplicaLoadBalanceAlgorithm. (#14880)

[menghaoran]

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

menghaoran 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 53d0b08  Add unit test for WeightReplicaLoadBalanceAlgorithm. (#14880)
53d0b08 is described below

commit 53d0b08d2c438f3771a76a3840736a733c686c15
Author: zhaojinchao <33742097+zj...@users.noreply.github.com>
AuthorDate: Wed Jan 19 16:23:42 2022 +0800

    Add unit test for WeightReplicaLoadBalanceAlgorithm. (#14880)
    
    * Add unit for WeightReplicaLoadBalanceAlgorithm.
    
    * update
---
 .../loadbalance/WeightReplicaLoadBalanceAlgorithmTest.java       | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git 
a/shardingsphere-features/shardingsphere-readwrite-splitting/shardingsphere-readwrite-splitting-core/src/test/java/org/apache/shardingsphere/readwritesplitting/algorithm/loadbalance/WeightReplicaLoadBalanceAlgorithmTest.java
 
b/shardingsphere-features/shardingsphere-readwrite-splitting/shardingsphere-readwrite-splitting-core/src/test/java/org/apache/shardingsphere/readwritesplitting/algorithm/loadbalance/WeightReplicaLoadBalanceAlgorithmTest.java
index 29b7847..56caf32 100644
--- 
a/shardingsphere-features/shardingsphere-readwrite-splitting/shardingsphere-readwrite-splitting-core/src/test/java/org/apache/shardingsphere/readwritesplitting/algorithm/loadbalance/WeightReplicaLoadBalanceAlgorithmTest.java
+++ 
b/shardingsphere-features/shardingsphere-readwrite-splitting/shardingsphere-readwrite-splitting-core/src/test/java/org/apache/shardingsphere/readwritesplitting/algorithm/loadbalance/WeightReplicaLoadBalanceAlgorithmTest.java
@@ -23,6 +23,7 @@ import java.util.Arrays;
 import java.util.List;
 import java.util.Properties;
 
+import static org.hamcrest.CoreMatchers.is;
 import static org.hamcrest.CoreMatchers.notNullValue;
 import static org.junit.Assert.assertThat;
 
@@ -31,6 +32,14 @@ public final class WeightReplicaLoadBalanceAlgorithmTest {
     private final WeightReplicaLoadBalanceAlgorithm 
weightReplicaLoadBalanceAlgorithm = new WeightReplicaLoadBalanceAlgorithm();
     
     @Test
+    public void assertGetOneReadDataSource() {
+        Properties properties = new Properties();
+        properties.setProperty("test_read_ds_1", "5");
+        weightReplicaLoadBalanceAlgorithm.setProps(properties);
+        assertThat(weightReplicaLoadBalanceAlgorithm.getDataSource("ds", 
"test_write_ds", Arrays.asList("test_read_ds_1")), is("test_read_ds_1"));
+    }
+    
+    @Test
     public void assertGetDataSource() {
         final Properties properties = new Properties();
         properties.setProperty("test_read_ds_1", "5");