gomu/playlist_test.go

package main

import (
	"os"
	"path"
	"path/filepath"
	"testing"

	"github.com/rivo/tview"
)

// Prepares for test
func prepareTest() *Gomu {

	gomu := newGomu()
	gomu.player = &Player{}
	gomu.queue = newQueue()
	gomu.playlist = &Playlist{
		TreeView: tview.NewTreeView(),
	}
	gomu.app = tview.NewApplication()

	rootDir, err := filepath.Abs("./test")
	if err != nil {
		panic(err)
	}

	root := tview.NewTreeNode("music")
	rootAudioFile := &AudioFile{
		name: root.GetText(),
		path: rootDir,
	}

	root.SetReference(rootAudioFile)
	populate(root, rootDir)
	gomu.playlist.SetRoot(root)

	return gomu
}

func TestPopulate(t *testing.T) {

	gomu = newGomu()
	rootDir, err := filepath.Abs("./test")

	if err != nil {
		panic(err)
	}

	expected := 0
	walkFn := func(path string, info os.FileInfo, err error) error {

		if info.IsDir() {
			expected++
			return nil
		}

		f, e := os.Open(path)
		if e != nil {
			return e
		}

		defer f.Close()

		expected++

		return nil
	}

	// calculate the amount of mp3 files and directories
	filepath.Walk(rootDir, walkFn)

	root := tview.NewTreeNode(path.Base(rootDir))

	root.SetReference(&AudioFile{
		name:        "Music",
		isAudioFile: false,
	})

	populate(root, rootDir)
	gotItems := 1
	root.Walk(func(node, _ *tview.TreeNode) bool {
		gotItems++
		return true
	})

	if gotItems != expected {
		t.Errorf("Invalid amount of file; expected %d got %d", expected, gotItems)
	}

}

func TestAddAllToQueue(t *testing.T) {

	gomu = prepareTest()
	var songs []*tview.TreeNode

	gomu.playlist.GetRoot().Walk(func(node, parent *tview.TreeNode) bool {

		if node.GetReference().(*AudioFile).name == "rap" {
			gomu.playlist.addAllToQueue(node)
		}

		return true
	})

	queue := gomu.queue.getItems()

	for i, song := range songs {

		audioFile := song.GetReference().(*AudioFile)

		// strips the path of the song in the queue
		s := filepath.Base(queue[i])

		if audioFile.name != s {
			t.Errorf("Expected \"%s\", got \"%s\"", audioFile.name, s)
		}

	}

}