Function TestStreamingErrorComparison

grpc/codegen/streaming_errors_test.go:117–191

TestStreamingErrorComparison compares error handling between unary and streaming methods to ensure consistency.

(t *testing.T)

// TestStreamingErrorComparison compares error handling between unary and
// streaming methods to ensure consistency.
func TestStreamingErrorComparison(t *testing.T) {
	// DSL with both unary and streaming methods with errors
	dsl := func() {
		var CustomError = Type("CustomError", func() {
			ErrorName("name", String, "error name")
			Attribute("message", String, "error message")
			Required("name", "message")
		})

		Service("MixedService", func() {
			// Unary method with custom error
			Method("UnaryMethod", func() {
				Payload(String)
				Result(String)
				Error("custom_error", CustomError)
				GRPC(func() {
					Response("custom_error", CodeInvalidArgument)
				})
			})

			// Streaming method with same error
			Method("StreamingMethod", func() {
				Payload(String)
				StreamingResult(String)
				Error("custom_error", CustomError)
				GRPC(func() {
					Response("custom_error", CodeInvalidArgument)
				})
			})
		})
	}

	root := RunGRPCDSL(t, dsl)
	services := CreateGRPCServices(root)
	clientfs := ClientFiles("", services)
	require.Greater(t, len(clientfs), 0, "should have client files")

	// Find unary and streaming code in different sections
	var unaryCode, streamCode string

	// For unary, look in client-endpoint-init
	if sections := clientfs[0].Section("client-endpoint-init"); len(sections) > 0 {
		var code strings.Builder
		for _, section := range sections {
			require.NoError(t, section.Write(&code))
		}
		unaryCode = code.String()
	}

	// For streaming, look in client-stream-recv
	if sections := clientfs[0].Section("client-stream-recv"); len(sections) > 0 {
		var code strings.Builder
		for _, section := range sections {
			require.NoError(t, section.Write(&code))
		}
		streamCode = code.String()
	}