trojan-go/stat/db_test.go

package stat

import (
	"context"
	"database/sql"
	"fmt"
	"math/rand"
	"strings"
	"testing"
	"time"

	_ "github.com/go-sql-driver/mysql"
	"github.com/p4gefau1t/trojan-go/common"
	"github.com/p4gefau1t/trojan-go/conf"
)

func TestDBTrafficMeter(t *testing.T) {
	userName := "root"
	password := "password"
	ip := "127.0.0.1"
	port := "3306"
	dbName := "trojan"
	path := strings.Join([]string{userName, ":", password, "@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
	db, err := sql.Open("mysql", path)
	hash := common.SHA224String("hashhash")
	common.Must(err)
	defer db.Close()
	c := &DBTrafficMeter{
		db:             db,
		trafficChan:    make(chan *trafficInfo, 1024),
		ctx:            context.Background(),
		updateDuration: time.Second * 5,
	}
	simulation := func() {
		for i := 0; i < 100; i++ {
			c.Count(hash, uint64(rand.Intn(500)), uint64(rand.Intn(500)))
			time.Sleep(time.Duration(int64(time.Millisecond) * rand.Int63n(100)))
		}
		fmt.Println("done")
	}
	for i := 0; i < 100; i++ {
		go simulation()
	}
	go c.dbDaemon()
	time.Sleep(time.Second * 30)
}

func TestDBAuthenticator(t *testing.T) {
	userName := "root"
	password := "password"
	ip := "127.0.0.1"
	port := "3306"
	dbName := "trojan"
	path := strings.Join([]string{userName, ":", password, "@tcp(", ip, ":", port, ")/", dbName, "?charset=utf8"}, "")
	db, err := sql.Open("mysql", path)
	common.Must(err)
	defer db.Close()
	config := conf.GlobalConfig{
		MySQL: conf.MySQLConfig{
			CheckRate: 2,
		},
	}
	a, err := NewDBAuthenticator(&config, db)
	common.Must(err)
	time.Sleep(time.Second * 5)
	hash := common.SHA224String("hashhash")
	fmt.Println(common.SHA224String("hashhash"))
	fmt.Println(a.CheckHash(hash), a.CheckHash("jasdlkflfejlqjef"))
}