toki
db4ee8f78e
- Add server_test.go for socket server testing - Update socket-session-loop.md roadmap milestone - Refactor services/api/internal/socket/server.go - Add agent-task documentation for socket session loop
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package socket
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"testing"
|
|
"time"
|
|
|
|
altv1 "git.toki-labs.com/toki/alt/packages/contracts/gen/go/alt/v1"
|
|
"git.toki-labs.com/toki/alt/services/api/internal/config"
|
|
apiContracts "git.toki-labs.com/toki/alt/services/api/internal/contracts"
|
|
protoSocket "git.toki-labs.com/toki/proto-socket/go"
|
|
)
|
|
|
|
func TestServerRespondsToHelloRequest(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
defer cancel()
|
|
|
|
cfg := config.Config{
|
|
Host: "127.0.0.1",
|
|
Port: freeTCPPort(t),
|
|
SocketPath: "/socket",
|
|
HeartbeatIntervalSec: 0,
|
|
HeartbeatWaitSec: 0,
|
|
}
|
|
server := NewServer(cfg)
|
|
if err := server.Start(ctx); err != nil {
|
|
t.Fatalf("failed to start server: %v", err)
|
|
}
|
|
defer server.Stop()
|
|
|
|
client, err := protoSocket.DialWsWithHeartbeat(ctx, cfg.Host, cfg.Port, cfg.SocketPath, 0, 0, apiContracts.ParserMap())
|
|
if err != nil {
|
|
t.Fatalf("failed to dial server: %v", err)
|
|
}
|
|
defer client.Close()
|
|
|
|
res, err := protoSocket.SendRequestTyped[*altv1.HelloRequest, *altv1.HelloResponse](
|
|
&client.Communicator,
|
|
&altv1.HelloRequest{
|
|
ClientName: "alt-test",
|
|
ClientVersion: "test",
|
|
AltProtocolVersion: "alt.v1",
|
|
},
|
|
2*time.Second,
|
|
)
|
|
if err != nil {
|
|
t.Fatalf("failed to send hello request: %v", err)
|
|
}
|
|
|
|
if res.GetServerName() != serverName {
|
|
t.Errorf("server name mismatch: expected %q, got %q", serverName, res.GetServerName())
|
|
}
|
|
if res.GetServerVersion() != serverVersion {
|
|
t.Errorf("server version mismatch: expected %q, got %q", serverVersion, res.GetServerVersion())
|
|
}
|
|
if res.GetAltProtocolVersion() != "alt.v1" {
|
|
t.Errorf("protocol version mismatch: expected %q, got %q", "alt.v1", res.GetAltProtocolVersion())
|
|
}
|
|
if len(res.GetCapabilities()) == 0 {
|
|
t.Fatal("expected capabilities to be set")
|
|
}
|
|
}
|
|
|
|
func freeTCPPort(t *testing.T) int {
|
|
t.Helper()
|
|
|
|
listener, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
t.Fatalf("failed to reserve TCP port: %v", err)
|
|
}
|
|
defer listener.Close()
|
|
|
|
addr, ok := listener.Addr().(*net.TCPAddr)
|
|
if !ok {
|
|
t.Fatalf("unexpected listener address type %T", listener.Addr())
|
|
}
|
|
return addr.Port
|
|
}
|