Use count argument in run_policy

This argument was not used anywhere, it worked by chance
since it had the same value everywhere.

Author: martin-sucha

tests/src/unit/tests/test_load_balancing.cpp

@@ -93,7 +93,7 @@ typedef Map<Address, int> QueryCounts;
 
 QueryCounts run_policy(LoadBalancingPolicy& policy, int count) {
   QueryCounts counts;
-  for (int i = 0; i < 12; ++i) {
+  for (int i = 0; i < count; ++i) {
     ScopedPtr<QueryPlan> qp(policy.new_query_plan("ks", NULL, NULL));
     Host::Ptr host(qp->compute_next());
     if (host) {