package notify

import (
	"context"
	"errors"
	"testing"
	"time"
)

func TestBulkOpenDedicatedUDPRejected(t *testing.T) {
	server := NewServer().(*ServerCommon)
	if err := UseModernPSKServer(server, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
		t.Fatalf("UseModernPSKServer failed: %v", err)
	}
	server.SetBulkHandler(func(info BulkAcceptInfo) error {
		return nil
	})
	if err := server.Listen("udp", "127.0.0.1:0"); err != nil {
		t.Fatalf("server Listen failed: %v", err)
	}
	defer func() {
		_ = server.Stop()
	}()

	client := NewClient().(*ClientCommon)
	if err := UseModernPSKClient(client, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
		t.Fatalf("UseModernPSKClient failed: %v", err)
	}
	if err := client.Connect("udp", signalRoundTripServerAddr(server, "")); err != nil {
		t.Fatalf("client Connect failed: %v", err)
	}
	defer func() {
		_ = client.Stop()
	}()

	_, err := client.OpenBulk(context.Background(), BulkOpenOptions{
		Range: BulkRange{
			Offset: 0,
			Length: 128,
		},
		Dedicated: true,
	})
	if !errors.Is(err, errBulkDedicatedStreamOnly) {
		t.Fatalf("client OpenBulk dedicated over udp error = %v, want %v", err, errBulkDedicatedStreamOnly)
	}
}

func TestServerOpenBulkLogicalDedicatedUDPRejected(t *testing.T) {
	server := NewServer().(*ServerCommon)
	if err := UseModernPSKServer(server, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
		t.Fatalf("UseModernPSKServer failed: %v", err)
	}
	server.SetBulkHandler(func(info BulkAcceptInfo) error {
		return nil
	})
	if err := server.Listen("udp", "127.0.0.1:0"); err != nil {
		t.Fatalf("server Listen failed: %v", err)
	}
	defer func() {
		_ = server.Stop()
	}()

	client := NewClient().(*ClientCommon)
	if err := UseModernPSKClient(client, integrationSharedSecret, integrationModernPSKOptions()); err != nil {
		t.Fatalf("UseModernPSKClient failed: %v", err)
	}
	if err := client.Connect("udp", signalRoundTripServerAddr(server, "")); err != nil {
		t.Fatalf("client Connect failed: %v", err)
	}
	defer func() {
		_ = client.Stop()
	}()

	logical := waitForTransferControlLogicalConn(t, server, 2*time.Second)
	_, err := server.OpenBulkLogical(context.Background(), logical, BulkOpenOptions{
		Range: BulkRange{
			Offset: 0,
			Length: 128,
		},
		Dedicated: true,
	})
	if !errors.Is(err, errBulkDedicatedStreamOnly) {
		t.Fatalf("server OpenBulkLogical dedicated over udp error = %v, want %v", err, errBulkDedicatedStreamOnly)
	}
}