sendqueue_test.go

package zsocket

import (
	"context"
	"fmt"
	"sync"
	"testing"
	"time"
)

func TestSendQueueConcurrent(t *testing.T) {
	client, server := testPair(t)
	client.EnableSendQueue(32)

	const n = 100
	errCh := make(chan error, 1)
	go func() {
		for i := 0; i < n; i++ {
			_, msg, err := server.Read(context.Background())
			if err != nil {
				errCh <- err
				return
			}
			if len(msg) == 0 {
				errCh <- fmt.Errorf("empty message")
				return
			}
		}
		errCh <- nil
	}()

	var wg sync.WaitGroup
	for i := 0; i < n; i++ {
		wg.Add(1)
		go func(i int) {
			defer wg.Done()
			ctx, cancel := context.WithTimeout(context.Background(), time.Second)
			defer cancel()
			_ = client.Send(ctx, TextMessage, []byte(fmt.Sprintf("m-%d", i)))
		}(i)
	}
	wg.Wait()

	if err := <-errCh; err != nil {
		t.Fatalf("server read: %v", err)
	}
}