Refactor MaxBitrate and UploadLimit in GuildHelper (#2854)

This commit is contained in:
FluffyDoggo
2024-02-19 02:59:14 -06:00
committed by GitHub
parent 1eb7a53f78
commit 6256de42bc
5 changed files with 23 additions and 36 deletions

View File

@@ -32,7 +32,7 @@ public class DiscordRestApiClientTests : IClassFixture<RestGuildFixture>, IAsync
[Fact]
public async Task UploadFile_WithMaximumSize_DontThrowsException()
{
var fileSize = GuildHelper.GetUploadLimit(_guild);
var fileSize = GuildHelper.GetUploadLimit(_guild.PremiumTier);
using var stream = new MemoryStream(new byte[fileSize]);
await _apiClient.UploadFileAsync(_channel.Id, new UploadFileParams(new FileAttachment(stream, "filename")));
@@ -41,7 +41,7 @@ public class DiscordRestApiClientTests : IClassFixture<RestGuildFixture>, IAsync
[Fact]
public async Task UploadFile_WithOverSize_ThrowsException()
{
var fileSize = GuildHelper.GetUploadLimit(_guild) + 1;
var fileSize = GuildHelper.GetUploadLimit(_guild.PremiumTier) + 1;
using var stream = new MemoryStream(new byte[fileSize]);
Func<Task> upload = async () =>

View File

@@ -14,12 +14,9 @@ public class GuildHelperTests
[InlineData(PremiumTier.Tier3, 100)]
public void GetUploadLimit(PremiumTier tier, ulong factor)
{
var guild = Substitute.For<IGuild>();
guild.PremiumTier.Returns(tier);
var expected = factor * (ulong)Math.Pow(2, 20);
var actual = GuildHelper.GetUploadLimit(guild);
var actual = GuildHelper.GetUploadLimit(tier);
Assert.Equal(expected, actual);
}