EdgeAPI/internal/db/models/ip_list_dao_test.go

package models

import (
	"errors"
	"runtime"
	"testing"

	_ "github.com/go-sql-driver/mysql"
	"github.com/iwind/TeaGo/dbs"
)

func TestIPListDAO_IncreaseVersion(t *testing.T) {
	dbs.NotifyReady()

	var tx *dbs.Tx

	dao := NewIPListDAO()
	version, err := dao.IncreaseVersion(tx)
	if err != nil {
		t.Fatal(err)
	}
	t.Log("version:", version)
}

func TestIPListDAO_CheckUserIPList(t *testing.T) {
	dbs.NotifyReady()

	var tx *dbs.Tx

	{
		err := NewIPListDAO().CheckUserIPList(tx, 1, 100)
		if err != nil && errors.Is(err, ErrNotFound) {
			t.Log("not found")
		} else {
			t.Log(err)
		}
	}

	{
		err := NewIPListDAO().CheckUserIPList(tx, 1, 85)
		if err != nil && errors.Is(err, ErrNotFound) {
			t.Log("not found")
		} else {
			t.Log(err)
		}
	}

	{
		err := NewIPListDAO().CheckUserIPList(tx, 1, 17)
		if err != nil && errors.Is(err, ErrNotFound) {
			t.Log("not found")
		} else {
			t.Log(err)
		}
	}
}

func TestIPListDAO_NotifyUpdate(t *testing.T) {
	dbs.NotifyReady()

	var dao = NewIPListDAO()
	var tx *dbs.Tx
	err := dao.NotifyUpdate(tx, 104, NodeTaskTypeIPListDeleted)
	if err != nil {
		t.Fatal(err)
	}
}

func BenchmarkIPListDAO_IncreaseVersion(b *testing.B) {
	runtime.GOMAXPROCS(1)

	dbs.NotifyReady()

	var tx *dbs.Tx

	dao := NewIPListDAO()
	for i := 0; i < b.N; i++ {
		_, _ = dao.IncreaseVersion(tx)
	}
}