package lock import ( "log" "os" "os/exec" "strconv" "testing" "github.com/spf13/viper" ) var testDirectory = "autorestic_test_tmp" // All tests must share the same lock file as it is only initialized once func setup(t *testing.T) { d, err := os.MkdirTemp("", testDirectory) if err != nil { log.Fatalf("error creating temp dir: %v", err) return } // set config file location viper.SetConfigFile(d + "/.autorestic.yml") t.Cleanup(func() { os.RemoveAll(d) viper.Reset() }) } func TestLock(t *testing.T) { setup(t) t.Run("getLock", func(t *testing.T) { result := getLock().GetBool(RUNNING) if result { t.Errorf("got %v, want %v", result, false) } }) t.Run("lock", func(t *testing.T) { lock, err := setLockValue(RUNNING, true) if err != nil { t.Errorf("unexpected error: %v", err) } result := lock.GetBool(RUNNING) if !result { t.Errorf("got %v, want %v", result, true) } }) t.Run("unlock", func(t *testing.T) { lock, err := setLockValue(RUNNING, false) if err != nil { t.Errorf("unexpected error: %v", err) } result := lock.GetBool(RUNNING) if result { t.Errorf("got %v, want %v", result, false) } }) // locking a locked instance exits the instance // this trick to capture os.Exit(1) is discussed here: // https://talks.golang.org/2014/testing.slide#23 t.Run("lock twice", func(t *testing.T) { if os.Getenv("CRASH") == "1" { err := Lock() if err != nil { t.Errorf("unexpected error: %v", err) } // should fail Lock() } cmd := exec.Command(os.Args[0], "-test.run=TestLock/lock_twice") cmd.Env = append(os.Environ(), "CRASH=1") err := cmd.Run() err, ok := err.(*exec.ExitError) if !ok { t.Error("unexpected error") } expected := "exit status 1" if err.Error() != expected { t.Errorf("got %q, want %q", err.Error(), expected) } }) t.Run("set cron", func(t *testing.T) { expected := int64(5) SetCron("foo", expected) result, err := strconv.ParseInt(getLock().GetString("cron.foo"), 10, 64) if err != nil { t.Errorf("unexpected error: %v", err) } if result != expected { t.Errorf("got %d, want %d", result, expected) } }) t.Run("get cron", func(t *testing.T) { expected := int64(5) result := GetCron("foo") if result != expected { t.Errorf("got %d, want %d", result, expected) } }) }