aria2/test/MetalinkEntryTest.cc

#include "MetalinkEntry.h"
#include <cppunit/extensions/HelperMacros.h>

using namespace std;

class MetalinkEntryTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(MetalinkEntryTest);
  CPPUNIT_TEST(testDropUnsupportedResource);
  CPPUNIT_TEST(testReorderResourcesByPreference);
  CPPUNIT_TEST(testSetLocationPreference);
  CPPUNIT_TEST(testSetProtocolPreference);
  CPPUNIT_TEST_SUITE_END();
private:

public:
  void setUp() {
  }
  void tearDown() {
  }

  void testDropUnsupportedResource();
  void testReorderResourcesByPreference();
  void testSetLocationPreference();
  void testSetProtocolPreference();
};


CPPUNIT_TEST_SUITE_REGISTRATION( MetalinkEntryTest );

MetalinkEntryHandle createTestEntry() {
  MetalinkEntryHandle entry = new MetalinkEntry();
  MetalinkResource* res1 = new MetalinkResource();
  res1->url = "ftp://myhost/aria2.tar.bz2";
  res1->type = MetalinkResource::TYPE_FTP;
  res1->location = "RO";
  res1->preference = 50;
  MetalinkResource* res2 = new MetalinkResource();
  res2->url = "http://myhost/aria2.tar.bz2";
  res2->type = MetalinkResource::TYPE_HTTP;
  res2->location = "AT";
  res2->preference = 100;
  MetalinkResource* res3 = new MetalinkResource();
  res3->url = "http://myhost/aria2.torrent";
  res3->type = MetalinkResource::TYPE_BITTORRENT;
  res3->location = "AL";
  res3->preference = 60;
  MetalinkResource* res4 = new MetalinkResource();
  res4->url = "http://myhost/aria2.ext";
  res4->type = MetalinkResource::TYPE_NOT_SUPPORTED;
  res4->location = "AD";
  res4->preference = 10;
  MetalinkResource* res5 = new MetalinkResource();
  res5->url = "https://myhost/aria2.tar.bz2";
  res5->type = MetalinkResource::TYPE_HTTPS;
  res5->location = "JP";
  res5->preference = 90;

  entry->resources.push_back(res1);
  entry->resources.push_back(res2);
  entry->resources.push_back(res3);
  entry->resources.push_back(res4);
  entry->resources.push_back(res5);
  return entry;
}

void MetalinkEntryTest::testDropUnsupportedResource() {
  MetalinkEntryHandle entry = createTestEntry();

  entry->dropUnsupportedResource();
#if defined ENABLE_SSL && ENABLE_BITTORRENT
  CPPUNIT_ASSERT_EQUAL(4, (int)entry->resources.size());
#elif defined ENABLE_SSL || ENABLE_BITTORRENT
  CPPUNIT_ASSERT_EQUAL(3, (int)entry->resources.size());
#else
  CPPUNIT_ASSERT_EQUAL(2, (int)entry->resources.size());
#endif // ENABLE_MESSAGE_DIGEST
  
  MetalinkResources::const_iterator itr = entry->resources.begin();
  CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_FTP,
		       (*itr++)->type);
  CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_HTTP,
		       (*itr++)->type);
#ifdef ENABLE_BITTORRENT
  CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_BITTORRENT,
		       (*itr++)->type);
#endif // ENABLE_BITTORRENT
#ifdef ENABLE_SSL
  CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_HTTPS,
		       (*itr++)->type);
#endif // ENABLE_SSL
}

void MetalinkEntryTest::testReorderResourcesByPreference() {
  MetalinkEntryHandle entry = createTestEntry();
  
  entry->reorderResourcesByPreference();

  CPPUNIT_ASSERT_EQUAL((int32_t)100, entry->resources.at(0)->preference);
  CPPUNIT_ASSERT_EQUAL((int32_t)90, entry->resources.at(1)->preference);
  CPPUNIT_ASSERT_EQUAL((int32_t)60, entry->resources.at(2)->preference);
  CPPUNIT_ASSERT_EQUAL((int32_t)50, entry->resources.at(3)->preference);
  CPPUNIT_ASSERT_EQUAL((int32_t)10, entry->resources.at(4)->preference);
}

void MetalinkEntryTest::testSetLocationPreference()
{
  MetalinkEntryHandle entry = createTestEntry();

  const char* locationsSrc[] = { "jp", "al", "RO" };

  Strings locations(&locationsSrc[0], &locationsSrc[3]);

  entry->setLocationPreference(locations, 100);

  CPPUNIT_ASSERT_EQUAL(string("RO"), entry->resources[0]->location);
  CPPUNIT_ASSERT_EQUAL((int32_t)150, entry->resources[0]->preference);
  CPPUNIT_ASSERT_EQUAL(string("AT"), entry->resources[1]->location);
  CPPUNIT_ASSERT_EQUAL((int32_t)100, entry->resources[1]->preference);
  CPPUNIT_ASSERT_EQUAL(string("AL"), entry->resources[2]->location);
  CPPUNIT_ASSERT_EQUAL((int32_t)160, entry->resources[2]->preference);
  CPPUNIT_ASSERT_EQUAL(string("AD"), entry->resources[3]->location);
  CPPUNIT_ASSERT_EQUAL((int32_t)10, entry->resources[3]->preference);
  CPPUNIT_ASSERT_EQUAL(string("JP"), entry->resources[4]->location);
  CPPUNIT_ASSERT_EQUAL((int32_t)190, entry->resources[4]->preference);
}

void MetalinkEntryTest::testSetProtocolPreference()
{
  MetalinkEntryHandle entry = createTestEntry();
  entry->setProtocolPreference("http", 1);
  CPPUNIT_ASSERT_EQUAL(50, entry->resources[0]->preference); // ftp
  CPPUNIT_ASSERT_EQUAL(101, entry->resources[1]->preference); // http, +1
  CPPUNIT_ASSERT_EQUAL(60, entry->resources[2]->preference); // bittorrent
  CPPUNIT_ASSERT_EQUAL(10, entry->resources[3]->preference); // not supported
  CPPUNIT_ASSERT_EQUAL(90, entry->resources[4]->preference); // https
}