notify/send_state_test.go

package notify

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

func TestClientSendReturnsServiceShutdownWhenNotRunning(t *testing.T) {
	client := NewClient().(*ClientCommon)
	UseLegacySecurityClient(client)

	err := client.Send("notify", []byte("hello"))
	if !errors.Is(err, errServiceShutdown) {
		t.Fatalf("client Send error = %v, want %v", err, errServiceShutdown)
	}
}

func TestServerSendReturnsServiceShutdownWhenNotRunning(t *testing.T) {
	server := NewServer().(*ServerCommon)
	UseLegacySecurityServer(server)

	err := server.Send(nil, "notify", []byte("hello"))
	if !errors.Is(err, errServiceShutdown) {
		t.Fatalf("server Send error = %v, want %v", err, errServiceShutdown)
	}
}

func TestClientSendReturnsTransportDetachedWhenSessionAliveWithoutTransport(t *testing.T) {
	client := NewClient().(*ClientCommon)
	client.markSessionStarted()
	client.clearClientSessionRuntimeTransport()

	err := client.Send("notify", []byte("hello"))
	if !errors.Is(err, errTransportDetached) {
		t.Fatalf("client Send error = %v, want %v", err, errTransportDetached)
	}
}

func TestServerSendReturnsTransportDetachedWhenPeerTransportMissing(t *testing.T) {
	server := NewServer().(*ServerCommon)
	server.markSessionStarted()

	err := server.Send(&ClientConn{}, "notify", []byte("hello"))
	if !errors.Is(err, errTransportDetached) {
		t.Fatalf("server Send error = %v, want %v", err, errTransportDetached)
	}
}

func TestPendingWaitClosedErrorReturnsTransportDetachedWhenServiceStillRunning(t *testing.T) {
	if err := pendingWaitClosedError(nil); !errors.Is(err, errTransportDetached) {
		t.Fatalf("pendingWaitClosedError(nil) = %v, want %v", err, errTransportDetached)
	}
	openCh := make(chan struct{})
	if err := pendingWaitClosedError(openCh); !errors.Is(err, errTransportDetached) {
		t.Fatalf("pendingWaitClosedError(open) = %v, want %v", err, errTransportDetached)
	}
}

func TestPendingWaitClosedErrorReturnsServiceShutdownWhenStopped(t *testing.T) {
	stopCh := make(chan struct{})
	close(stopCh)
	if err := pendingWaitClosedError(stopCh); !errors.Is(err, errServiceShutdown) {
		t.Fatalf("pendingWaitClosedError(stopped) = %v, want %v", err, errServiceShutdown)
	}
}

func TestClientSendCtxReturnsContextCanceled(t *testing.T) {
	client := NewClient().(*ClientCommon)
	secret := []byte("0123456789abcdef0123456789abcdef")
	client.SetSecretKey(secret)
	server := newRunningPeerAttachServerForTest(t, func(server *ServerCommon) {
		server.SetSecretKey(secret)
	})

	left, right := net.Pipe()
	defer right.Close()
	bootstrapPeerAttachConnForTest(t, server, right)
	if err := client.ConnectByConn(left); err != nil {
		t.Fatalf("client ConnectByConn failed: %v", err)
	}
	defer func() {
		client.setByeFromServer(true)
		_ = client.Stop()
	}()

	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	_, err := client.SendCtx(ctx, "ctx-canceled", []byte("payload"))
	if !errors.Is(err, context.Canceled) {
		t.Fatalf("client SendCtx error = %v, want %v", err, context.Canceled)
	}
}

func TestServerSendCtxReturnsContextCanceled(t *testing.T) {
	client := NewClient().(*ClientCommon)
	secret := []byte("0123456789abcdef0123456789abcdef")
	client.SetSecretKey(secret)
	server := newRunningPeerAttachServerForTest(t, func(server *ServerCommon) {
		server.SetSecretKey(secret)
	})

	left, right := net.Pipe()
	defer right.Close()
	bootstrapPeerAttachConnForTest(t, server, right)
	if err := client.ConnectByConn(left); err != nil {
		t.Fatalf("client ConnectByConn failed: %v", err)
	}
	defer func() {
		client.setByeFromServer(true)
		_ = client.Stop()
	}()

	var logical *LogicalConn
	deadline := time.Now().Add(time.Second)
	for time.Now().Before(deadline) {
		logical = server.GetLogicalConn(client.peerIdentity)
		if logical != nil {
			break
		}
		time.Sleep(10 * time.Millisecond)
	}
	if logical == nil {
		t.Fatal("server logical conn not found")
	}

	ctx, cancel := context.WithCancel(context.Background())
	cancel()
	_, err := server.SendCtxLogical(ctx, logical, "ctx-canceled", []byte("payload"))
	if !errors.Is(err, context.Canceled) {
		t.Fatalf("server SendCtxLogical error = %v, want %v", err, context.Canceled)
	}
}