account_test.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122
  1. package state
  2. import (
  3. "context"
  4. "testing"
  5. "github.com/stretchr/testify/assert"
  6. )
  7. func TestNewAccountCreator(t *testing.T) {
  8. tests := []struct {
  9. name string
  10. screenName DisplayScreenName
  11. password string
  12. insertUserErr error
  13. wantErr error
  14. validateUser func(*testing.T, User)
  15. }{
  16. {
  17. name: "Success with valid AIM handle",
  18. screenName: "TestUser123",
  19. password: "validpass123",
  20. wantErr: nil,
  21. validateUser: func(t *testing.T, u User) {
  22. assert.Equal(t, DisplayScreenName("TestUser123"), u.DisplayScreenName)
  23. assert.Equal(t, NewIdentScreenName("testuser123"), u.IdentScreenName)
  24. assert.False(t, u.IsICQ)
  25. assert.NotEmpty(t, u.AuthKey)
  26. assert.NotNil(t, u.WeakMD5Pass)
  27. assert.NotNil(t, u.StrongMD5Pass)
  28. },
  29. },
  30. {
  31. name: "Success with valid UIN",
  32. screenName: "12345678",
  33. password: "valid12",
  34. wantErr: nil,
  35. validateUser: func(t *testing.T, u User) {
  36. assert.Equal(t, DisplayScreenName("12345678"), u.DisplayScreenName)
  37. assert.Equal(t, NewIdentScreenName("12345678"), u.IdentScreenName)
  38. assert.True(t, u.IsICQ)
  39. assert.NotEmpty(t, u.AuthKey)
  40. assert.NotNil(t, u.WeakMD5Pass)
  41. assert.NotNil(t, u.StrongMD5Pass)
  42. },
  43. },
  44. {
  45. name: "Success with AIM handle containing spaces",
  46. screenName: "Test User 123",
  47. password: "validpass123",
  48. wantErr: nil,
  49. validateUser: func(t *testing.T, u User) {
  50. assert.Equal(t, DisplayScreenName("Test User 123"), u.DisplayScreenName)
  51. assert.Equal(t, NewIdentScreenName("test user 123"), u.IdentScreenName)
  52. assert.False(t, u.IsICQ)
  53. },
  54. },
  55. {
  56. name: "Invalid AIM handle - too short",
  57. screenName: "Us",
  58. password: "validpass123",
  59. wantErr: ErrAIMHandleLength,
  60. },
  61. {
  62. name: "Invalid AIM handle - starts with number",
  63. screenName: "1User",
  64. password: "validpass123",
  65. wantErr: ErrAIMHandleInvalidFormat,
  66. },
  67. {
  68. name: "Invalid AIM handle - ends with space",
  69. screenName: "User123 ",
  70. password: "validpass123",
  71. wantErr: ErrAIMHandleInvalidFormat,
  72. },
  73. {
  74. name: "Invalid UIN - too small",
  75. screenName: "9999",
  76. password: "valid12",
  77. wantErr: ErrICQUINInvalidFormat,
  78. },
  79. {
  80. name: "Invalid UIN - too large",
  81. screenName: "2147483647",
  82. password: "valid12",
  83. wantErr: ErrICQUINInvalidFormat,
  84. },
  85. {
  86. name: "Invalid UIN - contains non-digits",
  87. screenName: "12345abc",
  88. password: "valid12",
  89. wantErr: ErrAIMHandleInvalidFormat, // Will be validated as AIM handle since IsUIN() returns false
  90. },
  91. {
  92. name: "insertUser error",
  93. screenName: "TestUser123",
  94. password: "validpass123",
  95. insertUserErr: assert.AnError,
  96. wantErr: assert.AnError,
  97. },
  98. }
  99. for _, tt := range tests {
  100. t.Run(tt.name, func(t *testing.T) {
  101. var capturedUser User
  102. insertUser := func(ctx context.Context, u User) error {
  103. capturedUser = u
  104. return tt.insertUserErr
  105. }
  106. createAccount := NewAccountCreator(insertUser)
  107. err := createAccount(context.Background(), tt.screenName, tt.password)
  108. assert.ErrorIs(t, err, tt.wantErr)
  109. if tt.validateUser != nil {
  110. tt.validateUser(t, capturedUser)
  111. }
  112. })
  113. }
  114. }