app-cloudep-member-server/internal/logic/account/bind_user_info_logic_test.go

package accountlogic

import (
	"app-cloudep-member-server/gen_result/pb/member"
	libMock "app-cloudep-member-server/internal/mock/lib"
	modelMock "app-cloudep-member-server/internal/mock/model"
	"app-cloudep-member-server/internal/svc"
	"context"
	"fmt"
	"testing"

	"github.com/alicebob/miniredis/v2"
	"github.com/go-sql-driver/mysql"
	"github.com/stretchr/testify/assert"
	"github.com/zeromicro/go-zero/core/stores/redis"
	"go.uber.org/mock/gomock"
	"google.golang.org/protobuf/proto"
)

func TestBindUserInfoLogic_BindUserInfo(t *testing.T) {
	// mock
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	mockUserModel := modelMock.NewMockUserTableModel(ctrl)
	mockValidate := libMock.NewMockValidate(ctrl)

	r1, err := miniredis.Run()
	assert.NoError(t, err)
	defer r1.Close()

	newRedis, err := redis.NewRedis(redis.RedisConf{
		Host: r1.Addr(),
		Type: redis.ClusterType,
		Pass: "",
	})
	assert.NoError(t, err)

	sc := svc.ServiceContext{
		UserModel: mockUserModel,
		Redis:     *newRedis,
		Validate:  mockValidate,
	}

	l := NewBindUserInfoLogic(context.Background(), &sc)

	tests := []struct {
		name        string
		input       *member.CreateUserInfoReq
		setupMocks  func()
		expectError bool
		expected    *member.OKResp
	}{
		{
			name: "ok",
			input: &member.CreateUserInfoReq{
				Uid:        "2024101598765",
				VerifyType: 1,
				AlarmType:  2,
				Status:     3,
				Language:   "en",
				Currency:   "USDT",
				NickName:   proto.String("Test User"),
			},
			setupMocks: func() {
				mockValidate.EXPECT().
					ValidateAll(gomock.Any()).
					Return(nil)
				mockUserModel.EXPECT().
					Insert(gomock.Any(), gomock.Any()).
					Return(nil, nil)
			},
			expectError: false,
			expected:    &member.OKResp{},
		},
		{
			name: "invalid user info",
			input: &member.CreateUserInfoReq{
				Uid: "",
			},
			setupMocks: func() {
				mockValidate.EXPECT().
					ValidateAll(gomock.Any()).
					Return(fmt.Errorf("validation error"))
			},
			expectError: true,
		},
		{
			name: "duplicate entry",
			input: &member.CreateUserInfoReq{
				Uid:        "test_uid",
				VerifyType: 1,
				AlarmType:  2,
				Status:     3,
				Language:   "en",
				Currency:   "USD",
				NickName:   proto.String("Test User"),
			},
			setupMocks: func() {
				mockValidate.EXPECT().
					ValidateAll(gomock.Any()).
					Return(nil)
				mockUserModel.EXPECT().
					Insert(gomock.Any(), gomock.Any()).
					Return(nil, &mysql.MySQLError{Number: 1062, Message: "Duplicate entry"})
			},
			expectError: true,
		},
		{
			name: "database error",
			input: &member.CreateUserInfoReq{
				Uid:        "test_uid",
				VerifyType: 1,
				AlarmType:  2,
				Status:     3,
				Language:   "en",
				Currency:   "USD",
				NickName:   proto.String("Test User"),
			},
			setupMocks: func() {
				mockValidate.EXPECT().
					ValidateAll(gomock.Any()).
					Return(nil)
				mockUserModel.EXPECT().
					Insert(gomock.Any(), gomock.Any()).
					Return(nil, fmt.Errorf("database error"))
			},
			expectError: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.setupMocks()
			resp, err := l.BindUserInfo(tt.input)
			if tt.expectError {
				assert.Error(t, err)
			} else {
				assert.NoError(t, err)
				assert.Equal(t, tt.expected, resp)
			}
		})
	}
}