Function TestWebSocketGoldenFiles

http/codegen/websocket_golden_test.go:29–116

TestWebSocketGoldenFiles tests WebSocket code generation against golden files to ensure comprehensive coverage of all WebSocket templates and scenarios.

(t *testing.T)

// TestWebSocketGoldenFiles tests WebSocket code generation against golden files
// to ensure comprehensive coverage of all WebSocket templates and scenarios.
func TestWebSocketGoldenFiles(t *testing.T) {
	cases := []struct {
		name     string
		dsl      func()
		fileType string // "server" or "client"
	}{
		// Server Streaming (Result only)
		{"websocket-server-streaming-primitive", serverStreamingPrimitiveDSL, "server"},
		{"websocket-server-streaming-array", serverStreamingArrayDSL, "server"},
		{"websocket-server-streaming-object", serverStreamingObjectDSL, "server"},
		{"websocket-server-streaming-user-type", serverStreamingUserTypeDSL, "server"},
		{"websocket-server-streaming-with-views", serverStreamingWithViewsDSL, "server"},

		// Client Streaming (Payload only)
		{"websocket-client-streaming-primitive", clientStreamingPrimitiveDSL, "client"},
		{"websocket-client-streaming-array", clientStreamingArrayDSL, "client"},
		{"websocket-client-streaming-object", clientStreamingObjectDSL, "client"},
		{"websocket-client-streaming-user-type", clientStreamingUserTypeDSL, "client"},
		{"websocket-client-streaming-with-validation", clientStreamingWithValidationDSL, "client"},

		// Bidirectional Streaming
		{"websocket-bidirectional-streaming-primitive", bidirectionalStreamingPrimitiveDSL, "server"},
		{"websocket-bidirectional-streaming-complex", bidirectionalStreamingComplexDSL, "server"},
		{"websocket-bidirectional-streaming-with-views", bidirectionalStreamingWithViewsDSL, "server"},

		// Client-side Bidirectional
		{"websocket-bidirectional-streaming-primitive-client", bidirectionalStreamingPrimitiveDSL, "client"},
		{"websocket-bidirectional-streaming-complex-client", bidirectionalStreamingComplexDSL, "client"},
		{"websocket-bidirectional-streaming-with-views-client", bidirectionalStreamingWithViewsDSL, "client"},

		// Edge Cases
		{"websocket-no-payload-streaming", noPayloadStreamingDSL, "server"},
		{"websocket-no-result-streaming", noResultStreamingDSL, "server"},
		{"websocket-mixed-endpoints", mixedEndpointsDSL, "server"},
		{"websocket-mixed-endpoints-client", mixedEndpointsDSL, "client"},

		// Template Coverage Tests
		{"websocket-conn-configurer", connConfigurerDSL, "server"},
		{"websocket-conn-configurer-client", connConfigurerDSL, "client"},
		{"websocket-struct-types", structTypesDSL, "server"},
		{"websocket-struct-types-client", structTypesDSL, "client"},
	}

	for _, c := range cases {
		t.Run(c.name, func(t *testing.T) {
			root := RunHTTPDSL(t, c.dsl)
			services := CreateHTTPServices(root)

			var files []*codegen.File
			if c.fileType == "server" {
				files = ServerFiles("", services)
			} else {
				files = ClientFiles("", services)
			}

			// Find the websocket.go file
			var wsFile *codegen.File
			for _, f := range files {