From 7b480efe68bc09533b795d33801870d60ccebe93 Mon Sep 17 00:00:00 2001
From: Arvid Norberg <arvid@libtorrent.org>
Date: Thu, 25 Feb 2010 19:56:30 +0000
Subject: [PATCH] fix piece picker test

---
 test/test_piece_picker.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/test_piece_picker.cpp b/test/test_piece_picker.cpp
index b8dfccc5a..f967af5b7 100644
--- a/test/test_piece_picker.cpp
+++ b/test/test_piece_picker.cpp
@@ -70,7 +70,7 @@ boost::shared_ptr<piece_picker> setup_picker(
 	assert(int(strlen(have_str)) == num_pieces);
 
 	boost::shared_ptr<piece_picker> p(new piece_picker);
-	p->init(blocks_per_piece, num_pieces * blocks_per_piece);
+	p->init(blocks_per_piece, blocks_per_piece, num_pieces);
 
 	for (int i = 0; i < num_pieces; ++i)
 	{
@@ -430,7 +430,7 @@ int test_main()
 	TEST_CHECK(p->num_have_filtered() == 1);
 	TEST_CHECK(p->num_have() == 1);
 
-	p->init(blocks_per_piece, blocks_per_piece * 7);
+	p->init(blocks_per_piece, blocks_per_piece, blocks_per_piece * 7);
 	TEST_CHECK(p->piece_priority(0) == 0);
 	TEST_CHECK(p->num_filtered() == 1);
 	TEST_CHECK(p->num_have_filtered() == 0);