Xml2MetalinkProcessorTest.cc 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138
  1. #include "Xml2MetalinkProcessor.h"
  2. #include "Exception.h"
  3. #include "DefaultDiskWriter.h"
  4. #include <cppunit/extensions/HelperMacros.h>
  5. using namespace std;
  6. class Xml2MetalinkProcessorTest:public CppUnit::TestFixture {
  7. CPPUNIT_TEST_SUITE(Xml2MetalinkProcessorTest);
  8. CPPUNIT_TEST(testParseFile);
  9. CPPUNIT_TEST(testParseFromBinaryStream);
  10. CPPUNIT_TEST_SUITE_END();
  11. private:
  12. public:
  13. void setUp() {
  14. xmlInitParser();
  15. }
  16. void tearDown() {
  17. xmlCleanupParser();
  18. }
  19. void testParseFile();
  20. void testParseFromBinaryStream();
  21. };
  22. CPPUNIT_TEST_SUITE_REGISTRATION( Xml2MetalinkProcessorTest );
  23. void Xml2MetalinkProcessorTest::testParseFile() {
  24. Xml2MetalinkProcessor proc;
  25. try {
  26. MetalinkerHandle metalinker = proc.parseFile("test.xml");
  27. MetalinkEntries::iterator entryItr = metalinker->entries.begin();
  28. MetalinkEntryHandle entry1 = *entryItr;
  29. CPPUNIT_ASSERT_EQUAL(string("aria2-0.5.2.tar.bz2"), entry1->getPath());
  30. CPPUNIT_ASSERT_EQUAL((int64_t)0, entry1->getLength());
  31. CPPUNIT_ASSERT_EQUAL(string("0.5.2"), entry1->version);
  32. CPPUNIT_ASSERT_EQUAL(string("en-US"), entry1->language);
  33. CPPUNIT_ASSERT_EQUAL(string("Linux-x86"), entry1->os);
  34. CPPUNIT_ASSERT_EQUAL((int32_t)1, entry1->maxConnections);
  35. #ifdef ENABLE_MESSAGE_DIGEST
  36. CPPUNIT_ASSERT_EQUAL(string("a96cf3f0266b91d87d5124cf94326422800b627d"),
  37. entry1->checksum->getMessageDigest());
  38. CPPUNIT_ASSERT_EQUAL(string("sha1"), entry1->checksum->getAlgo());
  39. #endif // ENABLE_MESSAGE_DIGEST
  40. MetalinkResources::iterator resourceItr1 = entry1->resources.begin();
  41. MetalinkResourceHandle resource1 = *resourceItr1;
  42. CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_FTP, resource1->type);
  43. CPPUNIT_ASSERT_EQUAL(string("JP"), resource1->location);
  44. CPPUNIT_ASSERT_EQUAL((int32_t)100, resource1->preference);
  45. CPPUNIT_ASSERT_EQUAL(string("ftp://ftphost/aria2-0.5.2.tar.bz2"),
  46. resource1->url);
  47. CPPUNIT_ASSERT_EQUAL((int32_t)1, resource1->maxConnections);
  48. resourceItr1++;
  49. MetalinkResourceHandle resource2 = *resourceItr1;
  50. CPPUNIT_ASSERT_EQUAL(MetalinkResource::TYPE_HTTP, resource2->type);
  51. CPPUNIT_ASSERT_EQUAL(string("US"), resource2->location);
  52. CPPUNIT_ASSERT_EQUAL((int32_t)100, resource2->preference);
  53. CPPUNIT_ASSERT_EQUAL(string("http://httphost/aria2-0.5.2.tar.bz2"),
  54. resource2->url);
  55. CPPUNIT_ASSERT_EQUAL((int32_t)-1, resource2->maxConnections);
  56. entryItr++;
  57. MetalinkEntryHandle entry2 = *entryItr;
  58. CPPUNIT_ASSERT_EQUAL(string("aria2-0.5.1.tar.bz2"), entry2->getPath());
  59. CPPUNIT_ASSERT_EQUAL((int64_t)345689, entry2->getLength());
  60. CPPUNIT_ASSERT_EQUAL(string("0.5.1"), entry2->version);
  61. CPPUNIT_ASSERT_EQUAL(string("ja-JP"), entry2->language);
  62. CPPUNIT_ASSERT_EQUAL(string("Linux-m68k"), entry2->os);
  63. CPPUNIT_ASSERT_EQUAL((int32_t)-1, entry2->maxConnections);
  64. #ifdef ENABLE_MESSAGE_DIGEST
  65. CPPUNIT_ASSERT_EQUAL(string("4c255b0ed130f5ea880f0aa061c3da0487e251cc"),
  66. entry2->checksum->getMessageDigest());
  67. CPPUNIT_ASSERT_EQUAL((int32_t)2, entry2->chunkChecksum->countChecksum());
  68. CPPUNIT_ASSERT_EQUAL((int32_t)262144, entry2->chunkChecksum->getChecksumLength());
  69. CPPUNIT_ASSERT_EQUAL(string("179463a88d79cbf0b1923991708aead914f26142"),
  70. entry2->chunkChecksum->getChecksum(0));
  71. CPPUNIT_ASSERT_EQUAL(string("fecf8bc9a1647505fe16746f94e97a477597dbf3"),
  72. entry2->chunkChecksum->getChecksum(1));
  73. CPPUNIT_ASSERT_EQUAL(string("sha1"), entry2->checksum->getAlgo());
  74. #endif // ENABLE_MESSAGE_DIGEST
  75. entryItr++;
  76. // test case: verification hash is not provided
  77. MetalinkEntryHandle entry3 = *entryItr;
  78. CPPUNIT_ASSERT_EQUAL(string("NoVerificationHash"), entry3->getPath());
  79. #ifdef ENABLE_MESSAGE_DIGEST
  80. CPPUNIT_ASSERT(entry3->checksum.isNull());
  81. CPPUNIT_ASSERT(entry3->chunkChecksum.isNull());
  82. #endif // ENABLE_MESSAGE_DIGEST
  83. entryItr++;
  84. // test case: unsupported verification hash is included
  85. MetalinkEntryHandle entry4 = *entryItr;
  86. CPPUNIT_ASSERT_EQUAL(string("UnsupportedVerificationHashTypeIncluded"), entry4->getPath());
  87. #ifdef ENABLE_MESSAGE_DIGEST
  88. CPPUNIT_ASSERT_EQUAL(string("sha1"),
  89. entry4->checksum->getAlgo());
  90. CPPUNIT_ASSERT_EQUAL(string("4c255b0ed130f5ea880f0aa061c3da0487e251cc"),
  91. entry4->checksum->getMessageDigest());
  92. CPPUNIT_ASSERT_EQUAL(string("sha1"),
  93. entry4->chunkChecksum->getAlgo());
  94. #endif // ENABLE_MESSAGE_DIGEST
  95. } catch(Exception* e) {
  96. CPPUNIT_FAIL(e->getMsg());
  97. delete e;
  98. }
  99. }
  100. void Xml2MetalinkProcessorTest::testParseFromBinaryStream() {
  101. Xml2MetalinkProcessor proc;
  102. try {
  103. DefaultDiskWriterHandle dw = new DefaultDiskWriter();
  104. dw->openExistingFile("test.xml");
  105. MetalinkerHandle metalinker = proc.parseFromBinaryStream(dw);
  106. dw->closeFile();
  107. MetalinkEntries::iterator entryItr = metalinker->entries.begin();
  108. MetalinkEntryHandle entry1 = *entryItr;
  109. CPPUNIT_ASSERT_EQUAL(string("aria2-0.5.2.tar.bz2"), entry1->getPath());
  110. } catch(Exception* e) {
  111. cerr << *e;
  112. CPPUNIT_FAIL(e->getMsg());
  113. delete e;
  114. throw;
  115. }
  116. }