Commit cf082b3c authored by terrymanu's avatar terrymanu
Browse files

For #647: add YamlMasterSlaveRuleConfiguration test case

parent 17289461

sharding-core/src/test/java/io/shardingjdbc/core/yaml/AllYamlTests.java

+3 −1
Original line number Diff line number Diff line
@@ -18,6 +18,7 @@ 
package io.shardingjdbc.core.yaml;



import io.shardingjdbc.core.yaml.masterslave.YamlMasterSlaveConfigurationTest;

import io.shardingjdbc.core.yaml.masterslave.YamlMasterSlaveRuleConfigurationTest;

import io.shardingjdbc.core.yaml.sharding.YamlShardingConfigurationTest;

import org.junit.runner.RunWith;

import org.junit.runners.Suite;
@@ -26,7 +27,8 @@ import org.junit.runners.Suite.SuiteClasses; 
@RunWith(Suite.class)

@SuiteClasses({

        YamlShardingConfigurationTest.class, 

        YamlMasterSlaveConfigurationTest.class

        YamlMasterSlaveConfigurationTest.class, 

        YamlMasterSlaveRuleConfigurationTest.class

})

public class AllYamlTests {

}

sharding-core/src/test/java/io/shardingjdbc/core/yaml/masterslave/YamlMasterSlaveRuleConfigurationTest.java

0 → 100644
+56 −0
Original line number Diff line number Diff line 
/*

 * Copyright 1999-2015 dangdang.com.

 * <p>

 * Licensed 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.

 * </p>

 */



package io.shardingjdbc.core.yaml.masterslave;



import io.shardingjdbc.core.api.algorithm.masterslave.MasterSlaveLoadBalanceAlgorithmType;

import io.shardingjdbc.core.api.config.MasterSlaveRuleConfiguration;

import org.hamcrest.CoreMatchers;

import org.junit.Test;



import java.util.Arrays;

import java.util.Collection;



import static org.hamcrest.CoreMatchers.is;

import static org.junit.Assert.assertThat;



public final class YamlMasterSlaveRuleConfigurationTest {

    

    @Test

    public void assertGetMasterSlaveRuleConfiguration() {

        MasterSlaveRuleConfiguration actual = createYamlMasterSlaveRuleConfig().getMasterSlaveRuleConfiguration();

        assertMasterSlaveRuleConfig(actual);

    }

        

    private YamlMasterSlaveRuleConfiguration createYamlMasterSlaveRuleConfig() {

        YamlMasterSlaveRuleConfiguration result = new YamlMasterSlaveRuleConfiguration();

        result.setName("master_slave_ds");

        result.setMasterDataSourceName("master_ds");

        result.setSlaveDataSourceNames(Arrays.asList("slave_ds_0", "slave_ds_1"));

        result.setLoadBalanceAlgorithmType(MasterSlaveLoadBalanceAlgorithmType.RANDOM);

        result.setLoadBalanceAlgorithmClassName("TestLoadBalanceAlgorithmClassName");

        return result;

    }

    

    private void assertMasterSlaveRuleConfig(final MasterSlaveRuleConfiguration actual) {

        assertThat(actual.getName(), is("master_slave_ds"));

        assertThat(actual.getMasterDataSourceName(), is("master_ds"));

        assertThat(actual.getSlaveDataSourceNames(), CoreMatchers.<Collection<String>>is(Arrays.asList("slave_ds_0", "slave_ds_1")));

        assertThat(actual.getLoadBalanceAlgorithmType(), is(MasterSlaveLoadBalanceAlgorithmType.RANDOM));

        assertThat(actual.getLoadBalanceAlgorithmClassName(), is("TestLoadBalanceAlgorithmClassName"));

    }

}