InviteCodeApplicationTest.cs 3.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. using Hotline.Api.Controllers;
  2. using Hotline.Application.Snapshot;
  3. using Hotline.Identity.Accounts;
  4. using Hotline.Identity.Roles;
  5. using Hotline.Repository.SqlSugar.Snapshot;
  6. using Hotline.Settings;
  7. using Hotline.Share.Dtos.Snapshot;
  8. using Hotline.Snapshot.Interfaces;
  9. using Hotline.ThirdAccountDomainServices;
  10. using Hotline.ThirdAccountDomainServices.Interfaces;
  11. using Hotline.Users;
  12. using Microsoft.AspNetCore.Http;
  13. using Microsoft.Extensions.DependencyInjection;
  14. using Shouldly;
  15. using System;
  16. using System.Collections.Generic;
  17. using System.Linq;
  18. using System.Text;
  19. using System.Threading.Tasks;
  20. using XF.Domain.Authentications;
  21. using XF.Domain.Cache;
  22. using XF.Domain.Repository;
  23. namespace Hotline.Tests.Application;
  24. public class InviteCodeApplicationTest : TestBase
  25. {
  26. private readonly IInviteCodeApplication _inviteCodeApplication;
  27. private readonly ISnapshotApplication _snapshotApplication;
  28. private readonly IInviteCodeRepository _inviteCodeRepository;
  29. private readonly ISessionContext _sessionContext;
  30. private readonly IInviteCodeRecordRepository _inviteCodeRecordRepository;
  31. private readonly ISnapshotUserInfoRepository _snapshotUserInfoRepository;
  32. public InviteCodeApplicationTest(IAccountRepository accountRepository, IRepository<Role> roleRepository, UserController userController, IServiceScopeFactory scopeFactory, IRepository<User> userRepository, IHttpContextAccessor httpContextAccessor, IThirdIdentiyService thirdIdentiyService, IThirdAccountRepository thirdAccountRepository, IInviteCodeApplication inviteCodeApplication, ISnapshotApplication snapshotApplication, IInviteCodeRepository inviteCodeRepository, ISessionContext sessionContext, IInviteCodeRecordRepository inviteCodeRecordRepository, ITypedCache<SystemSetting> cacheSettingData, ISnapshotUserInfoRepository snapshotUserInfoRepository, ThirdAccountDomainFactory thirdAccountDomainFactory) : base(accountRepository, roleRepository, userController, scopeFactory, userRepository, httpContextAccessor, thirdIdentiyService, thirdAccountRepository, cacheSettingData, thirdAccountDomainFactory)
  33. {
  34. _inviteCodeApplication = inviteCodeApplication;
  35. _snapshotApplication = snapshotApplication;
  36. this._inviteCodeRepository = inviteCodeRepository;
  37. _sessionContext = sessionContext;
  38. _inviteCodeRecordRepository = inviteCodeRecordRepository;
  39. _snapshotUserInfoRepository = snapshotUserInfoRepository;
  40. }
  41. [Fact]
  42. public async Task InviteCode_Test()
  43. {
  44. var inDto = new AddInviteCodeInDto
  45. {
  46. OrgName = "测试部门" + DateTime.Now.ToString("ss"),
  47. BeginCode = 100,
  48. EndCode = 200,
  49. };
  50. var id = await _inviteCodeApplication.AddInviteCodeAsync(inDto);
  51. var items = _inviteCodeApplication.GetInviteCodeItems().ToList();
  52. items.Count.ShouldNotBe(0);
  53. await _inviteCodeRepository.RemoveAsync(id);
  54. SetWeiXin();
  55. await _snapshotUserInfoRepository.Updateable()
  56. .SetColumns(m => m.InvitationCode, null)
  57. .Where(m => m.ThirdAccountId == _sessionContext.UserId)
  58. .ExecuteCommandAsync();
  59. await _inviteCodeRecordRepository.Removeable().Where(m => m.InviteCode == "110").ExecuteCommandAsync();
  60. await _snapshotApplication.SaveInvitationCodeAsync(new SaveInvitationCodeInDto { InvitationCode = "110"});
  61. var statics = await _inviteCodeApplication.GetInviteCodeStatisticAsync(new GetInviteCodeStatisticInDto
  62. {
  63. StartTime = DateTime.Now.AddDays(-1),
  64. EndTime = DateTime.Now
  65. }).ToListAsync();
  66. statics.Count.ShouldNotBe(0);
  67. }
  68. }