github.com/ChicK00o/awgo@v0.29.4/testutils_test.go

// Copyright (c) 2018 Dean Jackson <deanishe@deanishe.net>
// MIT Licence - http://opensource.org/licenses/MIT

package aw

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
	"go.deanishe.net/env"
)

var (
	tVersion                  = "1.2.0"
	tName                     = "AwGo"
	tBundleID                 = "net.deanishe.awgo"
	tUID                      = "user.workflow.4B0E9731-E139-4179-BC50-D7FFF82B269A"
	tDebug                    = true
	tAlfredVersion            = "3.6"
	tAlfredBuild              = "901"
	tTheme                    = "alfred.theme.custom.DE3D17CA-64A2-4B42-A3F6-C71DB1201F88"
	tThemeBackground          = "rgba(255,255,255,1.00)"
	tThemeSelectionBackground = "rgba(255,255,255,1.00)"
	tPreferences              = os.ExpandEnv("$HOME/Library/Application Support/Alfred/Alfred.alfredpreferences")
	tLocalhash                = "0dd4500828b5c675862eaa7786cf0f374823b965"
	tCacheDir                 = os.ExpandEnv("$HOME/Library/Caches/com.runningwithcrayons.Alfred/Workflow Data/net.deanishe.awgo")
	tDataDir                  = os.ExpandEnv("$HOME/Library/Application Support/Alfred/Workflow Data/net.deanishe.awgo")

	testEnv = env.MapEnv{
		EnvVarVersion:          tVersion,
		EnvVarName:             tName,
		EnvVarBundleID:         tBundleID,
		EnvVarUID:              tUID,
		EnvVarDebug:            fmt.Sprintf("%v", tDebug),
		EnvVarAlfredVersion:    tAlfredVersion,
		EnvVarAlfredBuild:      tAlfredBuild,
		EnvVarTheme:            tTheme,
		EnvVarThemeBG:          tThemeBackground,
		EnvVarThemeSelectionBG: tThemeSelectionBackground,
		EnvVarPreferences:      tPreferences,
		EnvVarLocalhash:        tLocalhash,
		EnvVarCacheDir:         tCacheDir,
		EnvVarDataDir:          tDataDir,
	}
)

// Mock os.Exit
type mockExit struct {
	code int
}

func (me *mockExit) Exit(code int) { me.code = code }

// Mock exec.Command
type mockExec struct {
	name string
	args []string
}

func (me *mockExec) Run(name string, arg ...string) error {
	me.name = name
	me.args = append([]string{name}, arg...)
	return nil
}

func panicOnErr(err error) {
	if err != nil {
		panic(err)
	}
}

// create a temporary directory, call function fn, delete the directory.
func withTempDir(fn func(dir string)) {
	p, err := ioutil.TempDir("", "awgo-")
	if err != nil {
		panic(err)
	}
	if p, err = filepath.EvalSymlinks(p); err != nil {
		panic(err)
	}
	defer panicOnErr(os.RemoveAll(p))
	fn(p)
}

// Call function with a test environment.
func withTestEnv(fn func(e env.MapEnv)) {
	e := env.MapEnv{
		EnvVarVersion:          tVersion,
		EnvVarName:             tName,
		EnvVarBundleID:         tBundleID,
		EnvVarUID:              tUID,
		EnvVarDebug:            fmt.Sprintf("%v", tDebug),
		EnvVarAlfredVersion:    tAlfredVersion,
		EnvVarAlfredBuild:      tAlfredBuild,
		EnvVarTheme:            tTheme,
		EnvVarThemeBG:          tThemeBackground,
		EnvVarThemeSelectionBG: tThemeSelectionBackground,
		EnvVarPreferences:      tPreferences,
		EnvVarLocalhash:        tLocalhash,
		EnvVarCacheDir:         tCacheDir,
		EnvVarDataDir:          tDataDir,
	}

	fn(e)
}

// Call function in a test workflow environment.
func withTestWf(fn func(wf *Workflow)) {
	withTestEnv(func(e env.MapEnv) {
		var (
			dir string
			err error
		)

		if dir, err = ioutil.TempDir("", "awgo-"); err != nil {
			panic(err)
		}
		// TempDir() returns a symlink on my macOS :(
		if dir, err = filepath.EvalSymlinks(dir); err != nil {
			panic(err)
		}

		defer func() {
			if err := os.RemoveAll(dir); err != nil {
				panic(err)
			}
		}()

		var (
			dataDir  = filepath.Join(dir, "data")
			cacheDir = filepath.Join(dir, "cache")
		)

		// Update env to point to cache & data dirs
		e[EnvVarCacheDir] = cacheDir
		e[EnvVarDataDir] = dataDir

		// Create test files & directories
		for _, p := range []string{dataDir, cacheDir} {
			if err := os.MkdirAll(p, os.ModePerm); err != nil {
				panic(err)
			}
		}

		// Create workflow for current environment and pass it to function.
		var wf = NewFromEnv(e)
		fn(wf)
	})
}

// TestWithTestWf verifies the withTestEnv helper.
func TestWithTestWf(t *testing.T) {
	withTestWf(func(wf *Workflow) {
		wd, err := os.Getwd()
		if err != nil {
			t.Fatal(err)
		}
		data := []struct {
			name, x, v string
		}{
			{"Workflow.Dir", wd, wf.Dir()},
			{"Name", tName, wf.Name()},
			{"BundleID", tBundleID, wf.BundleID()},

			{"Config.UID", tUID, wf.Config.Get(EnvVarUID)},
			{"Config.AlfredVersion", tAlfredVersion, wf.Config.Get(EnvVarAlfredVersion)},
			{"Config.AlfredBuild", tAlfredBuild, wf.Config.Get(EnvVarAlfredBuild)},
			{"Config.Theme", tTheme, wf.Config.Get(EnvVarTheme)},
			{"Config.ThemeBackground", tThemeBackground, wf.Config.Get(EnvVarThemeBG)},
			{"Config.ThemeSelectionBackground", tThemeSelectionBackground,
				wf.Config.Get(EnvVarThemeSelectionBG)},
			{"Config.Preferences", tPreferences, wf.Config.Get(EnvVarPreferences)},
			{"Config.Localhash", tLocalhash, wf.Config.Get(EnvVarLocalhash)},
		}

		assert.Equal(t, tDebug, wf.Debug(), "unexpected debug")

		for _, td := range data {
			td := td
			t.Run(td.name, func(t *testing.T) {
				assert.Equal(t, td.x, td.v, "unexpected variable")
			})
		}
	})
}