测试覆盖率

WebApi.Test/Transaction/TransactionPeriodicServiceTest.cs

@@ -0,0 +1,608 @@
+using Microsoft.Extensions.Logging;
+using Service.Transaction;
+
+namespace WebApi.Test.Transaction;
+
+public class TransactionPeriodicServiceTest : BaseTest
+{
+    private readonly ITransactionPeriodicRepository _periodicRepository = Substitute.For<ITransactionPeriodicRepository>();
+    private readonly ITransactionRecordRepository _transactionRepository = Substitute.For<ITransactionRecordRepository>();
+    private readonly IMessageRecordRepository _messageRepository = Substitute.For<IMessageRecordRepository>();
+    private readonly ILogger<TransactionPeriodicService> _logger = Substitute.For<ILogger<TransactionPeriodicService>>();
+    private readonly ITransactionPeriodicService _service;
+
+    public TransactionPeriodicServiceTest()
+    {
+        _service = new TransactionPeriodicService(
+            _periodicRepository,
+            _transactionRepository,
+            _messageRepository,
+            _logger
+        );
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_每日账单()
+    {
+        // Arrange
+        var today = new DateTime(2024, 1, 15, 10, 0, 0);
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Daily,
+            Amount = 100m,
+            Type = TransactionType.Expense,
+            Classify = "餐饮",
+            Reason = "每日餐费",
+            IsEnabled = true,
+            LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+            NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(Task.CompletedTask);
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _transactionRepository.Received(1).AddAsync(Arg.Is<TransactionRecord>(t =>
+            t.Amount == -100m &&
+            t.Type == TransactionType.Expense &&
+            t.Classify == "餐饮" &&
+            t.Reason == "每日餐费" &&
+            t.Card == "周期性账单" &&
+            t.ImportFrom == "周期性账单自动生成"
+        ));
+
+        await _messageRepository.Received(1).AddAsync(Arg.Is<MessageRecord>(m =>
+            m.Title == "周期性账单提醒" &&
+            m.Content.Contains("支出") &&
+            m.Content.Contains("100.00") &&
+            m.Content.Contains("每日餐费") &&
+            m.IsRead == false
+        ));
+
+        await _periodicRepository.Received(1).UpdateExecuteTimeAsync(
+            Arg.Is(1L),
+            Arg.Is<DateTime>(dt => dt.Date == today.Date),
+            Arg.Is<DateTime?>(dt => dt.HasValue && dt.Value.Date == today.Date.AddDays(1))
+        );
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_每周账单()
+    {
+        // Arrange
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Weekly,
+            PeriodicConfig = "1,3,5", // 周一、三、五
+            Amount = 200m,
+            Type = TransactionType.Expense,
+            Classify = "交通",
+            Reason = "每周通勤费",
+            IsEnabled = true,
+            LastExecuteTime = new DateTime(2024, 1, 10, 10, 0, 0),
+            NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(Task.CompletedTask);
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _transactionRepository.Received(1).AddAsync(Arg.Is<TransactionRecord>(t =>
+            t.Amount == -200m &&
+            t.Type == TransactionType.Expense &&
+            t.Classify == "交通" &&
+            t.Reason == "每周通勤费"
+        ));
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_每月账单()
+    {
+        // Arrange
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Monthly,
+            PeriodicConfig = "1,15", // 每月1号和15号
+            Amount = 5000m,
+            Type = TransactionType.Income,
+            Classify = "工资",
+            Reason = "每月工资",
+            IsEnabled = true,
+            LastExecuteTime = new DateTime(2024, 1, 1, 10, 0, 0),
+            NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(Task.CompletedTask);
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _transactionRepository.Received(1).AddAsync(Arg.Is<TransactionRecord>(t =>
+            t.Amount == 5000m &&
+            t.Type == TransactionType.Income &&
+            t.Classify == "工资" &&
+            t.Reason == "每月工资"
+        ));
+
+        await _messageRepository.Received(1).AddAsync(Arg.Is<MessageRecord>(m =>
+            m.Content.Contains("收入") &&
+            m.Content.Contains("5000.00") &&
+            m.Content.Contains("每月工资")
+        ));
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_未达到执行时间()
+    {
+        // Arrange
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Weekly,
+            PeriodicConfig = "1,3,5", // 只在周一(1)、三(3)、五(5)执行
+            Amount = 200m,
+            Type = TransactionType.Expense,
+            Classify = "交通",
+            Reason = "每周通勤费",
+            IsEnabled = true,
+            LastExecuteTime = new DateTime(2024, 1, 10, 10, 0, 0),
+            NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _transactionRepository.Received(0).AddAsync(Arg.Any<TransactionRecord>());
+        await _messageRepository.Received(0).AddAsync(Arg.Any<MessageRecord>());
+        await _periodicRepository.Received(0).UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>());
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_今天已执行过()
+    {
+        // Arrange
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Daily,
+            Amount = 100m,
+            Type = TransactionType.Expense,
+            Classify = "餐饮",
+            Reason = "每日餐费",
+            IsEnabled = true,
+            LastExecuteTime = new DateTime(2024, 1, 15, 8, 0, 0), // 今天已经执行过
+            NextExecuteTime = new DateTime(2024, 1, 16, 10, 0, 0)
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        // 由于 LastExecuteTime 日期是今天，所以不会再次执行
+        await _transactionRepository.Received(0).AddAsync(Arg.Any<TransactionRecord>());
+        await _messageRepository.Received(0).AddAsync(Arg.Any<MessageRecord>());
+        await _periodicRepository.Received(0).UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>());
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_从未执行过()
+    {
+        // Arrange
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Weekly,
+            PeriodicConfig = "1,3,5",
+            Amount = 200m,
+            Type = TransactionType.Expense,
+            Classify = "交通",
+            Reason = "每周通勤费",
+            IsEnabled = true,
+            LastExecuteTime = null, // 从未执行过
+            NextExecuteTime = null
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(Task.CompletedTask);
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _transactionRepository.Received(1).AddAsync(Arg.Any<TransactionRecord>());
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_添加交易记录失败()
+    {
+        // Arrange
+        var periodicBill = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Daily,
+            Amount = 100m,
+            Type = TransactionType.Expense,
+            Classify = "餐饮",
+            Reason = "每日餐费",
+            IsEnabled = true,
+            LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+            NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(new[] { periodicBill });
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(false); // 添加失败
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _messageRepository.Received(0).AddAsync(Arg.Any<MessageRecord>());
+        await _periodicRepository.Received(0).UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>());
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_多条账单()
+    {
+        // Arrange
+        var periodicBills = new[]
+        {
+            new TransactionPeriodic
+            {
+                Id = 1,
+                PeriodicType = PeriodicType.Daily,
+                Amount = 100m,
+                Type = TransactionType.Expense,
+                Classify = "餐饮",
+                Reason = "每日餐费",
+                IsEnabled = true,
+                LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+                NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+            },
+            new TransactionPeriodic
+            {
+                Id = 2,
+                PeriodicType = PeriodicType.Daily,
+                Amount = 200m,
+                Type = TransactionType.Expense,
+                Classify = "交通",
+                Reason = "每日交通",
+                IsEnabled = true,
+                LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+                NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+            }
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(periodicBills);
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(Task.CompletedTask);
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        await _transactionRepository.Received(2).AddAsync(Arg.Any<TransactionRecord>());
+        await _messageRepository.Received(2).AddAsync(Arg.Any<MessageRecord>());
+        await _periodicRepository.Received(2).UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>());
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每日()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Daily,
+            PeriodicConfig = ""
+        };
+        var baseTime = new DateTime(2024, 1, 15, 10, 0, 0);
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 1, 16, 0, 0, 0));
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每周_本周内()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Weekly,
+            PeriodicConfig = "1,3,5" // 周一、三、五
+        };
+        var baseTime = new DateTime(2024, 1, 15, 10, 0, 0); // 周一
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 1, 17, 0, 0, 0)); // 下周三
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每周_跨周()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Weekly,
+            PeriodicConfig = "1,3,5" // 周一、三、五
+        };
+        var baseTime = new DateTime(2024, 1, 19, 10, 0, 0); // 周五
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 1, 22, 0, 0, 0)); // 下周一
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每月_本月内()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Monthly,
+            PeriodicConfig = "1,15" // 每月1号和15号
+        };
+        var baseTime = new DateTime(2024, 1, 10, 10, 0, 0);
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 1, 15, 0, 0, 0));
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每月_跨月()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Monthly,
+            PeriodicConfig = "1,15" // 每月1号和15号
+        };
+        var baseTime = new DateTime(2024, 1, 16, 10, 0, 0);
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 2, 1, 0, 0, 0));
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每月_月末()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Monthly,
+            PeriodicConfig = "30,31" // 每月30号和31号
+        };
+        var baseTime = new DateTime(2024, 1, 15, 10, 0, 0); // 1月只有31天
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 1, 30, 0, 0, 0));
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每月_小月()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Monthly,
+            PeriodicConfig = "30,31" // 每月30号和31号
+        };
+        var baseTime = new DateTime(2024, 4, 25, 10, 0, 0); // 4月只有30天
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 4, 30, 0, 0, 0)); // 30号，31号不存在
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每季度()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Quarterly,
+            PeriodicConfig = "15" // 每季度第15天
+        };
+        var baseTime = new DateTime(2024, 1, 15, 10, 0, 0);
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2024, 4, 15, 0, 0, 0)); // 下季度
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_每年()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = PeriodicType.Yearly,
+            PeriodicConfig = "100" // 每年第100天
+        };
+        var baseTime = new DateTime(2024, 4, 10, 10, 0, 0); // 第100天是4月9日
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().Be(new DateTime(2025, 4, 9, 0, 0, 0)); // 下一年
+    }
+
+    [Fact]
+    public void CalculateNextExecuteTime_未知周期类型()
+    {
+        // Arrange
+        var periodic = new TransactionPeriodic
+        {
+            Id = 1,
+            PeriodicType = (PeriodicType)99, // 未知类型
+            PeriodicConfig = ""
+        };
+        var baseTime = new DateTime(2024, 1, 15, 10, 0, 0);
+
+        // Act
+        var result = _service.CalculateNextExecuteTime(periodic, baseTime);
+
+        // Assert
+        result.Should().BeNull();
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_处理异常不中断()
+    {
+        // Arrange
+        var periodicBills = new[]
+        {
+            new TransactionPeriodic
+            {
+                Id = 1,
+                PeriodicType = PeriodicType.Daily,
+                Amount = 100m,
+                Type = TransactionType.Expense,
+                Classify = "餐饮",
+                Reason = "每日餐费",
+                IsEnabled = true,
+                LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+                NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+            },
+            new TransactionPeriodic
+            {
+                Id = 2,
+                PeriodicType = PeriodicType.Daily,
+                Amount = 200m,
+                Type = TransactionType.Expense,
+                Classify = "交通",
+                Reason = "每日交通",
+                IsEnabled = true,
+                LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+                NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+            }
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(periodicBills);
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(args =>
+            {
+                var id = (long)args[0];
+                if (id == 1)
+                {
+                    throw new Exception("更新失败");
+                }
+                return Task.CompletedTask;
+            });
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        // 第二条记录应该成功处理
+        await _transactionRepository.Received(2).AddAsync(Arg.Any<TransactionRecord>());
+    }
+
+    [Fact]
+    public async Task ExecutePeriodicBillsAsync_处理所有账单()
+    {
+        // Arrange
+        var periodicBills = new[]
+        {
+            new TransactionPeriodic
+            {
+                Id = 1,
+                PeriodicType = PeriodicType.Daily,
+                Amount = 100m,
+                Type = TransactionType.Expense,
+                Classify = "餐饮",
+                Reason = "每日餐费",
+                IsEnabled = false, // 禁用
+                LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+                NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+            },
+            new TransactionPeriodic
+            {
+                Id = 2,
+                PeriodicType = PeriodicType.Daily,
+                Amount = 200m,
+                Type = TransactionType.Expense,
+                Classify = "交通",
+                Reason = "每日交通",
+                IsEnabled = true,
+                LastExecuteTime = new DateTime(2024, 1, 14, 10, 0, 0),
+                NextExecuteTime = new DateTime(2024, 1, 15, 10, 0, 0)
+            }
+        };
+
+        _periodicRepository.GetPendingPeriodicBillsAsync().Returns(periodicBills);
+        _transactionRepository.AddAsync(Arg.Any<TransactionRecord>()).Returns(Task.FromResult(true));
+        _messageRepository.AddAsync(Arg.Any<MessageRecord>()).Returns(Task.CompletedTask);
+        _periodicRepository.UpdateExecuteTimeAsync(Arg.Any<long>(), Arg.Any<DateTime>(), Arg.Any<DateTime>())
+            .Returns(Task.CompletedTask);
+
+        // Act
+        await _service.ExecutePeriodicBillsAsync();
+
+        // Assert
+        // 只有启用的账单会被处理
+        await _transactionRepository.Received(1).AddAsync(Arg.Is<TransactionRecord>(t => t.Reason == "每日交通"));
+        await _transactionRepository.Received(0).AddAsync(Arg.Is<TransactionRecord>(t => t.Reason == "每日餐费"));
+    }
+}