TestParser_DisplayTokens tests token display
(t *testing.T)
| 204 | |
| 205 | // TestParser_DisplayTokens tests token display |
| 206 | func TestParser_DisplayTokens(t *testing.T) { |
| 207 | var outBuf, errBuf bytes.Buffer |
| 208 | parser := NewParser(&outBuf, &errBuf, CLIParserOptions{ |
| 209 | Format: "table", |
| 210 | ShowTokens: true, |
| 211 | }) |
| 212 | |
| 213 | result, err := parser.Parse("SELECT * FROM users") |
| 214 | if err != nil { |
| 215 | t.Fatalf("Unexpected error: %v", err) |
| 216 | } |
| 217 | |
| 218 | err = parser.Display(result) |
| 219 | if err != nil { |
| 220 | t.Fatalf("Failed to display tokens: %v", err) |
| 221 | } |
| 222 | |
| 223 | output := outBuf.String() |
| 224 | if output == "" { |
| 225 | t.Error("Expected token output but got empty string") |
| 226 | } |
| 227 | |
| 228 | // Check for token output format |
| 229 | if !strings.Contains(output, "Tokens") { |
| 230 | t.Error("Expected 'Tokens' in output") |
| 231 | } |
| 232 | } |
| 233 | |
| 234 | // TestParser_DisplayTree tests tree visualization |
| 235 | func TestParser_DisplayTree(t *testing.T) { |