package test

import (
	"fmt"
	"main/dao/mysql"
	"main/logic"
	"main/models"
	"regexp"
	"testing"

	"github.com/DATA-DOG/go-sqlmock"
	"github.com/jmoiron/sqlx"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestGetAllStudentInfo(t *testing.T) {
	// 创建一个模拟数据库
	db, mock, err := sqlmock.New()
	require.NoError(t, err)
	defer db.Close()

	sqlxDB := sqlx.NewDb(db, "mysql")
	originalDB := mysql.GetDb()
	mysql.ChangeDb(sqlxDB)
	defer func() { mysql.ChangeDb(originalDB) }()

	// 定义期望的查询和结果
	rows := sqlmock.NewRows([]string{"user_id", "user_name", "user_gender", "user_score"}).
		AddRow(1, "Alice", 1, 90).
		AddRow(2, "Bob", 1, 85)

	mock.ExpectQuery("(?i)SELECT user_id, user_name, user_gender, user_score FROM user ORDER BY user_score ASC").
		WillReturnRows(rows)

	// 调用被测试函数
	result, err := mysql.GetAllStudentInfo() // 确保函数路径正确
	require.NoError(t, err)

	// 验证结果
	expected := &[]models.User{
		{User_id: 1, User_name: "Alice", User_gender: 1, User_score: 90},
		{User_id: 2, User_name: "Bob", User_gender: 1, User_score: 85},
	}
	assert.Equal(t, expected, result)

	// 验证所有期望的操作都已执行
	err = mock.ExpectationsWereMet()
	require.NoError(t, err)
}

func TestChangeStuScore(t *testing.T) {
	db, mock, err := sqlmock.New()
	require.NoError(t, err)
	defer db.Close()
	sqlxDB := sqlx.NewDb(db, "mysql")
	originalDB := mysql.GetDb()
	mysql.ChangeDb(sqlxDB)
	defer func() { mysql.ChangeDb(originalDB) }()
	if err != nil {
		fmt.Println(err)
	}
	sqlStr := `update user set
	user_score = ?
	where user_id=?`
	mock.ExpectExec(regexp.QuoteMeta(sqlStr)).
		WithArgs(95, 1).
		WillReturnResult(sqlmock.NewResult(1, 1))
	user := &models.UserScore{
		User_id:    1,
		User_score: 95,
	}
	err = mysql.ChangeStuScore(user)
	require.NoError(t, err)
	err = mock.ExpectationsWereMet()
	require.NoError(t, err)
}

func TestWeightedRandomSelectWithMockRNG(t *testing.T) {
	users := []models.User{
		{User_id: 1, User_score: 0},
		{User_id: 2, User_score: 10},
		{User_id: 3, User_score: 20},
	}
	// 确定种子
	rng := 42
	expectedUser := users[0]
	selectedUser := logic.WeightedRandomSelect(users, int64(rng))
	require.Equal(t, expectedUser, selectedUser)
}