Related refactoring to updated of ThrottleList to use new Scheduler.

test/helpers/test_thread.cc

@@ -0,0 +1,102 @@
+#include "config.h"
+
+#include "test_thread.h"
+
+#include <unistd.h>
+#include <cppunit/extensions/HelperMacros.h>
+#include <torrent/exceptions.h>
+#include <torrent/poll.h>
+
+#include "test/helpers/mock_function.h"
+
+const int test_thread::test_flag_pre_stop;
+const int test_thread::test_flag_long_timeout;
+
+const int test_thread::test_flag_do_work;
+const int test_thread::test_flag_pre_poke;
+const int test_thread::test_flag_post_poke;
+
+// TODO: Remove PollSelect.
+
+torrent::Poll*
+create_poll() {
+  torrent::Poll* poll = torrent::Poll::create(256);
+
+  if (poll == nullptr)
+    throw torrent::internal_error("Unable to create poll object");
+
+  return poll;
+}
+
+void
+set_create_poll() {
+  torrent::Poll::slot_create_poll() = []() {
+    return create_poll();
+  };
+}
+
+std::unique_ptr<test_thread>
+test_thread::create() {
+  // Needs to be called before Thread is created.
+  mock_redirect_defaults();
+
+  auto thread = new test_thread();
+  return std::unique_ptr<test_thread>(thread);
+}
+
+test_thread::test_thread() :
+  m_test_state(TEST_NONE),
+  m_test_flags(0) {
+}
+
+test_thread::~test_thread() {
+  if (is_active())
+    stop_thread_wait();
+
+  m_self = nullptr;
+}
+
+void
+test_thread::init_thread() {
+  m_state = STATE_INITIALIZED;
+  m_test_state = TEST_PRE_START;
+
+  m_poll = std::unique_ptr<torrent::Poll>(create_poll());
+}
+
+void
+test_thread::call_events() {
+  m_loop_count++;
+
+  if ((m_test_flags & test_flag_pre_stop) && m_test_state == TEST_PRE_START && m_state == STATE_ACTIVE)
+    m_test_state = TEST_PRE_STOP;
+
+  if ((m_flags & flag_do_shutdown)) {
+    if ((m_flags & flag_did_shutdown))
+      throw torrent::internal_error("Already trigged shutdown.");
+
+    m_flags |= flag_did_shutdown;
+    throw torrent::shutdown_exception();
+  }
+
+  if ((m_test_flags & test_flag_pre_poke)) {
+  }
+
+  if ((m_test_flags & test_flag_do_work)) {
+    usleep(10 * 1000); // TODO: Don't just sleep, as that give up core.
+    m_test_flags &= ~test_flag_do_work;
+  }
+
+  if ((m_test_flags & test_flag_post_poke)) {
+  }
+
+  process_callbacks();
+}
+
+std::chrono::microseconds
+test_thread::next_timeout() {
+  if ((m_test_flags & test_flag_long_timeout))
+    return std::chrono::microseconds(10s);
+  else
+    return std::chrono::microseconds(100ms);
+}