ConstantReconnectionPolicyFactoryTest.java

package systems.composable.dropwizard.cassandra.reconnection;

import com.datastax.driver.core.policies.ConstantReconnectionPolicy;
import io.dropwizard.util.Duration;
import org.junit.Test;

import static org.assertj.core.api.Assertions.assertThat;

public class ConstantReconnectionPolicyFactoryTest {

    @Test
    public void buildsPolicyWithDelayInMillis() throws Exception {
        final ConstantReconnectionPolicyFactory factory = new ConstantReconnectionPolicyFactory();
        factory.setDelay(Duration.seconds(5));

        final ConstantReconnectionPolicy policy = (ConstantReconnectionPolicy) factory.build();

        assertThat(policy.getConstantDelayMs()).isEqualTo(5000L);
    }
}