X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;ds=sidebyside;f=tests%2Ffunction.c;fp=tests%2Ffunction.c;h=c95d3348ae353d79854b716cad55c2f9db49567c;hb=604540c15315c6ed876082529f639697ba3eceab;hp=84a053a4ddc529c000f01105aa6ed74dd238e628;hpb=f10c3a3677bdd5a1848e7497a674a938956d7f3d;p=awesomized%2Flibmemcached

diff --git a/tests/function.c b/tests/function.c
index 84a053a4..c95d3348 100644
--- a/tests/function.c
+++ b/tests/function.c
@@ -2856,7 +2856,8 @@ static test_return_t auto_eject_hosts(memcached_st *trash)
   /* and re-added when it's back. */
   memc->hosts[2].next_retry = time(NULL) - 1;
   memc->next_distribution_rebuild= time(NULL) - 1;
-  run_distribution(memc);
+  memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION,
+                         memc->distribution);
   for (int x= 0; x < 99; x++)
   {
     uint32_t server_idx = memcached_generate_hash(memc, ketama_test_cases[x].key, strlen(ketama_test_cases[x].key));
@@ -4351,7 +4352,7 @@ static test_return_t set_udp_behavior_test(memcached_st *memc)
 
   memcached_quit(memc);
   memc->number_of_hosts= 0;
-  run_distribution(memc);
+  memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_DISTRIBUTION, memc->distribution);
   test_truth(memcached_behavior_set(memc, MEMCACHED_BEHAVIOR_USE_UDP, 1) == MEMCACHED_SUCCESS);
   test_truth(memc->flags & MEM_USE_UDP);
   test_truth(memc->flags & MEM_NOREPLY);;