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,
			Actions: []string{"New"},
			ActionEntries: []ActionEntry{
				{
					Name: "New Terminal",
					Exec: "alacritty",
				},
			},
			Name:        "Alacritty",
			GenericName: "Terminal",
			Comment:     "A cross-platform, GPU enhanced terminal emulator",
			Icon:        "Alacritty",
			Path:        "/home/test",
			Exec:        "alacritty",
		},
	},
	{
		Filename: "app-vivaldi.desktop",
		Entry: &Entry{
			Type:        Application,
			Name:        "Vivaldi",
			GenericName: "Web Browser",
			Comment:     "Access the Internet",
			Exec:        "/usr/bin/vivaldi-stable %U",
			Terminal:    false,
			Icon:        "vivaldi",
			Actions:     []string{"new-window", "new-private-window"},
			ActionEntries: []ActionEntry{
				{
					Name: "New Window",
					Exec: "/usr/bin/vivaldi-stable --new-window",
				},
				{
					Name: "New Private Window",
					Exec: "/usr/bin/vivaldi-stable --incognito",
				},
			},
		},
	},
	{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)
			}
		}
	}
}