Files
s3/weed/query/engine/select_test.go
zooqueen dd74d3e6d6 Replace AGPL MinIO fork with Apache-2.0 SeaweedFS fork; rebrand CLI weed->s3
hanzoai/s3 was an AGPL-3.0 MinIO fork whose network-copyleft blocks embedding
and reselling as a managed service. Replace it wholesale with a fork of
SeaweedFS (github.com/seaweedfs/seaweedfs, Apache-2.0, Go) which ships a
built-in S3 gateway + Filer.

- License: AGPL-3.0 -> Apache-2.0 (upstream LICENSE kept verbatim, incl.
  "Copyright 2025 Chris Lu"); NOTICE added crediting SeaweedFS. No AGPL/BUSL/
  SSPL/GPL or enterprise ("ee") files remain.
- Module: go.mod path set to github.com/hanzoai/s3 with a self-replace
  (replace github.com/seaweedfs/seaweedfs => ./) so the ~1800 internal Go files
  keep their upstream import paths with zero churn. Internal identifiers and
  import paths are intentionally NOT renamed.
- CLI rebrand: the binary is built as `s3` (was `weed`); root usage, per-command
  help/usage/flag strings, shell autocomplete program name, and the product name
  in CLI output now read "Hanzo S3" / `s3`. Users never see `weed`.
  (e.g. `s3 server`, `s3 filer`, `s3 master`, `s3 volume`, `s3 s3`, `s3 mount`.)
- Build: `make` (or `go build -o s3 ./weed`) builds the `s3` binary green;
  `cmd/weed-db` and `cmd/weed-sql` also build; `go vet ./weed/command/` clean.
- README/LLM.md rewritten for Hanzo S3; AGENTS.md/CLAUDE.md symlinks preserved.
- Dropped a 33MB prebuilt Mach-O `stress` binary upstream had committed; `.gitignore`
  updated to ignore the `s3` build output.
2026-06-18 23:41:25 -07:00

214 lines
6.1 KiB
Go

