115 lines
3.4 KiB
Go
115 lines
3.4 KiB
Go
package format_test
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"efprojects.com/kitten-ipc/kitcom/internal/tsgo/ast"
|
|
"efprojects.com/kitten-ipc/kitcom/internal/tsgo/core"
|
|
"efprojects.com/kitten-ipc/kitcom/internal/tsgo/format"
|
|
"efprojects.com/kitten-ipc/kitcom/internal/tsgo/parser"
|
|
"efprojects.com/kitten-ipc/kitcom/internal/tsgo/printer"
|
|
"efprojects.com/kitten-ipc/kitcom/internal/tsgo/repo"
|
|
"gotest.tools/v3/assert"
|
|
)
|
|
|
|
func applyBulkEdits(text string, edits []core.TextChange) string {
|
|
b := strings.Builder{}
|
|
b.Grow(len(text))
|
|
lastEnd := 0
|
|
for _, e := range edits {
|
|
start := e.TextRange.Pos()
|
|
if start != lastEnd {
|
|
b.WriteString(text[lastEnd:e.TextRange.Pos()])
|
|
}
|
|
b.WriteString(e.NewText)
|
|
|
|
lastEnd = e.TextRange.End()
|
|
}
|
|
b.WriteString(text[lastEnd:])
|
|
|
|
return b.String()
|
|
}
|
|
|
|
func TestFormat(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
t.Run("format checker.ts", func(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := format.WithFormatCodeSettings(t.Context(), &format.FormatCodeSettings{
|
|
EditorSettings: format.EditorSettings{
|
|
TabSize: 4,
|
|
IndentSize: 4,
|
|
BaseIndentSize: 4,
|
|
NewLineCharacter: "\n",
|
|
ConvertTabsToSpaces: true,
|
|
IndentStyle: format.IndentStyleSmart,
|
|
TrimTrailingWhitespace: true,
|
|
},
|
|
InsertSpaceBeforeTypeAnnotation: core.TSTrue,
|
|
}, "\n")
|
|
repo.SkipIfNoTypeScriptSubmodule(t)
|
|
filePath := filepath.Join(repo.TypeScriptSubmodulePath, "src/compiler/checker.ts")
|
|
fileContent, err := os.ReadFile(filePath)
|
|
assert.NilError(t, err)
|
|
text := string(fileContent)
|
|
sourceFile := parser.ParseSourceFile(ast.SourceFileParseOptions{
|
|
FileName: "/checker.ts",
|
|
Path: "/checker.ts",
|
|
}, text, core.ScriptKindTS)
|
|
edits := format.FormatDocument(ctx, sourceFile)
|
|
newText := applyBulkEdits(text, edits)
|
|
assert.Assert(t, len(newText) > 0)
|
|
assert.Assert(t, text != newText)
|
|
})
|
|
}
|
|
|
|
func BenchmarkFormat(b *testing.B) {
|
|
ctx := format.WithFormatCodeSettings(b.Context(), &format.FormatCodeSettings{
|
|
EditorSettings: format.EditorSettings{
|
|
TabSize: 4,
|
|
IndentSize: 4,
|
|
BaseIndentSize: 4,
|
|
NewLineCharacter: "\n",
|
|
ConvertTabsToSpaces: true,
|
|
IndentStyle: format.IndentStyleSmart,
|
|
TrimTrailingWhitespace: true,
|
|
},
|
|
InsertSpaceBeforeTypeAnnotation: core.TSTrue,
|
|
}, "\n")
|
|
repo.SkipIfNoTypeScriptSubmodule(b)
|
|
filePath := filepath.Join(repo.TypeScriptSubmodulePath, "src/compiler/checker.ts")
|
|
fileContent, err := os.ReadFile(filePath)
|
|
assert.NilError(b, err)
|
|
text := string(fileContent)
|
|
sourceFile := parser.ParseSourceFile(ast.SourceFileParseOptions{
|
|
FileName: "/checker.ts",
|
|
Path: "/checker.ts",
|
|
}, text, core.ScriptKindTS)
|
|
|
|
b.Run("format checker.ts", func(b *testing.B) {
|
|
for b.Loop() {
|
|
edits := format.FormatDocument(ctx, sourceFile)
|
|
newText := applyBulkEdits(text, edits)
|
|
assert.Assert(b, len(newText) > 0)
|
|
}
|
|
})
|
|
|
|
b.Run("format checker.ts (no edit application)", func(b *testing.B) { // for comparison (how long does applying many edits take?)
|
|
for b.Loop() {
|
|
edits := format.FormatDocument(ctx, sourceFile)
|
|
assert.Assert(b, len(edits) > 0)
|
|
}
|
|
})
|
|
|
|
p := printer.NewPrinter(printer.PrinterOptions{}, printer.PrintHandlers{}, printer.NewEmitContext())
|
|
b.Run("pretty print checker.ts", func(b *testing.B) { // for comparison
|
|
for b.Loop() {
|
|
newText := p.EmitSourceFile(sourceFile)
|
|
assert.Assert(b, len(newText) > 0)
|
|
}
|
|
})
|
|
}
|