func TestUnsynchTextFrame_RoundTrip(t *testing.T) {
var (
err error
tempfile *os.File
f *File
tagger *v2.Tag
ft v2.FrameType
utextFrame *v2.UnsynchTextFrame
parsedFrame v2.Framer
resultFrame *v2.UnsynchTextFrame
ok bool
expected, actual string
)
tempfile, err = ioutil.TempFile("", "id3v2")
if err != nil {
t.Fatal(err)
}
tagger = v2.NewTag(3)
ft = v2.V23FrameTypeMap["COMM"]
utextFrame = v2.NewUnsynchTextFrame(ft, "Comment", "Foo")
tagger.AddFrames(utextFrame)
_, err = tempfile.Write(tagger.Bytes())
tempfile.Close()
if err != nil {
t.Fatal(err)
}
f, err = Open(tempfile.Name())
if err != nil {
t.Fatal(err)
}
parsedFrame = f.Frame("COMM")
if resultFrame, ok = parsedFrame.(*v2.UnsynchTextFrame); !ok {
t.Error("Couldn't cast frame")
} else {
expected = utextFrame.Description()
actual = resultFrame.Description()
if expected != actual {
t.Errorf("Expected %q, got %q", expected, actual)
}
}
}
// Opens a new tagged file
func Open(name string) (*File, error) {
fi, err := os.OpenFile(name, os.O_RDWR, 0666)
if err != nil {
return nil, err
}
file := &File{file: fi}
if v2Tag := v2.ParseTag(fi); v2Tag != nil {
file.Tagger = v2Tag
file.originalSize = v2Tag.Size()
} else if v1Tag := v1.ParseTag(fi); v1Tag != nil {
file.Tagger = v1Tag
} else {
// Add a new tag if none exists
file.Tagger = v2.NewTag(LatestVersion)
}
return file, nil
}
func TestFind(t *testing.T) {
var (
dir string
err error
f *os.File
)
for testNum, tt := range findTests {
dir, err = ioutil.TempDir("", "pavarotti")
if err != nil {
t.Errorf("test %d: %s", testNum, err)
continue
}
t.Logf("test %d: %s", testNum, dir)
defer os.RemoveAll(dir)
songPath := filepath.Join(dir, tt.path)
songDir := filepath.Dir(songPath)
if songDir != "" {
err = os.MkdirAll(songDir, 0755)
if err != nil {
t.Errorf("test %d: %s", testNum, err)
continue
}
}
f, err = os.Create(songPath)
if err != nil {
t.Errorf("test %d: %s", testNum, err)
continue
}
if tt.tagMajor > 0 {
var tagger id3.Tagger
if tt.tagMajor == 1 {
tagger = new(v1.Tag)
} else if tt.tagMajor == 2 {
tagger = v2.NewTag(tt.tagMinor)
var (
ft v2.FrameType
textFrame *v2.TextFrame
utextFrame *v2.UnsynchTextFrame
)
// add track
if tt.tagMinor == 2 {
ft = v2.V22FrameTypeMap["TRK"]
} else if tt.tagMinor == 3 {
ft = v2.V23FrameTypeMap["TRCK"]
}
textFrame = v2.NewTextFrame(ft, strconv.FormatUint(uint64(tt.tagData.track), 10))
tagger.AddFrames(textFrame)
// add album artist
if tt.tagMinor == 2 {
ft = v2.V22FrameTypeMap["TP2"]
} else if tt.tagMinor == 3 {
ft = v2.V23FrameTypeMap["TPE2"]
}
textFrame = v2.NewTextFrame(ft, tt.tagData.albumArtist)
tagger.AddFrames(textFrame)
// add comments
for _, comment := range tt.tagData.comments {
if tt.tagMinor == 2 {
ft = v2.V22FrameTypeMap["COM"]
} else if tt.tagMinor == 3 {
ft = v2.V23FrameTypeMap["COMM"]
}
utextFrame = v2.NewUnsynchTextFrame(ft, "Comment", comment)
tagger.AddFrames(utextFrame)
}
// add copyright
if tt.tagMinor == 2 {
ft = v2.V22FrameTypeMap["TCR"]
} else if tt.tagMinor == 3 {
ft = v2.V23FrameTypeMap["TCOP"]
}
textFrame = v2.NewTextFrame(ft, tt.tagData.copyright)
tagger.AddFrames(textFrame)
// add composer
if tt.tagMinor == 2 {
ft = v2.V22FrameTypeMap["TCM"]
} else if tt.tagMinor == 3 {
ft = v2.V23FrameTypeMap["TCOM"]
}
textFrame = v2.NewTextFrame(ft, tt.tagData.composer)
tagger.AddFrames(textFrame)
}
tagger.SetTitle(tt.tagData.title)
tagger.SetArtist(tt.tagData.artist)
tagger.SetAlbum(tt.tagData.album)
tagger.SetYear(strconv.Itoa(tt.tagData.year))
tagger.SetGenre(tt.tagData.genre)
if _, err = f.Write(tagger.Bytes()); err != nil {
f.Close()
t.Errorf("test %d: %s", testNum, err)
continue
}
}
f.Close()
ch := make(chan Song)
quit := make(chan bool)
go func() {
var song Song
select {
case song = <-ch:
if song.Path != songPath {
t.Errorf("test %d: expected path to be <%q>, got <%q>", testNum, songPath, song.Path)
}
if song.Artist != tt.expected.artist {
t.Errorf("test %d: expected artist to be <%q>, got <%q>", testNum, tt.expected.artist, song.Artist)
}
if song.AlbumArtist != tt.expected.albumArtist {
t.Errorf("test %d: expected albumArtist to be <%q>, got <%q>", testNum, tt.expected.albumArtist, song.AlbumArtist)
}
if song.Album != tt.expected.album {
t.Errorf("test %d: expected album to be <%q>, got <%q>", testNum, tt.expected.album, song.Album)
}
if song.Title != tt.expected.title {
t.Errorf("test %d: expected title to be <%q>, got <%q>", testNum, tt.expected.title, song.Title)
}
if song.Year != tt.expected.year {
t.Errorf("test %d: expected year to be %d, got %d", testNum, tt.expected.year, song.Year)
}
if song.Track != tt.expected.track {
t.Errorf("test %d: expected track to be %d, got %d", testNum, tt.expected.track, song.Track)
}
if song.Genre != tt.expected.genre {
t.Errorf("test %d: expected genre to be <%q>, got <%q>", testNum, tt.expected.genre, song.Genre)
}
if len(tt.expected.comments) != len(song.Comments) {
t.Errorf("test %d: expected comments to be %q, got %q", testNum, tt.expected.comments, song.Comments)
} else {
for i, comment := range tt.expected.comments {
if comment != song.Comments[i] {
t.Errorf("test %d: expected comments to be %q, got %q", testNum, tt.expected.comments, song.Comments)
break
}
}
}
if song.Composer != tt.expected.composer {
t.Errorf("test %d: expected composer to be <%q>, got <%q>", testNum, tt.expected.composer, song.Composer)
}
if song.Copyright != tt.expected.copyright {
t.Errorf("test %d: expected copyright to be <%q>, got <%q>", testNum, tt.expected.copyright, song.Copyright)
}
case <-quit:
t.Errorf("test %d: no song found", testNum)
return
}
select {
case <-ch:
t.Errorf("test %d: extra song found", testNum)
case <-quit:
}
}()
err = Find(dir, ch)
if err != nil {
t.Errorf("test %d: %s", testNum, err)
}
quit <- true
}
}