Arne Moerman
96 lines
3.5 KiB
C#
96 lines
3.5 KiB
C#
using Microsoft.Extensions.Time.Testing;
|
|
using SeerrAutoApprove.Options;
|
|
using SeerrAutoApprove.Quota;
|
|
|
|
namespace SeerrAutoApprove.Tests;
|
|
|
|
public sealed class QuotaRulesTests
|
|
{
|
|
// Thursday
|
|
private static readonly DateTimeOffset Now = new(2026, 10, 15, 12, 30, 0, TimeSpan.Zero);
|
|
|
|
private static FakeTimeProvider Time(TimeZoneInfo? zone = null)
|
|
{
|
|
var time = new FakeTimeProvider(Now);
|
|
time.SetLocalTimeZone(zone ?? TimeZoneInfo.Utc);
|
|
return time;
|
|
}
|
|
|
|
[Fact]
|
|
public void Rolling_period_starts_n_days_ago()
|
|
{
|
|
var start = QuotaRules.GetPeriodStart(new AutoApproveOptions { PeriodType = PeriodType.Rolling, RollingDays = 7 }, Time());
|
|
|
|
Assert.Equal(Now.AddDays(-7), start);
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData(CalendarPeriod.Day, 2026, 10, 15)]
|
|
[InlineData(CalendarPeriod.Week, 2026, 10, 12)]
|
|
[InlineData(CalendarPeriod.Month, 2026, 10, 1)]
|
|
[InlineData(CalendarPeriod.Year, 2026, 1, 1)]
|
|
public void Calendar_period_starts_at_beginning_of_period(CalendarPeriod period, int year, int month, int day)
|
|
{
|
|
var options = new AutoApproveOptions { PeriodType = PeriodType.Calendar, CalendarPeriod = period };
|
|
|
|
var start = QuotaRules.GetPeriodStart(options, Time());
|
|
|
|
Assert.Equal(new DateTimeOffset(year, month, day, 0, 0, 0, TimeSpan.Zero), start);
|
|
}
|
|
|
|
[Fact]
|
|
public void Calendar_period_respects_local_time_zone()
|
|
{
|
|
var zone = TimeZoneInfo.CreateCustomTimeZone("UTC+2", TimeSpan.FromHours(2), "UTC+2", "UTC+2");
|
|
var options = new AutoApproveOptions { PeriodType = PeriodType.Calendar, CalendarPeriod = CalendarPeriod.Month };
|
|
|
|
var start = QuotaRules.GetPeriodStart(options, Time(zone));
|
|
|
|
Assert.Equal(new DateTimeOffset(2026, 10, 1, 0, 0, 0, TimeSpan.FromHours(2)), start);
|
|
}
|
|
|
|
[Fact]
|
|
public void Week_starts_on_monday_when_today_is_sunday()
|
|
{
|
|
var time = new FakeTimeProvider(new DateTimeOffset(2026, 10, 18, 10, 0, 0, TimeSpan.Zero));
|
|
time.SetLocalTimeZone(TimeZoneInfo.Utc);
|
|
var options = new AutoApproveOptions { PeriodType = PeriodType.Calendar, CalendarPeriod = CalendarPeriod.Week };
|
|
|
|
Assert.Equal(new DateTimeOffset(2026, 10, 12, 0, 0, 0, TimeSpan.Zero), QuotaRules.GetPeriodStart(options, time));
|
|
}
|
|
|
|
[Fact]
|
|
public void Categorizes_requests()
|
|
{
|
|
Assert.Equal(QuotaCategory.Movie, QuotaRules.GetCategory(Requests.Movie(1)));
|
|
Assert.Equal(QuotaCategory.Movie4k, QuotaRules.GetCategory(Requests.Movie(1, is4k: true)));
|
|
Assert.Equal(QuotaCategory.Series, QuotaRules.GetCategory(Requests.Series(1, 1)));
|
|
Assert.Equal(QuotaCategory.Series4k, QuotaRules.GetCategory(Requests.Series(1, 1, is4k: true)));
|
|
}
|
|
|
|
[Fact]
|
|
public void Falls_back_to_media_type_when_request_type_missing()
|
|
{
|
|
var request = Requests.Movie(1) with { Type = null, Media = new Seerr.SeerrMedia { MediaType = "tv" } };
|
|
|
|
Assert.Equal(QuotaCategory.Series, QuotaRules.GetCategory(request));
|
|
}
|
|
|
|
[Theory]
|
|
[InlineData(SeriesCountMode.PerSeason, 3, 3)]
|
|
[InlineData(SeriesCountMode.PerSeason, 0, 1)]
|
|
[InlineData(SeriesCountMode.PerRequest, 3, 1)]
|
|
public void Series_units_follow_count_mode(SeriesCountMode mode, int seasons, int expected)
|
|
{
|
|
var units = QuotaRules.GetUnits(Requests.Series(1, seasons), new AutoApproveOptions { SeriesCountMode = mode });
|
|
|
|
Assert.Equal(expected, units);
|
|
}
|
|
|
|
[Fact]
|
|
public void Movies_always_count_as_one()
|
|
{
|
|
Assert.Equal(1, QuotaRules.GetUnits(Requests.Movie(1), new AutoApproveOptions()));
|
|
}
|
|
}
|