kevinschoon-pomo/pkg/cmd/cmd_test.go

package cmd

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

	pomo "github.com/kevinschoon/pomo/pkg/internal"
)

func checkErr(t *testing.T, err error) {
	if err != nil {
		t.Helper()
		t.Fatal(err)
	}
}

func initTestConfig(t *testing.T) (*pomo.Store, *pomo.Config) {
	tmpPath, err := ioutil.TempDir(os.TempDir(), "pomo-test")
	checkErr(t, err)
	config := &pomo.Config{
		DateTimeFmt: "2006-01-02 15:04",
		BasePath:    tmpPath,
		DBPath:      filepath.Join(tmpPath, "pomo.db"),
		SocketPath:  filepath.Join(tmpPath, "pomo.sock"),
		IconPath:    filepath.Join(tmpPath, "icon.png"),
	}
	store, err := pomo.NewStore(config.DBPath)
	checkErr(t, err)
	checkErr(t, pomo.InitDB(store))
	return store, config
}

func TestPomoCreate(t *testing.T) {
	store, config := initTestConfig(t)
	cmd := New(config)
	checkErr(t, cmd.Run([]string{"pomo", "create", "fuu"}))
	// verify the task was created
	store.With(func(tx *sql.Tx) error {
		task, err := store.ReadTask(tx, 1)
		checkErr(t, err)
		if task.Message != "fuu" {
			checkErr(t, fmt.Errorf("task should have message fuu, got %s", task.Message))
		}
		return nil
	})
}