152 lines
4.8 KiB
C++
152 lines
4.8 KiB
C++
// This file Copyright (C) 2022 Mnemosyne LLC.
|
|
// It may be used under GPLv2 (SPDX: GPL-2.0-only), GPLv3 (SPDX: GPL-3.0-only),
|
|
// or any future license endorsed by Mnemosyne LLC.
|
|
// License text can be found in the licenses/ folder.
|
|
|
|
#include <chrono>
|
|
#include <ctime>
|
|
#include <vector>
|
|
|
|
#include <libtransmission/transmission.h>
|
|
#include <libtransmission/session-alt-speeds.h>
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
using namespace std::literals;
|
|
|
|
class SessionAltSpeedsTest : public ::testing::Test
|
|
{
|
|
protected:
|
|
using ChangeReason = tr_session_alt_speeds::ChangeReason;
|
|
|
|
class MockMediator final : public tr_session_alt_speeds::Mediator
|
|
{
|
|
public:
|
|
void is_active_changed(bool is_active, ChangeReason reason) override
|
|
{
|
|
changelog_.emplace_back(is_active, reason, time());
|
|
}
|
|
|
|
[[nodiscard]] time_t time() override
|
|
{
|
|
return current_time_;
|
|
}
|
|
|
|
time_t current_time_ = 0;
|
|
|
|
struct Change
|
|
{
|
|
Change() = default;
|
|
|
|
Change(bool is_active_in, ChangeReason reason_in, time_t timestamp_in)
|
|
: is_active{ is_active_in }
|
|
, reason{ reason_in }
|
|
, timestamp{ timestamp_in }
|
|
{
|
|
}
|
|
|
|
bool is_active = false;
|
|
ChangeReason reason = ChangeReason::User;
|
|
time_t timestamp = 0;
|
|
};
|
|
|
|
std::vector<Change> changelog_;
|
|
};
|
|
|
|
static auto constexpr ArbitraryTimestamp1 = time_t{ 666 };
|
|
|
|
static auto midnightSundayMorning()
|
|
{
|
|
// new year's, Sun Jan 1 2023
|
|
auto tmdate = tm{};
|
|
tmdate.tm_sec = 0;
|
|
tmdate.tm_min = 0;
|
|
tmdate.tm_hour = 0;
|
|
tmdate.tm_mday = 1;
|
|
tmdate.tm_mon = 0;
|
|
tmdate.tm_year = 2023 - 1900;
|
|
tmdate.tm_wday = 0; // Sunday
|
|
tmdate.tm_yday = 0;
|
|
tmdate.tm_isdst = 0;
|
|
return mktime(&tmdate);
|
|
}
|
|
};
|
|
|
|
namespace libtransmission::test
|
|
{
|
|
|
|
TEST_F(SessionAltSpeedsTest, canInstantiate)
|
|
{
|
|
auto mediator = MockMediator{};
|
|
auto alt_speeds = tr_session_alt_speeds{ mediator };
|
|
EXPECT_FALSE(alt_speeds.is_active());
|
|
}
|
|
|
|
TEST_F(SessionAltSpeedsTest, canActivate)
|
|
{
|
|
static auto constexpr Now = ArbitraryTimestamp1;
|
|
auto mediator = MockMediator{};
|
|
mediator.current_time_ = Now;
|
|
|
|
auto alt_speeds = tr_session_alt_speeds{ mediator };
|
|
auto const changed_value = !alt_speeds.is_active();
|
|
EXPECT_EQ(0U, std::size(mediator.changelog_));
|
|
|
|
static auto constexpr Reason = ChangeReason::User;
|
|
alt_speeds.set_active(changed_value, Reason);
|
|
EXPECT_EQ(changed_value, alt_speeds.is_active());
|
|
ASSERT_EQ(1U, std::size(mediator.changelog_));
|
|
EXPECT_EQ(changed_value, mediator.changelog_[0].is_active);
|
|
EXPECT_EQ(Reason, mediator.changelog_[0].reason);
|
|
EXPECT_EQ(Now, mediator.changelog_[0].timestamp);
|
|
}
|
|
|
|
TEST_F(SessionAltSpeedsTest, canSchedule)
|
|
{
|
|
auto mediator = MockMediator{};
|
|
|
|
auto now = midnightSundayMorning(); // midnight
|
|
mediator.current_time_ = now;
|
|
|
|
auto alt_speeds = tr_session_alt_speeds{ mediator };
|
|
alt_speeds.set_start_minute(60); // start at 1AM
|
|
alt_speeds.set_end_minute(120); // end at 2AM
|
|
alt_speeds.set_weekdays(TR_SCHED_ALL); // every day
|
|
alt_speeds.set_scheduler_enabled(true);
|
|
auto n_changes = std::size(mediator.changelog_);
|
|
EXPECT_EQ(0U, n_changes);
|
|
|
|
// Confirm that walking up to the threshold, but not crossing it, does not enable
|
|
now += std::chrono::duration_cast<std::chrono::seconds>(59min).count();
|
|
mediator.current_time_ = now;
|
|
alt_speeds.check_scheduler();
|
|
EXPECT_EQ(n_changes, std::size(mediator.changelog_));
|
|
|
|
// Confirm that crossing the threshold does enable
|
|
now += std::chrono::duration_cast<std::chrono::seconds>(1min).count();
|
|
mediator.current_time_ = now;
|
|
alt_speeds.check_scheduler();
|
|
ASSERT_EQ(n_changes + 1, std::size(mediator.changelog_));
|
|
EXPECT_EQ(true, mediator.changelog_[n_changes].is_active);
|
|
EXPECT_EQ(ChangeReason::Scheduler, mediator.changelog_[n_changes].reason);
|
|
EXPECT_EQ(now, mediator.changelog_[n_changes].timestamp);
|
|
++n_changes;
|
|
|
|
// Confirm that walking up to the threshold, but not crossing it, does not disable
|
|
now += std::chrono::duration_cast<std::chrono::seconds>(59min).count();
|
|
mediator.current_time_ = now;
|
|
alt_speeds.check_scheduler();
|
|
EXPECT_EQ(n_changes, std::size(mediator.changelog_));
|
|
|
|
// Confirm that crossing the threshold does disable
|
|
now += std::chrono::duration_cast<std::chrono::seconds>(1min).count();
|
|
mediator.current_time_ = now;
|
|
alt_speeds.check_scheduler();
|
|
ASSERT_EQ(n_changes + 1, std::size(mediator.changelog_));
|
|
EXPECT_EQ(false, mediator.changelog_[n_changes].is_active);
|
|
EXPECT_EQ(ChangeReason::Scheduler, mediator.changelog_[n_changes].reason);
|
|
EXPECT_EQ(now, mediator.changelog_[n_changes].timestamp);
|
|
}
|
|
|
|
} // namespace libtransmission::test
|