TinySkin/service/user_test.go

package service

import (
	"context"
	"crypto/rand"
	"crypto/rsa"
	"net/http"
	"os"
	"testing"

	"tinyskin/config"
	"tinyskin/db/cache"
	"tinyskin/db/ent"
	"tinyskin/db/ent/migrate"
	"tinyskin/model"
	"tinyskin/service/auth"
	"tinyskin/service/captcha"
	"tinyskin/service/email"

	_ "github.com/go-sql-driver/mysql"
	"github.com/samber/lo"
)

var (
	userSerice  *UserService
	adminSerice *AdminService
)

func TestMain(m *testing.M) {
	ctx := context.Background()

	clean := initSerice(ctx)
	code := m.Run()

	clean()

	os.Exit(code)
}

func initSerice(ctx context.Context) func() {
	c := lo.Must(ent.Open("mysql", "root:root@tcp(127.0.0.1)/test"))
	lo.Must0(c.Schema.Create(context.Background(), migrate.WithForeignKeys(false), migrate.WithDropIndex(true), migrate.WithDropColumn(true)))
	rsa4 := lo.Must(rsa.GenerateKey(rand.Reader, 4096))
	cache := cache.NewFastCache(100000)
	config := config.Default()
	authService := auth.NewAuthService(c, cache, &rsa4.PublicKey, rsa4)
	email := lo.Must(email.NewEmail(rsa4, config, cache))

	userSerice = NewUserSerice(config, c, captcha.NewCaptchaService(config, &http.Client{}), authService, cache, email)
	adminSerice = NewAdminService(authService, c, config, cache)

	return func() {
		c.User.Delete().Exec(ctx)
		c.Texture.Delete().Exec(ctx)
		c.UserProfile.Delete().Exec(ctx)
		c.UserTexture.Delete().Exec(ctx)
		c.UserToken.Delete().Exec(ctx)
	}
}

func TestUserSerice_Reg(t *testing.T) {
	ctx := context.Background()
	userSerice.config.MaxIpUser = 1
	type args struct {
		ctx      context.Context
		u        model.UserReg
		ipPrefix string
		ip       string
	}
	tests := []struct {
		name    string
		w       *UserService
		args    args
		wantErr bool
	}{
		{
			name: "1",
			w:    userSerice,
			args: args{
				ctx: ctx,
				u: model.UserReg{
					Email:        "1@xmdhs.com",
					Password:     "123456",
					Name:         "111",
					CaptchaToken: "",
				},
				ipPrefix: "127.0.0.0/24",
				ip:       "127.0.0.1",
			},
			wantErr: false,
		},
		{
			name: "email duplicate",
			w:    userSerice,
			args: args{
				ctx: ctx,
				u: model.UserReg{
					Email:        "1@xmdhs.com",
					Password:     "123456",
					Name:         "111",
					CaptchaToken: "",
				},
				ipPrefix: "127.0.0.0/24",
				ip:       "127.0.0.1",
			},
			wantErr: true,
		},
		{
			name: "name duplicate",
			w:    userSerice,
			args: args{
				ctx: ctx,
				u: model.UserReg{
					Email:        "2@xmdhs.com",
					Password:     "123456",
					Name:         "111",
					CaptchaToken: "",
				},
				ipPrefix: "127.0.0.0/24",
				ip:       "127.0.0.1",
			},
			wantErr: true,
		},
		{
			name: "MaxIpUser",
			w:    userSerice,
			args: args{
				ctx: ctx,
				u: model.UserReg{
					Email:        "3@xmdhs.com",
					Password:     "123456",
					Name:         "333",
					CaptchaToken: "",
				},
				ipPrefix: "127.0.0.0/24",
				ip:       "127.0.0.1",
			},
			wantErr: true,
		},
		{
			name: "MaxIpUser",
			w:    userSerice,
			args: args{
				ctx: ctx,
				u: model.UserReg{
					Email:        "4@xmdhs.com",
					Password:     "123456",
					Name:         "444",
					CaptchaToken: "",
				},
				ipPrefix: "127.0.0.2/24",
				ip:       "127.0.0.1",
			},
			wantErr: false,
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if _, err := tt.w.Reg(tt.args.ctx, tt.args.u, tt.args.ipPrefix, tt.args.ip); (err != nil) != tt.wantErr {
				t.Errorf("WebService.Reg() error = %v, wantErr %v", err, tt.wantErr)
			}
		})
	}
	userSerice.config.MaxIpUser = 0
}