Function TestParseCopyLine

server/conn_test.go:2809–2884

(t *testing.T)

	copySQL := BuildDuckDBCopyFromSQL("copy_large_record_test", "", path, &CopyFromOptions{
		Delimiter:  "\t",
		NullString: "\\N",
	})
	if _, err = db.Exec(copySQL); err != nil {
		t.Fatalf("COPY 11 MiB record: %v", err)
	}

	var storedSize int64
	if err = db.QueryRow("SELECT length(payload) FROM copy_large_record_test").Scan(&storedSize); err != nil {
		t.Fatalf("query copied payload size: %v", err)
	}
	if storedSize != payloadSize {
		t.Fatalf("copied payload size = %d, want %d", storedSize, payloadSize)
	}
}

func TestCopyCommandTypeDetection(t *testing.T) {
	c := &clientConn{}

	tests := []struct {
		name     string
		query    string
		expected string
	}{
		{
			name:     "COPY TO STDOUT",
			query:    "COPY users TO STDOUT",
			expected: "COPY",
		},
		{
			name:     "COPY FROM STDIN",
			query:    "COPY users FROM STDIN",
			expected: "COPY",
		},
		{
			name:     "COPY with options",
			query:    "COPY users FROM STDIN WITH (FORMAT CSV)",
			expected: "COPY",
		},
		{
			name:     "lowercase copy",
			query:    "copy users to stdout",
			expected: "COPY",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := c.getCommandType(strings.ToUpper(tt.query))
			if result != tt.expected {
				t.Errorf("getCommandType(%q) = %q, want %q", tt.query, result, tt.expected)
			}
		})
	}
}

func TestParseCopyLine(t *testing.T) {
	c := &clientConn{}