utils/network/udp_test.go

package network

import (
	"bytes"
	"strings"
	"testing"
	"time"
)

const (
	testUDPServerAddress           = "127.0.0.1:10060"
	testUDPServerTimeout           = time.Second
	testUDPServerReceiveBufferSize = 1024
)

func TestUDP(t *testing.T) {
	server := &UDPServer{}

	err := server.Connect(testUDPServerAddress, NewUDPServerOptions(
		WithUDPServerReadTimeout(testUDPServerTimeout),
		WithUDPServerWriteTimeout(testUDPServerTimeout),
		WithUDPServerReceiveBufferSize(testUDPServerReceiveBufferSize),
		WithUDPServerRequestCallback(func(data []byte) ([]byte, error) {
			requestBuffer := bytes.NewReader(data)
			requestDataReader := NewDataReader(requestBuffer)
			requestBytes, err := requestDataReader.Bytes(len(data))
			if err != nil {
				return nil, err
			}

			responseBytes := []byte(strings.ToUpper(string(requestBytes)))
			responseBuffer := &bytes.Buffer{}
			responseReader := NewDataWriter(responseBuffer)
			err = responseReader.Bytes(responseBytes)
			if err != nil {
				return nil, err
			}

			return responseBuffer.Bytes(), nil
		}),
	))
	if err != nil {
		t.Fatal(err)
	}

	defer server.Disconnect()

	// TODO 完成客户端后补充
}