61 lines
1 KiB
Go
61 lines
1 KiB
Go
|
package progdoc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestParseGoFile(t *testing.T) {
|
||
|
|
sampleGoFile := `package main
|
||
|
|
|
||
|
|
/// :meta this is some meta
|
||
|
|
//
|
||
|
|
// This is some content
|
||
|
|
|
||
|
|
import "fmt"
|
||
|
|
|
||
|
|
// Ignore me
|
||
|
|
|
||
|
|
/// :more meta
|
||
|
|
//
|
||
|
|
// :arg stuff
|
||
|
|
// :arg more stuff
|
||
|
|
// :foo bar
|
||
|
|
//
|
||
|
|
// This is some more content
|
||
|
|
//
|
||
|
|
// this is indented
|
||
|
|
//
|
||
|
|
// More content goes here and there
|
||
|
|
|
||
|
|
func main() {
|
||
|
|
fmt.Println("Hello, World!")
|
||
|
|
}`
|
||
|
|
|
||
|
|
want := goFileDocs{
|
||
|
|
Parts: []goFilePart{
|
||
|
|
{
|
||
|
|
Meta: []goFileMeta{
|
||
|
|
{Name: "meta", Value: "this is some meta"},
|
||
|
|
},
|
||
|
|
Body: "This is some content",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Meta: []goFileMeta{
|
||
|
|
{Name: "more", Value: "meta"},
|
||
|
|
{Name: "arg", Value: "stuff"},
|
||
|
|
{Name: "arg", Value: "more stuff"},
|
||
|
|
{Name: "foo", Value: "bar"},
|
||
|
|
},
|
||
|
|
Body: "This is some more content\n\n this is indented\n\nMore content goes here and there",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
got, err := parseGoFileDocs(strings.NewReader(sampleGoFile))
|
||
|
|
assert.NoError(t, err)
|
||
|
|
assert.Equal(t, want, got)
|
||
|
|
}
|