#include "config.h" #include "test/src/test_command_throttle.h" #include <torrent/throttle.h>#include <torrent/torrent.h> #include "core/manager.h"#include "control.h"#include "globals.h"#include "rpc/parse_commands.h" CPPUNIT_TEST_SUITE_REGISTRATION(TestCommandThrottle); void initialize_command_throttle(); static voidcall_set(const char* key, const char* value) {  rpc::commands.call_command(key, torrent::Object(std::string(value)));} static voidcall_named(const char* key, const char* name, const char* value) {  torrent::Object::list_type args;  args.push_back(torrent::Object(std::string(name)));  args.push_back(torrent::Object(std::string(value)));   rpc::commands.call_command(key, torrent::Object::create_list_range(args.begin(), args.end()));} static uint64_tdown_rate() {  return torrent::down_throttle_global()->max_rate();} voidTestCommandThrottle::setUp() {  torrent::initialize_main_thread();  torrent::initialize();   if (control == nullptr)    control = new Control;   if (!rpc::commands.has("throttle.global_down.max_rate.set_kb"))    initialize_command_throttle();} voidTestCommandThrottle::tearDown() {  torrent::cleanup();} voidTestCommandThrottle::test_global_rate_in_range() {  call_set("throttle.global_down.max_rate.set_kb", "1024");  CPPUNIT_ASSERT_EQUAL(uint64_t{1048576}, down_rate());   call_set("throttle.global_down.max_rate.set_kb", "4194303");  CPPUNIT_ASSERT_EQUAL(uint64_t{4294966272}, down_rate());   call_set("throttle.global_down.max_rate.set", "4294966272");  CPPUNIT_ASSERT_EQUAL(uint64_t{4294966272}, down_rate());} voidTestCommandThrottle::test_global_rate_kb_out_of_range() {  call_set("throttle.global_down.max_rate.set_kb", "1024");   CPPUNIT_ASSERT_THROW(call_set("throttle.global_down.max_rate.set_kb", "4194304"), torrent::input_error);  CPPUNIT_ASSERT_EQUAL(uint64_t{1048576}, down_rate());} voidTestCommandThrottle::test_global_rate_bytes_out_of_range() {  call_set("throttle.global_down.max_rate.set", "1048576");   CPPUNIT_ASSERT_THROW(call_set("throttle.global_down.max_rate.set", "4294967296"), torrent::input_error);  CPPUNIT_ASSERT_EQUAL(uint64_t{1048576}, down_rate());} voidTestCommandThrottle::test_global_rate_negative() {  call_set("throttle.global_down.max_rate.set", "1048576");   CPPUNIT_ASSERT_THROW(call_set("throttle.global_down.max_rate.set", "-1"), torrent::input_error);  CPPUNIT_ASSERT_EQUAL(uint64_t{1048576}, down_rate());} voidTestCommandThrottle::test_named_rate_in_range() {  call_named("throttle.down", "test_named_in_range", "1024");   auto itr = control->core()->throttles().find("test_named_in_range");   CPPUNIT_ASSERT(itr != control->core()->throttles().end());  CPPUNIT_ASSERT_EQUAL(uint64_t{1048576}, itr->second.second->max_rate());} voidTestCommandThrottle::test_named_rate_out_of_range() {  CPPUNIT_ASSERT_THROW(call_named("throttle.down", "test_named_out_of_range", "18014398509481984"), torrent::input_error);}