Function TestCopyCommandTypeDetection

server/conn_test.go:2769–2807

(t *testing.T)

	}
}

func TestBuildDuckDBCopyFromSQLAllowsLargeRecord(t *testing.T) {
	const payloadSize = 11 << 20

	db, err := sql.Open("duckdb", ":memory:")
	if err != nil {
		t.Fatalf("open duckdb: %v", err)
	}
	defer func() { _ = db.Close() }()

	if _, err = db.Exec("SET memory_limit = '878 MiB'"); err != nil {
		t.Fatalf("set worker memory limit: %v", err)
	}
	if _, err = db.Exec("CREATE TABLE copy_large_record_test (payload TEXT)"); err != nil {
		t.Fatalf("create table: %v", err)
	}

	path := t.TempDir() + "/large-record.tsv"
	payload := strings.Repeat("x", payloadSize)
	if err = os.WriteFile(path, []byte(payload+"\n"), 0o600); err != nil {
		t.Fatalf("write COPY input: %v", err)
	}

	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{}