mirror of
https://github.com/opentofu/opentofu.git
synced 2025-01-02 12:17:39 -06:00
f40800b3a4
This is part of a general effort to move all of Terraform's non-library package surface under internal in order to reinforce that these are for internal use within Terraform only. If you were previously importing packages under this prefix into an external codebase, you could pin to an earlier release tag as an interim solution until you've make a plan to achieve the same functionality some other way.
96 lines
1.8 KiB
Go
96 lines
1.8 KiB
Go
package statemgr
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"flag"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
_ "github.com/hashicorp/terraform/internal/logging"
|
|
)
|
|
|
|
func TestNewLockInfo(t *testing.T) {
|
|
info1 := NewLockInfo()
|
|
info2 := NewLockInfo()
|
|
|
|
if info1.ID == "" {
|
|
t.Fatal("LockInfo missing ID")
|
|
}
|
|
|
|
if info1.Version == "" {
|
|
t.Fatal("LockInfo missing version")
|
|
}
|
|
|
|
if info1.Created.IsZero() {
|
|
t.Fatal("LockInfo missing Created")
|
|
}
|
|
|
|
if info1.ID == info2.ID {
|
|
t.Fatal("multiple LockInfo with identical IDs")
|
|
}
|
|
|
|
// test the JSON output is valid
|
|
newInfo := &LockInfo{}
|
|
err := json.Unmarshal(info1.Marshal(), newInfo)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestLockWithContext(t *testing.T) {
|
|
s := NewFullFake(nil, TestFullInitialState())
|
|
|
|
id, err := s.Lock(NewLockInfo())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// use a cancelled context for an immediate timeout
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
info := NewLockInfo()
|
|
info.Info = "lock with context"
|
|
_, err = LockWithContext(ctx, s, info)
|
|
if err == nil {
|
|
t.Fatal("lock should have failed immediately")
|
|
}
|
|
|
|
// block until LockwithContext has made a first attempt
|
|
attempted := make(chan struct{})
|
|
postLockHook = func() {
|
|
close(attempted)
|
|
postLockHook = nil
|
|
}
|
|
|
|
// unlock the state during LockWithContext
|
|
unlocked := make(chan struct{})
|
|
var unlockErr error
|
|
go func() {
|
|
defer close(unlocked)
|
|
<-attempted
|
|
unlockErr = s.Unlock(id)
|
|
}()
|
|
|
|
ctx, cancel = context.WithTimeout(context.Background(), 2*time.Second)
|
|
defer cancel()
|
|
|
|
id, err = LockWithContext(ctx, s, info)
|
|
if err != nil {
|
|
t.Fatal("lock should have completed within 2s:", err)
|
|
}
|
|
|
|
// ensure the goruotine completes
|
|
<-unlocked
|
|
if unlockErr != nil {
|
|
t.Fatal(unlockErr)
|
|
}
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
flag.Parse()
|
|
os.Exit(m.Run())
|
|
}
|