desktop/entry_test.go

package desktop

import (
	"io"
	"os"
	"reflect"
	"testing"
)

type testData struct {
	Filename string
	Entry    *Entry
}

var testCases = []*testData{
	{Filename: "app-alacritty.desktop", Entry: &Entry{Type: Application, Name: "Alacritty", GenericName: "Terminal", Comment: "A cross-platform, GPU enhanced terminal emulator", Icon: "Alacritty", Path: "/home/test", Exec: "alacritty"}},
	{Filename: "app-vim.desktop", Entry: &Entry{Type: Application, Name: "Vim", GenericName: "Text Editor", Comment: "Edit text files", Icon: "gvim", Exec: "vim %F", Terminal: true}},
	{Filename: "app-vim-nodisplay.desktop", Entry: nil},
	{Filename: "link-google.desktop", Entry: &Entry{Type: Link, Name: "Link to Google", Icon: "text-html", URL: "https://google.com"}},
}

func TestParse(t *testing.T) {
	var (
		buf   = make([]byte, bufferSize)
		f     *os.File
		entry *Entry
		err   error
	)
	for _, c := range testCases {
		f, err = os.OpenFile("test/"+c.Filename, os.O_RDONLY, 0644)
		if err != nil {
			t.Fatal(err)
		}

		entry, err = Parse(f, buf)
		f.Close()
		if err != nil {
			t.Fatal(err)
		}

		if !reflect.DeepEqual(entry, c.Entry) {
			t.Fatalf("%s: entry incorrect: got %#v, want %#v", f.Name(), entry, c.Entry)
		}
	}
}

func BenchmarkParse(b *testing.B) {
	var (
		buf   = make([]byte, bufferSize)
		files = make([]*os.File, len(testCases))
		entry *Entry
		err   error
	)
	defer func() {
		for _, f := range files {
			f.Close()
		}
	}()

	for i, c := range testCases {
		f, err := os.OpenFile("test/"+c.Filename, os.O_RDONLY, 0644)
		if err != nil {
			b.Fatal(err)
		}

		files[i] = f
	}

	b.StopTimer()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		for ci, f := range files {
			b.StartTimer()
			entry, err = Parse(f, buf)
			if err != nil {
				b.Fatal(err)
			}
			b.StopTimer()

			if !reflect.DeepEqual(entry, testCases[ci].Entry) {
				b.Fatalf("%s: entry incorrect: got %#v, want %#v", f.Name(), entry, testCases[ci].Entry)
			}

			_, err = f.Seek(0, io.SeekStart)
			if err != nil {
				b.Fatal(err)
			}
		}
	}
}