| package source | |
| import ( | |
| "go/parser" | |
| "go/token" | |
| "testing" | |
| ) | |
| func TestTrimToImports(t *testing.T) { | |
| const input = `package source | |
| import ( | |
| m | |
| "fmt" | |
| ) | |
| func foo() { | |
| fmt.Println("hi") | |
| } | |
| ` | |
| fs := token.NewFileSet() | |
| f, _ := parser.ParseFile(fs, "foo.go", input, parser.ImportsOnly) | |
| trimToImports(fs, f, []byte(input)) | |
| } |