connect/server/server_test.go

package server

import (
	"fmt"
	"log"
	"testing"

	"github.com/google/uuid"
	"github.com/gorilla/websocket"
	"github.com/spf13/viper"
	"sikey.com/websocket/utils/keys"
)

func BenchmarkTest1(b *testing.B) {
	fmt.Println(b.N)
	for i := 0; i < b.N; i++ {
		_ = fmt.Sprintf("%d", i)
	}
}

func BenchmarkServer_WebsocketPressure(b *testing.B) {
	b.ReportAllocs()
	b.ResetTimer()

	b.SetParallelism(10)
	fmt.Println(b.N)
	b.RunParallel(func(p *testing.PB) {
		for p.Next() {
			websocketConnect()
		}
	})
}

func websocketConnect() {
	conn, _, err := websocket.DefaultDialer.Dial(
		fmt.Sprintf(`ws://127.0.0.1:10082/websocket/endpoint?X-Websocket-Header-ID=%s`, uuid.NewString()),
		// fmt.Sprintf(`ws://106.75.230.4:10082/websocket/endpoint?X-Websocket-Header-ID=%s`, uuid.NewString()),
		nil)
	if err != nil {
		log.Println(err)
	}
	defer conn.Close()

	// message := Message{Type: MessageTypePingPong, RequestId: strconv.Itoa(int(time.Now().UnixMilli())), Content: "ping"}
	// conn.SetWriteDeadline(time.Now().Add(time.Second * 60))
	// err = conn.WriteMessage(websocket.TextMessage, serializationMessage(&message))
	// if err != nil {
	// 	log.Println(err)
	// }

	// go func() {
	// 	for {
	// 		//conn.SetReadDeadline(time.Now().Add(time.Second * 2))
	// 		_, _, err = conn.ReadMessage()
	// 		if err != nil {
	// 			log.Println(err)
	// 			return
	// 		}
	// 		//conn.SetReadDeadline(time.Now().Add(time.Second * 2))
	// 		//log.Println(string(msg))
	// 	}
	// }()
}

func TestServer_JWT(t *testing.T) {
	// config.MustLoadConfig("../etc/websocket.release.yaml")
	viper.SetDefault("auth.secret", "a4176e95ad5d4d05b012b81dd5833b40")
	var headers = make(Headers)
	headers[keys.LocalizationHeader] = ""
	headers[keys.AccessTokenHeader] = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1aWQiOiJmOTU1NGE3OS03N2JiLTRkNTItYmYxMy01MDMyZTg2OTlmNDgiLCJpc19jaGlsZCI6dHJ1ZSwiZmFjdG9yIjoiMzEyNjg3NTMtODU1ZC00NzBmLWFmYzEtZmZhZjk0ZDJmMjJkIn0.HhCi-HpFSVy8UTL7LNUZyrZB25XiDtpu-Nl9-mM2rs0"
	headers[keys.SimpleHeader] = ""
	s, ok, err := jwtParse(headers)
	if !ok {
		log.Fatalln(err)
		return
	}
	log.Println(s)
}