blob: bca5c81ef016204893515c9420c5299496935620 (
plain) (
blame)
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
|
// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/storageframework/generic/thread/taskthread.h>
#include <vespa/vdstestlib/cppunit/macros.h>
namespace storage {
namespace framework {
struct TaskThreadTest : public CppUnit::TestFixture
{
void testNormalUsage();
CPPUNIT_TEST_SUITE(TaskThreadTest);
CPPUNIT_TEST(testNormalUsage);
CPPUNIT_TEST_SUITE_END();
};
CPPUNIT_TEST_SUITE_REGISTRATION(TaskThreadTest);
namespace {
struct Task {
std::string _name;
uint8_t _priority;
Task(const std::string& name, uint8_t priority)
: _name(name), _priority(priority) {}
bool operator<(const Task& other) const {
return (_priority > other._priority);
}
uint8_t getPriority() const { return _priority; }
};
struct MyThread : public TaskThread<Task> {
MyThread(ThreadLock& lock) : TaskThread<Task>(lock) {}
virtual ThreadWaitInfo doNonCriticalTick(ThreadIndex) override {
return ThreadWaitInfo::NO_MORE_CRITICAL_WORK_KNOWN;
}
};
}
void
TaskThreadTest::testNormalUsage()
{
TickingThreadPool::UP pool(TickingThreadPool::createDefault("testApp"));
MyThread t(*pool);
t.addTask(Task("a", 6));
t.addTask(Task("b", 3));
t.addTask(Task("c", 8));
t.addTask(Task("d", 4));
CPPUNIT_ASSERT(t.empty()); // Still empty before critical tick has run
dynamic_cast<TickingThread&>(t).doCriticalTick(0);
CPPUNIT_ASSERT(!t.empty());
CPPUNIT_ASSERT_EQUAL(3, (int) t.peek().getPriority());
std::ostringstream ost;
while (!t.empty()) {
Task task(t.peek());
ost << task._name << '(' << ((int) task.getPriority()) << ") ";
t.pop();
}
CPPUNIT_ASSERT_EQUAL(std::string("b(3) d(4) a(6) c(8) "), ost.str());
}
} // framework
} // storage
|