summaryrefslogtreecommitdiff
path: root/test/UTMetadataRequestTrackerTest.cc
blob: 1170d6dc30d9a4830a5ed08cf8a422a28a4b3181 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
#include "UTMetadataRequestTracker.h"

#include <cppunit/extensions/HelperMacros.h>

namespace aria2 {

class UTMetadataRequestTrackerTest:public CppUnit::TestFixture {

  CPPUNIT_TEST_SUITE(UTMetadataRequestTrackerTest);
  CPPUNIT_TEST(testAdd);
  CPPUNIT_TEST(testRemove);
  CPPUNIT_TEST(testGetAllTrackedIndex);
  CPPUNIT_TEST(testCount);
  CPPUNIT_TEST(testAvail);
  CPPUNIT_TEST_SUITE_END();
public:
  void testAdd();
  void testRemove();
  void testGetAllTrackedIndex();
  void testCount();
  void testAvail();
};


CPPUNIT_TEST_SUITE_REGISTRATION(UTMetadataRequestTrackerTest);

void UTMetadataRequestTrackerTest::testAdd()
{
  UTMetadataRequestTracker tr;
  tr.add(1);
  CPPUNIT_ASSERT(tr.tracks(1));
}

void UTMetadataRequestTrackerTest::testRemove()
{
  UTMetadataRequestTracker tr;
  tr.add(1);
  tr.remove(1);
  CPPUNIT_ASSERT(!tr.tracks(1));
}

void UTMetadataRequestTrackerTest::testGetAllTrackedIndex()
{
  UTMetadataRequestTracker tr;
  tr.add(1);
  tr.add(2);

  std::vector<size_t> indexes = tr.getAllTrackedIndex();
  CPPUNIT_ASSERT_EQUAL((size_t)2, indexes.size());
  CPPUNIT_ASSERT_EQUAL((size_t)1, indexes[0]);
  CPPUNIT_ASSERT_EQUAL((size_t)2, indexes[1]);
}

void UTMetadataRequestTrackerTest::testCount()
{
  UTMetadataRequestTracker tr;
  tr.add(1);
  tr.add(2);
  CPPUNIT_ASSERT_EQUAL((size_t)2, tr.count());
}

void UTMetadataRequestTrackerTest::testAvail()
{
  UTMetadataRequestTracker tr;
  CPPUNIT_ASSERT_EQUAL((size_t)1, tr.avail());
  tr.add(1);
  CPPUNIT_ASSERT_EQUAL((size_t)0, tr.avail());
  tr.add(2);
  CPPUNIT_ASSERT_EQUAL((size_t)0, tr.avail());
}

} // namespace aria2