(t *testing.T)
| 74 | } |
| 75 | |
| 76 | func TestSplitContextCancellation(t *testing.T) { |
| 77 | ctx, cancel := context.WithCancel(context.Background()) |
| 78 | cancel() // Cancel immediately |
| 79 | |
| 80 | reader := strings.NewReader("SELECT * FROM users;") |
| 81 | _, err := Split(ctx, reader) |
| 82 | |
| 83 | if err != context.Canceled { |
| 84 | t.Errorf("Expected context.Canceled error, got %v", err) |
| 85 | } |
| 86 | } |
| 87 | |
| 88 | func TestExtractDollarTag(t *testing.T) { |
| 89 | tests := []struct { |