57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package gofeed_test
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/mmcdole/gofeed"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestDetectFeedType(t *testing.T) {
|
|
var feedTypeTests = []struct {
|
|
file string
|
|
expected gofeed.FeedType
|
|
}{
|
|
{"atom03_feed.xml", gofeed.FeedTypeAtom},
|
|
{"atom10_feed.xml", gofeed.FeedTypeAtom},
|
|
{"rss_feed.xml", gofeed.FeedTypeRSS},
|
|
{"rdf_feed.xml", gofeed.FeedTypeRSS},
|
|
{"unknown_feed.xml", gofeed.FeedTypeUnknown},
|
|
{"empty_feed.xml", gofeed.FeedTypeUnknown},
|
|
}
|
|
|
|
for _, test := range feedTypeTests {
|
|
fmt.Printf("Testing %s... ", test.file)
|
|
|
|
// Get feed content
|
|
path := fmt.Sprintf("testdata/parser/universal/%s", test.file)
|
|
f, _ := ioutil.ReadFile(path)
|
|
|
|
// Get actual value
|
|
actual := gofeed.DetectFeedType(bytes.NewReader(f))
|
|
|
|
if assert.Equal(t, actual, test.expected, "Feed file %s did not match expected type %d", test.file, test.expected) {
|
|
fmt.Printf("OK\n")
|
|
} else {
|
|
fmt.Printf("Failed\n")
|
|
}
|
|
}
|
|
}
|
|
|
|
// Examples
|
|
|
|
func ExampleDetectFeedType() {
|
|
feedData := `<rss version="2.0">
|
|
<channel>
|
|
<title>Sample Feed</title>
|
|
</channel>
|
|
</rss>`
|
|
feedType := gofeed.DetectFeedType(strings.NewReader(feedData))
|
|
if feedType == gofeed.FeedTypeRSS {
|
|
fmt.Println("Wow! This is an RSS feed!")
|
|
}
|
|
}
|