package engine
import (
"context"
"fmt"
"strings"
"testing"
)
func TestSQLEngine_SelectBasic(t *testing.T) {
engine := NewTestSQLEngine()
// Test SELECT * FROM table
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events")
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if result.Error != nil {
t.Fatalf("Expected no query error, got %v", result.Error)
}
if len(result.Columns) == 0 {
t.Error("Expected columns in result")
}
if len(result.Rows) == 0 {
t.Error("Expected rows in result")
}
// Should have sample data with 4 columns (SELECT * excludes system columns)
expectedColumns := []string{"id", "user_id", "event_type", "data"}
if len(result.Columns) != len(expectedColumns) {
t.Errorf("Expected %d columns, got %d", len(expectedColumns), len(result.Columns))
}
// In mock environment, only live_log data from unflushed messages
// parquet_archive data would come from parquet files in a real system
if len(result.Rows) == 0 {
t.Error("Expected rows in result")
}
}
func TestSQLEngine_SelectWithLimit(t *testing.T) {
engine := NewTestSQLEngine()
// Test SELECT with LIMIT
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 2")
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if result.Error != nil {
t.Fatalf("Expected no query error, got %v", result.Error)
}
// Should have exactly 2 rows due to LIMIT
if len(result.Rows) != 2 {
t.Errorf("Expected 2 rows with LIMIT 2, got %d", len(result.Rows))
}
}
func TestSQLEngine_SelectSpecificColumns(t *testing.T) {
engine := NewTestSQLEngine()
// Test SELECT specific columns (this will fall back to sample data)
result, err := engine.ExecuteSQL(context.Background(), "SELECT user_id, event_type FROM user_events")
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if result.Error != nil {
t.Fatalf("Expected no query error, got %v", result.Error)
}
// Should have all columns for now (sample data doesn't implement projection yet)
if len(result.Columns) == 0 {
t.Error("Expected columns in result")
}
}
func TestSQLEngine_SelectFromNonExistentTable(t *testing.T) {
t.Skip("Skipping non-existent table test - table name parsing issue needs investigation")
engine := NewTestSQLEngine()
// Test SELECT from non-existent table
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM nonexistent_table")
t.Logf("ExecuteSQL returned: err=%v, result.Error=%v", err, result.Error)
if result.Error == nil {
t.Error("Expected error for non-existent table")
return
}
if !strings.Contains(result.Error.Error(), "not found") {
t.Errorf("Expected 'not found' error, got: %v", result.Error)
}
}
func TestSQLEngine_SelectWithOffset(t *testing.T) {
engine := NewTestSQLEngine()
// Test SELECT with OFFSET only
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 10 OFFSET 1")
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if result.Error != nil {
t.Fatalf("Expected no query error, got %v", result.Error)
}
// Should have fewer rows than total since we skip 1 row
// Sample data has 10 rows, so OFFSET 1 should give us 9 rows
if len(result.Rows) != 9 {
t.Errorf("Expected 9 rows with OFFSET 1 (10 total - 1 offset), got %d", len(result.Rows))
}
}
func TestSQLEngine_SelectWithLimitAndOffset(t *testing.T) {
engine := NewTestSQLEngine()
// Test SELECT with both LIMIT and OFFSET
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 2 OFFSET 1")
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if result.Error != nil {
t.Fatalf("Expected no query error, got %v", result.Error)
}
// Should have exactly 2 rows (skip 1, take 2)
if len(result.Rows) != 2 {
t.Errorf("Expected 2 rows with LIMIT 2 OFFSET 1, got %d", len(result.Rows))
}
}
func TestSQLEngine_SelectWithOffsetExceedsRows(t *testing.T) {
engine := NewTestSQLEngine()
// Test OFFSET that exceeds available rows
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 10 OFFSET 10")
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if result.Error != nil {
t.Fatalf("Expected no query error, got %v", result.Error)
}
// Should have 0 rows since offset exceeds available data
if len(result.Rows) != 0 {
t.Errorf("Expected 0 rows with large OFFSET, got %d", len(result.Rows))
}
}
func TestSQLEngine_SelectWithOffsetZero(t *testing.T) {
engine := NewTestSQLEngine()
// Test OFFSET 0 (should be same as no offset)
result1, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 3")
if err != nil {
t.Fatalf("Expected no error for LIMIT query, got %v", err)
}
result2, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 3 OFFSET 0")
if err != nil {
t.Fatalf("Expected no error for LIMIT OFFSET query, got %v", err)
}
if result1.Error != nil {
t.Fatalf("Expected no query error for LIMIT, got %v", result1.Error)
}
if result2.Error != nil {
t.Fatalf("Expected no query error for LIMIT OFFSET, got %v", result2.Error)
}
// Both should return the same number of rows
if len(result1.Rows) != len(result2.Rows) {
t.Errorf("LIMIT 3 and LIMIT 3 OFFSET 0 should return same number of rows. Got %d vs %d", len(result1.Rows), len(result2.Rows))
}
}
func TestSQLEngine_SelectDifferentTables(t *testing.T) {
engine := NewTestSQLEngine()
// Test different sample tables
tables := []string{"user_events", "system_logs"}
for _, tableName := range tables {
result, err := engine.ExecuteSQL(context.Background(), fmt.Sprintf("SELECT * FROM %s", tableName))
if err != nil {
t.Errorf("Error querying table %s: %v", tableName, err)
continue
}
if result.Error != nil {
t.Errorf("Query error for table %s: %v", tableName, result.Error)
continue
}
if len(result.Columns) == 0 {
t.Errorf("No columns returned for table %s", tableName)
}
if len(result.Rows) == 0 {
t.Errorf("No rows returned for table %s", tableName)
}
t.Logf("Table %s: %d columns, %d rows", tableName, len(result.Columns), len(result.Rows))
}
}