Fix tests and warnings

This commit is contained in:
Dmitri Shimanski
2025-03-19 15:10:56 +02:00
parent 37bfd816e4
commit aa4b86d12f
6 changed files with 65 additions and 91 deletions

View File

@@ -12,9 +12,9 @@ jobs:
build:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v3
- uses: actions/checkout@v4
- name: Setup .NET
uses: actions/setup-dotnet@v3
uses: actions/setup-dotnet@v4
with:
dotnet-version: 7.0.x
- name: Restore dependencies
@@ -22,4 +22,4 @@ jobs:
- name: Build
run: dotnet build --no-restore /p:TreatWarningsAsErrors=false
- name: Test
run: dotnet test --no-build --verbosity normal
run: dotnet test --no-build --verbosity normal -e telegram_test_token=${{secrets.TELEGRAM_TEST_TOKEN}}

View File

@@ -5,6 +5,7 @@
<TargetFramework>net7.0</TargetFramework>
<ImplicitUsings>enable</ImplicitUsings>
<Nullable>enable</Nullable>
<SuppressTfmSupportBuildWarnings>true</SuppressTfmSupportBuildWarnings>
</PropertyGroup>
<ItemGroup>

View File

@@ -14,14 +14,14 @@ namespace Telegram.Net.Services;
public class TelegramHostedService : IHostedService
{
private IServiceCollection isc { get; }
private TelegramBotClient Client { get; }
internal TelegramBotClient Client { get; set; }
private ITelegramBotConfig Config { get; }
internal static Dictionary<string, Func<ITelegramBotClient, Message, CancellationToken, Task>> CommandHandler { get; } = new();
internal static List<Func<ITelegramBotClient, Message, CancellationToken, Task>> EditedMessageHandler { get; } = new();
internal static Dictionary<string, Func<ITelegramBotClient, CallbackQuery,CancellationToken, Task>> CallbackQueryHandler { get; } = new();
internal static Dictionary<string, Func<ITelegramBotClient, InlineQuery ,CancellationToken, Task>> InlineHandler { get; } = new();
internal static Func<ITelegramBotClient, PreCheckoutQuery,CancellationToken, Task>? PreCheckoutHandler { get; set; }
internal static List<Func<ITelegramBotClient, Update, CancellationToken, Task>> DefaultUpdateHandler { get; } = new();
internal Dictionary<string, Func<ITelegramBotClient, Message, CancellationToken, Task>> CommandHandler { get; } = new();
internal List<Func<ITelegramBotClient, Message, CancellationToken, Task>> EditedMessageHandler { get; } = new();
internal Dictionary<string, Func<ITelegramBotClient, CallbackQuery,CancellationToken, Task>> CallbackQueryHandler { get; } = new();
internal Dictionary<string, Func<ITelegramBotClient, InlineQuery ,CancellationToken, Task>> InlineHandler { get; } = new();
internal Func<ITelegramBotClient, PreCheckoutQuery,CancellationToken, Task>? PreCheckoutHandler { get; set; }
internal List<Func<ITelegramBotClient, Update, CancellationToken, Task>> DefaultUpdateHandler { get; } = new();
public TelegramHostedService(ITelegramBotConfig config, IServiceCollection isc)
{
@@ -47,10 +47,8 @@ public class TelegramHostedService : IHostedService
internal async Task AddAttributes(CancellationToken cancellationToken)
{
var mutex = new Mutex();
await Task.Run(async () =>
{
mutex.WaitOne();
var implementations = AppDomain.CurrentDomain.GetAssemblies()
.SelectMany(a => a.GetTypes())
.Where(t => typeof(IUpdatePollingService).IsAssignableFrom(t) && !t.IsInterface);
@@ -82,7 +80,7 @@ public class TelegramHostedService : IHostedService
if (IsValidHandlerMethod(method, typeof(Message)))
{
var handler = CreateDelegate<Message>(method);
CommandHandler.Add(commandAttr.Command, handler);
CommandHandler.TryAdd(commandAttr.Command, handler);
}
continue;
}
@@ -93,7 +91,7 @@ public class TelegramHostedService : IHostedService
if (IsValidHandlerMethod(method, typeof(CallbackQuery)))
{
var handler = CreateDelegate<CallbackQuery>(method);
CallbackQueryHandler.Add(callbackAttr.QueryId, handler);
CallbackQueryHandler.TryAdd(callbackAttr.QueryId, handler);
}
continue;
}
@@ -115,7 +113,7 @@ public class TelegramHostedService : IHostedService
if (IsValidHandlerMethod(method, typeof(InlineQuery)))
{
var handler = CreateDelegate<InlineQuery>(method);
InlineHandler.Add(inlineAttr.InlineId, handler);
InlineHandler.TryAdd(inlineAttr.InlineId, handler);
}
continue;
}
@@ -142,44 +140,44 @@ public class TelegramHostedService : IHostedService
continue;
}
}
mutex.ReleaseMutex();
}, cancellationToken);
}
internal async Task UpdateHandler(ITelegramBotClient client, Update update, CancellationToken ctx)
{
switch (update)
{
case { Message: { } message }:
await CommandHandler.FirstOrDefault(k => message.Text!.StartsWith(k.Key)).Value(client, message, ctx);
break;
case { EditedMessage: { } message }:
EditedMessageHandler.ForEach(async k => await k(client, message, ctx));
break;
case { CallbackQuery: { } callbackQuery }:
await CallbackQueryHandler.FirstOrDefault(k => callbackQuery.Data!.StartsWith(k.Key)).Value(client, callbackQuery, ctx);
break;
case { InlineQuery: { } inlineQuery }:
await InlineHandler.FirstOrDefault(k => inlineQuery.Id.StartsWith(k.Key)).Value(client, inlineQuery, ctx);
break;
case { PreCheckoutQuery: { } preCheckoutQuery }:
if (PreCheckoutHandler != null) await PreCheckoutHandler(client, preCheckoutQuery, ctx);
break;
default:
DefaultUpdateHandler.ForEach(async k => await k(client, update, ctx));
break;
}
}
[SuppressMessage("ReSharper", "AsyncVoidLambda")]
public async Task StartAsync(CancellationToken cancellationToken)
{
await AddAttributes(cancellationToken);
Client.StartReceiving(
async (client, update, ctx) =>
{
switch (update)
{
case { Message: { } message }:
await CommandHandler.FirstOrDefault(k => message.Text!.StartsWith(k.Key)).Value(client, message, cancellationToken);
break;
case { EditedMessage: { } message }:
EditedMessageHandler.ForEach(async k => await k(client, message, cancellationToken));
break;
case { CallbackQuery: { } callbackQuery }:
await CallbackQueryHandler.FirstOrDefault(k => callbackQuery.Data!.StartsWith(k.Key)).Value(client, callbackQuery, cancellationToken);
break;
case { InlineQuery: { } inlineQuery }:
await InlineHandler.FirstOrDefault(k => inlineQuery.Id.StartsWith(k.Key)).Value(client, inlineQuery, cancellationToken);
break;
case {PreCheckoutQuery: { } preCheckoutQuery}:
if (PreCheckoutHandler != null)
await PreCheckoutHandler(client, preCheckoutQuery, cancellationToken);
break;
default:
DefaultUpdateHandler.ForEach(async k => await k(client, update, ctx));
break;
}
},
UpdateHandler,
Config.errorHandler ?? ((_, _, _) => Task.CompletedTask),
Config.ReceiverOptions,
cancellationToken);

View File

@@ -1,6 +1,7 @@
<Project Sdk="Microsoft.NET.Sdk">
<PropertyGroup>
<SuppressTfmSupportBuildWarnings>true</SuppressTfmSupportBuildWarnings>
<TargetFramework>net7.0</TargetFramework>
<ImplicitUsings>enable</ImplicitUsings>
<Nullable>enable</Nullable>

View File

@@ -4,7 +4,7 @@
<TargetFramework>net7.0</TargetFramework>
<ImplicitUsings>enable</ImplicitUsings>
<Nullable>enable</Nullable>
<SuppressTfmSupportBuildWarnings>true</SuppressTfmSupportBuildWarnings>
<IsPackable>false</IsPackable>
<IsTestProject>true</IsTestProject>
</PropertyGroup>

View File

@@ -13,13 +13,14 @@ using Microsoft.Extensions.DependencyInjection;
using Moq;
using Telegram.Bot.Types.Enums;
using Telegram.Bot.Types.ReplyMarkups;
using Telegram.Net.Models;
namespace Telegram.Tests
{
[TestFixture]
public class TelegramHostedServiceTests
{
private Mock<ITelegramBotConfig> _configMock;
private TelegramBotConfig _configMock;
private ServiceCollection _services;
[SetUp]
@@ -30,14 +31,14 @@ namespace Telegram.Tests
.AddJsonFile("appsettings.json", false)
.AddEnvironmentVariables()
.Build();
_configMock = new Mock<ITelegramBotConfig>();
_configMock.Setup(c => c.Token).Returns(conf.GetValue<string>("telegram_test_token"));
_configMock = new TelegramBotConfig(conf.GetValue<string>("telegram_test_token") ?? throw new Exception("Provide telegram token first"));
_services = new ServiceCollection();
}
public class TestHandler : IUpdatePollingService
{
[Command("start")]
[Command("/start")]
public async Task HandleStart(ITelegramBotClient client, Message message, CancellationToken ct)
=> await client.SendTextMessageAsync(message.Chat.Id, "Started", cancellationToken: ct);
@@ -66,18 +67,18 @@ namespace Telegram.Tests
public void AddAttributes_RegistersCommandHandlersCorrectly()
{
_services.AddSingleton<TestHandler>();
var service = new TelegramHostedService(_configMock.Object, _services);
var service = new TelegramHostedService(_configMock, _services);
service.AddAttributes(CancellationToken.None).Wait();
Assert.Multiple(() =>
{
Assert.That(TelegramHostedService.CommandHandler, Contains.Key("start"));
Assert.That(TelegramHostedService.CallbackQueryHandler, Contains.Key("test_callback"));
Assert.That(TelegramHostedService.EditedMessageHandler, Has.Count.EqualTo(1));
Assert.That(TelegramHostedService.InlineHandler, Contains.Key("search"));
Assert.That(TelegramHostedService.PreCheckoutHandler, Is.Not.Null);
Assert.That(TelegramHostedService.DefaultUpdateHandler, Has.Count.EqualTo(1));
Assert.That(service.CommandHandler, Contains.Key("/start"));
Assert.That(service.CallbackQueryHandler, Contains.Key("test_callback"));
Assert.That(service.EditedMessageHandler, Has.Count.EqualTo(1));
Assert.That(service.InlineHandler, Contains.Key("search"));
Assert.That(service.PreCheckoutHandler, Is.Not.Null);
Assert.That(service.DefaultUpdateHandler, Has.Count.EqualTo(1));
});
}
@@ -118,8 +119,14 @@ namespace Telegram.Tests
[SetUp]
public void Setup()
{
var conf = new ConfigurationBuilder()
.SetBasePath(AppContext.BaseDirectory)
.AddJsonFile("appsettings.json", false)
.AddEnvironmentVariables()
.Build();
_configMock = new Mock<ITelegramBotConfig>();
_configMock.Setup(c => c.Token).Returns("test_token");
_configMock.Setup(c => c.Token).Returns(conf.GetValue<string>("telegram_test_token") ?? throw new Exception("Provide token first"));
_configMock.Setup(c => c.ReceiverOptions).Returns(new ReceiverOptions());
var services = new ServiceCollection();
@@ -137,23 +144,6 @@ namespace Telegram.Tests
var message = new Message { Text = "/start", Chat = new Chat { Id = 123 } };
var update = new Update { Message = message };
_botClientMock.Setup(b => b.SendMessage(
It.IsAny<ChatId>(),
"Started",
It.IsAny<ParseMode>(),
It.IsAny<ReplyParameters>(),
It.IsAny<ReplyMarkup>(),
It.IsAny<LinkPreviewOptions>(),
It.IsAny<int>(),
It.IsAny<IEnumerable<MessageEntity>>(),
It.IsAny<bool>(),
It.IsAny<bool>(),
It.IsAny<string>(),
It.IsAny<string>(),
It.IsAny<bool>(),
It.IsAny<CancellationToken>()
)).Verifiable();
// Act
await _hostedService.StartAsync(CancellationToken.None);
await InvokeUpdateHandler(update);
@@ -169,15 +159,6 @@ namespace Telegram.Tests
var callback = new CallbackQuery { Data = "test_callback", Id = "cb_id" };
var update = new Update { CallbackQuery = callback };
_botClientMock.Setup(b => b.AnswerCallbackQueryAsync(
"cb_id",
"Callback handled",
It.IsAny<bool>(),
It.IsAny<string>(),
It.IsAny<int>(),
It.IsAny<CancellationToken>()
)).Verifiable();
// Act
await _hostedService.StartAsync(CancellationToken.None);
await InvokeUpdateHandler(update);
@@ -188,14 +169,7 @@ namespace Telegram.Tests
private async Task InvokeUpdateHandler(Update update)
{
var clientField = typeof(TelegramHostedService).GetField("Client", BindingFlags.NonPublic | BindingFlags.Instance);
clientField.SetValue(_hostedService, _botClientMock.Object);
var updateHandler = _botClientMock.Invocations
.First(i => i.Method.Name == "StartReceiving")
.Arguments[0] as Func<ITelegramBotClient, Update, CancellationToken, Task>;
await updateHandler(_botClientMock.Object, update, CancellationToken.None);
await _hostedService.UpdateHandler(_botClientMock.Object, update, CancellationToken.None);
}
}
}