tests: fix tests

- register test model before each test

Author: srebrek

src/MaIN.Core.UnitTests/AgentContextTests.cs

@@ -3,6 +3,7 @@
 using MaIN.Domain.Entities.Agents;
 using MaIN.Domain.Entities.Agents.Knowledge;
 using MaIN.Domain.Exceptions.Agents;
+using MaIN.Domain.Models.Abstract;
 using MaIN.Services.Services.Abstract;
 using MaIN.Services.Services.Models;
 using Moq;
@@ -13,11 +14,14 @@ public class AgentContextTests
 {
     private readonly Mock<IAgentService> _mockAgentService;
     private readonly AgentContext _agentContext;
+    private readonly string _testModelId = "test-model";
 
     public AgentContextTests()
     {
         _mockAgentService = new Mock<IAgentService>();
         _agentContext = new AgentContext(_mockAgentService.Object);
+        var testModel = new GenericLocalModel(_testModelId);
+        ModelRegistry.RegisterOrReplace(testModel);
     }
 
     [Fact]
@@ -159,7 +163,7 @@ public async Task ProcessAsync_WithStringMessage_ShouldReturnChatResult()
     {
         // Arrange
         var message = "Hello, agent!";
-        var chat = new Chat { Id = _agentContext.GetAgentId(), Messages = new List<Message>(), Name = "test", ModelId = "default"};
+        var chat = new Chat { Id = _agentContext.GetAgentId(), Messages = new List<Message>(), Name = "test", ModelId = _testModelId};
         var chatResult = new ChatResult { Done = true, Model = "test-model", Message = new Message
             {
                 Role = "Assistant",

src/MaIN.Core.UnitTests/ChatContextTests.cs

@@ -12,11 +12,14 @@ public class ChatContextTests
 {
     private readonly Mock<IChatService> _mockChatService;
     private readonly ChatContext _chatContext;
+    private readonly string _testModelId = "test-model";
 
     public ChatContextTests()
     {
         _mockChatService = new Mock<IChatService>();
         _chatContext = new ChatContext(_mockChatService.Object);
+        var testModel = new GenericLocalModel(_testModelId);
+        ModelRegistry.RegisterOrReplace(testModel);
     }
 
     [Fact]
@@ -103,7 +106,7 @@ public async Task CompleteAsync_ShouldCallChatService()
     public async Task GetCurrentChat_ShouldCallChatService()
     {
         // Arrange
-        var chat = new Chat { Id = _chatContext.GetChatId(), ModelId = "default", Name = "test"};
+        var chat = new Chat { Id = _chatContext.GetChatId(), ModelId = _testModelId , Name = "test"};
         _mockChatService.Setup(s => s.GetById(chat.Id)).ReturnsAsync(chat);
 
         // Act
@@ -117,14 +120,14 @@ public async Task GetCurrentChat_ShouldCallChatService()
     public async Task WithModel_ShouldSetModelIdAndInstance()
     {
         // Arrange
-        var model = new GenericLocalModel("default");
+        var model = new GenericLocalModel(_testModelId);
 
         // Act
         await _chatContext.WithModel(model)
             .WithMessage("User message")
             .CompleteAsync();
 
         // Assert
-        _mockChatService.Verify(s => s.Completions(It.Is<Chat>(c => c.ModelId == "default" && c.ModelInstance == model), It.IsAny<bool>(), It.IsAny<bool>(), null), Times.Once);
+        _mockChatService.Verify(s => s.Completions(It.Is<Chat>(c => c.ModelId == _testModelId && c.ModelInstance == model), It.IsAny<bool>(), It.IsAny<bool>(), null), Times.Once);
     }
 }

src/MaIN.Core.UnitTests/FlowContextTests.cs

@@ -1,11 +1,12 @@
 using MaIN.Core.Hub.Contexts;
 using MaIN.Domain.Entities;
 using MaIN.Domain.Entities.Agents;
-using MaIN.Services.Services.Abstract;
-using Moq;
 using MaIN.Domain.Entities.Agents.AgentSource;
 using MaIN.Domain.Entities.Agents.Knowledge;
 using MaIN.Domain.Exceptions.Flows;
+using MaIN.Domain.Models.Abstract;
+using MaIN.Services.Services.Abstract;
+using Moq;
 
 namespace MaIN.Core.UnitTests;
 
@@ -14,12 +15,15 @@ public class FlowContextTests
     private readonly Mock<IAgentFlowService> _mockFlowService;
     private readonly Mock<IAgentService> _mockAgentService;
     private FlowContext _flowContext;
+    private readonly string _testModelId = "test-model";
 
     public FlowContextTests()
     {
         _mockFlowService = new Mock<IAgentFlowService>();
         _mockAgentService = new Mock<IAgentService>();
         _flowContext = new FlowContext(_mockFlowService.Object, _mockAgentService.Object);
+        var testModel = new GenericLocalModel(_testModelId);
+        ModelRegistry.RegisterOrReplace(testModel);
     }
 
     [Fact]
@@ -92,7 +96,7 @@ public async Task ProcessAsync_WithStringMessage_ShouldReturnChatResult()
         _flowContext.AddAgent(firstAgent);
 
         var message = "Hello, flow!";
-        var chat = new Chat { Id = firstAgent.Id, Messages = new List<Message>(), ModelId = "default", Name = "test"};
+        var chat = new Chat { Id = firstAgent.Id, Messages = new List<Message>(), ModelId = _testModelId, Name = "test"};
 
         _mockAgentService
             .Setup(s => s.GetChatByAgent(firstAgent.Id))
@@ -101,7 +105,7 @@ public async Task ProcessAsync_WithStringMessage_ShouldReturnChatResult()
         _mockAgentService
             .Setup(s => s.Process(It.IsAny<Chat>(), firstAgent.Id, It.IsAny<Knowledge>(), It.IsAny<bool>(), null, null))
             .ReturnsAsync(new Chat { 
-                ModelId = "test-model", 
+                ModelId = _testModelId, 
                 Name = "test",
                 Messages = new List<Message> { 
                     new() { Content = "Response", Role = "Assistant", Type = MessageType.LocalLLM} 
@@ -113,7 +117,7 @@ public async Task ProcessAsync_WithStringMessage_ShouldReturnChatResult()
 
         // Assert
         Assert.True(result.Done);
-        Assert.Equal("test-model", result.Model);
+        Assert.Equal(_testModelId, result.Model);
         Assert.NotNull(result.Message);
     }