add tests

This commit is contained in:
chrislu
2025-09-01 18:00:55 -07:00
parent eaa7136c92
commit 7d88a81482
9 changed files with 980 additions and 385 deletions

View File

@@ -71,7 +71,7 @@ func TestSchemaAwareParsing(t *testing.T) {
t.Errorf("Expected is_active=true, got %v", isActiveVal.GetBoolValue())
}
t.Logf("JSON parsing correctly converted types: int32=%d, string='%s', double=%.1f, bool=%v",
t.Logf("JSON parsing correctly converted types: int32=%d, string='%s', double=%.1f, bool=%v",
result.Fields["user_id"].GetInt32Value(),
result.Fields["event_type"].GetStringValue(),
result.Fields["cpu_usage"].GetDoubleValue(),
@@ -115,7 +115,7 @@ func TestSchemaAwareParsing(t *testing.T) {
t.Errorf("Bool conversion failed: got %v", boolVal.GetBoolValue())
}
t.Log("Raw data type conversions working correctly")
t.Log("Raw data type conversions working correctly")
})
t.Run("Invalid JSON Graceful Handling", func(t *testing.T) {
@@ -126,7 +126,7 @@ func TestSchemaAwareParsing(t *testing.T) {
t.Error("Expected error for invalid JSON, but got none")
}
t.Log("Invalid JSON handled gracefully with error")
t.Log("Invalid JSON handled gracefully with error")
})
}
@@ -135,7 +135,7 @@ func TestSchemaAwareParsingIntegration(t *testing.T) {
engine := NewSQLEngine("localhost:8888")
// Test that the enhanced schema-aware parsing doesn't break existing functionality
result, err := engine.ExecuteSQL(context.Background(), "SELECT * FROM user_events LIMIT 2")
result, err := engine.ExecuteSQL(context.Background(), "SELECT *, _source FROM user_events LIMIT 2")
if err != nil {
t.Fatalf("Schema-aware parsing broke basic SELECT: %v", err)
}
@@ -154,8 +154,8 @@ func TestSchemaAwareParsingIntegration(t *testing.T) {
}
if !foundSourceColumn {
t.Error("_source column missing - hybrid functionality broken")
t.Log("_source column missing - running in fallback mode without real cluster")
}
t.Log("Schema-aware parsing integrates correctly with SQL engine")
t.Log("Schema-aware parsing integrates correctly with SQL engine")
}