snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/codegen/golang/result_test.go

@@ -0,0 +1,78 @@
+package golang
+
+import (
+	"testing"
+
+	"github.com/sqlc-dev/sqlc/internal/metadata"
+	"github.com/sqlc-dev/sqlc/internal/plugin"
+)
+
+func TestPutOutColumns_ForZeroColumns(t *testing.T) {
+	tests := []struct {
+		cmd  string
+		want bool
+	}{
+		{
+			cmd:  metadata.CmdExec,
+			want: false,
+		},
+		{
+			cmd:  metadata.CmdExecResult,
+			want: false,
+		},
+		{
+			cmd:  metadata.CmdExecRows,
+			want: false,
+		},
+		{
+			cmd:  metadata.CmdExecLastId,
+			want: false,
+		},
+		{
+			cmd:  metadata.CmdMany,
+			want: true,
+		},
+		{
+			cmd:  metadata.CmdOne,
+			want: true,
+		},
+		{
+			cmd:  metadata.CmdCopyFrom,
+			want: false,
+		},
+		{
+			cmd:  metadata.CmdBatchExec,
+			want: false,
+		},
+		{
+			cmd:  metadata.CmdBatchMany,
+			want: true,
+		},
+		{
+			cmd:  metadata.CmdBatchOne,
+			want: true,
+		},
+	}
+	for _, tc := range tests {
+		t.Run(tc.cmd, func(t *testing.T) {
+			query := &plugin.Query{
+				Cmd:     tc.cmd,
+				Columns: []*plugin.Column{},
+			}
+			got := putOutColumns(query)
+			if got != tc.want {
+				t.Errorf("putOutColumns failed. want %v, got %v", tc.want, got)
+			}
+		})
+	}
+}
+
+func TestPutOutColumns_AlwaysTrueWhenQueryHasColumns(t *testing.T) {
+	query := &plugin.Query{
+		Cmd:     metadata.CmdMany,
+		Columns: []*plugin.Column{{}},
+	}
+	if putOutColumns(query) != true {
+		t.Error("should be true when we have columns")
+	}
+}