/**
   * @see <a href="https://jira.spring.io/browse/DATACASS-226">DATACASS-226</a>
   * @throws Exception
   */
  @Test
  public void shouldSetLoadBalancingPolicy() throws Exception {

    LoadBalancingPolicy loadBalancingPolicy = new RoundRobinPolicy();

    CassandraCqlClusterFactoryBean bean = new CassandraCqlClusterFactoryBean();
    bean.setLoadBalancingPolicy(loadBalancingPolicy);
    bean.afterPropertiesSet();

    assertThat(getConfiguration(bean).getPolicies().getLoadBalancingPolicy())
        .isEqualTo(loadBalancingPolicy);
  }