103 lines
3.7 KiB
C#
103 lines
3.7 KiB
C#
using Managing.Application.Abstractions;
|
|
using Managing.Application.Abstractions.Repositories;
|
|
using Managing.Application.Abstractions.Services;
|
|
using Managing.Application.Bots.Grains;
|
|
using Managing.Application.Bots.Models;
|
|
using Managing.Domain.Bots;
|
|
using Managing.Domain.Statistics;
|
|
using Microsoft.Extensions.DependencyInjection;
|
|
using Microsoft.Extensions.Logging;
|
|
using Moq;
|
|
using Xunit;
|
|
|
|
namespace Managing.Application.Tests;
|
|
|
|
public class AgentGrainTests
|
|
{
|
|
private readonly Mock<IPersistentState<AgentGrainState>> _mockState;
|
|
private readonly Mock<ILogger<AgentGrain>> _mockLogger;
|
|
private readonly Mock<IBotService> _mockBotService;
|
|
private readonly Mock<IAgentService> _mockAgentService;
|
|
private readonly Mock<IExchangeService> _mockExchangeService;
|
|
private readonly Mock<IUserService> _mockUserService;
|
|
private readonly Mock<IAccountService> _mockAccountService;
|
|
private readonly Mock<ITradingService> _mockTradingService;
|
|
private readonly Mock<IServiceScopeFactory> _mockScopeFactory;
|
|
private readonly Mock<IAgentBalanceRepository> _mockAgentBalanceRepository;
|
|
|
|
public AgentGrainTests()
|
|
{
|
|
_mockState = new Mock<IPersistentState<AgentGrainState>>();
|
|
_mockLogger = new Mock<ILogger<AgentGrain>>();
|
|
_mockBotService = new Mock<IBotService>();
|
|
_mockAgentService = new Mock<IAgentService>();
|
|
_mockExchangeService = new Mock<IExchangeService>();
|
|
_mockUserService = new Mock<IUserService>();
|
|
_mockAccountService = new Mock<IAccountService>();
|
|
_mockTradingService = new Mock<ITradingService>();
|
|
_mockScopeFactory = new Mock<IServiceScopeFactory>();
|
|
_mockAgentBalanceRepository = new Mock<IAgentBalanceRepository>();
|
|
|
|
// Setup default state
|
|
_mockState.Setup(x => x.State).Returns(new AgentGrainState
|
|
{
|
|
AgentName = "TestAgent",
|
|
BotIds = new HashSet<Guid> { Guid.NewGuid() }
|
|
});
|
|
}
|
|
|
|
[Fact]
|
|
public async Task RegisterBotAsync_ShouldUpdateSummary()
|
|
{
|
|
// Arrange
|
|
var agentGrain = CreateAgentGrain();
|
|
var newBotId = Guid.NewGuid();
|
|
|
|
// Setup mocks
|
|
_mockBotService.Setup(x => x.GetBotsByIdsAsync(It.IsAny<HashSet<Guid>>()))
|
|
.ReturnsAsync(new List<Bot>());
|
|
_mockAgentService.Setup(x => x.SaveOrUpdateAgentSummary(It.IsAny<AgentSummary>()))
|
|
.Returns(Task.CompletedTask);
|
|
|
|
// Act
|
|
await agentGrain.RegisterBotAsync(newBotId);
|
|
|
|
// Assert
|
|
_mockAgentService.Verify(x => x.SaveOrUpdateAgentSummary(It.IsAny<AgentSummary>()), Times.Once);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task UnregisterBotAsync_ShouldUpdateSummary()
|
|
{
|
|
// Arrange
|
|
var agentGrain = CreateAgentGrain();
|
|
var botId = _mockState.Object.State.BotIds.First();
|
|
|
|
// Setup mocks
|
|
_mockBotService.Setup(x => x.GetBotsByIdsAsync(It.IsAny<HashSet<Guid>>()))
|
|
.ReturnsAsync(new List<Bot>());
|
|
_mockAgentService.Setup(x => x.SaveOrUpdateAgentSummary(It.IsAny<AgentSummary>()))
|
|
.Returns(Task.CompletedTask);
|
|
|
|
// Act
|
|
await agentGrain.UnregisterBotAsync(botId);
|
|
|
|
// Assert
|
|
_mockAgentService.Verify(x => x.SaveOrUpdateAgentSummary(It.IsAny<AgentSummary>()), Times.Once);
|
|
}
|
|
|
|
private AgentGrain CreateAgentGrain()
|
|
{
|
|
return new AgentGrain(
|
|
_mockState.Object,
|
|
_mockLogger.Object,
|
|
_mockBotService.Object,
|
|
_mockAgentService.Object,
|
|
_mockExchangeService.Object,
|
|
_mockUserService.Object,
|
|
_mockAccountService.Object,
|
|
_mockTradingService.Object,
|
|
_mockAgentBalanceRepository.Object,
|
|
_mockScopeFactory.Object);
|
|
}
|
|
} |