udp_client_test.go

package transport

import (
	"net"
	"testing"
	"time"
)

func TestUDPClient(t *testing.T) {

	t.Run("specify", func(t *testing.T) {
		addr, _ := net.ResolveUDPAddr("udp", "0.0.0.0:20001")
		remoteAddr, _ := net.ResolveUDPAddr("udp", "192.168.31.112:40000")

		client := UDPClient{}
		err := client.Connect(addr, remoteAddr)
		if err != nil {
			panic(err)
		}
		client.SetHandler2(nil, func(conn net.Conn, data []byte) []byte {
			println("recv data:" + string(data))
			return nil
		}, nil)

		go client.Receive()

		for {
			msg := "hello!"
			err := client.Write([]byte(msg))
			if err != nil {
				panic(err)
			}

			time.Sleep(time.Second)
		}
	})

	t.Run("random", func(t *testing.T) {
		client := UDPClient{}
		client.SetHandler2(nil, func(conn net.Conn, data []byte) []byte {
			println("recv data:" + string(data))
			return nil
		},
			nil)

		client.Connect(nil, nil)

		go client.Receive()

		remoteAddr, _ := net.ResolveUDPAddr("udp", "192.168.2.148:40000")

		for {
			msg := "hello!"
			err := client.WriteTo([]byte(msg), remoteAddr)
			if err != nil {
				panic(err)
			}

			time.Sleep(time.Second)
		}
	})

}