120 lines
2.7 KiB
Go
120 lines
2.7 KiB
Go
![]() |
package kubernetes
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
"github.com/hashicorp/terraform/backend"
|
||
|
"github.com/hashicorp/terraform/state"
|
||
|
"github.com/hashicorp/terraform/state/remote"
|
||
|
)
|
||
|
|
||
|
func TestRemoteClient_impl(t *testing.T) {
|
||
|
var _ remote.Client = new(RemoteClient)
|
||
|
var _ remote.ClientLocker = new(RemoteClient)
|
||
|
}
|
||
|
|
||
|
func TestRemoteClient(t *testing.T) {
|
||
|
testACC(t)
|
||
|
defer cleanupK8sSecrets(t)
|
||
|
|
||
|
b := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
||
|
"secret_suffix": secretSuffix,
|
||
|
}))
|
||
|
|
||
|
state, err := b.StateMgr(backend.DefaultStateName)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
remote.TestClient(t, state.(*remote.State).Client)
|
||
|
}
|
||
|
|
||
|
func TestRemoteClientLocks(t *testing.T) {
|
||
|
testACC(t)
|
||
|
defer cleanupK8sSecrets(t)
|
||
|
|
||
|
b1 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
||
|
"secret_suffix": secretSuffix,
|
||
|
}))
|
||
|
|
||
|
b2 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
||
|
"secret_suffix": secretSuffix,
|
||
|
}))
|
||
|
|
||
|
s1, err := b1.StateMgr(backend.DefaultStateName)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
s2, err := b2.StateMgr(backend.DefaultStateName)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
remote.TestRemoteLocks(t, s1.(*remote.State).Client, s2.(*remote.State).Client)
|
||
|
}
|
||
|
|
||
|
func TestForceUnlock(t *testing.T) {
|
||
|
testACC(t)
|
||
|
defer cleanupK8sSecrets(t)
|
||
|
|
||
|
b1 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
||
|
"secret_suffix": secretSuffix,
|
||
|
}))
|
||
|
|
||
|
b2 := backend.TestBackendConfig(t, New(), backend.TestWrapConfig(map[string]interface{}{
|
||
|
"secret_suffix": secretSuffix,
|
||
|
}))
|
||
|
|
||
|
// first test with default
|
||
|
s1, err := b1.StateMgr(backend.DefaultStateName)
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
info := state.NewLockInfo()
|
||
|
info.Operation = "test"
|
||
|
info.Who = "clientA"
|
||
|
|
||
|
lockID, err := s1.Lock(info)
|
||
|
if err != nil {
|
||
|
t.Fatal("unable to get initial lock:", err)
|
||
|
}
|
||
|
|
||
|
// s1 is now locked, get the same state through s2 and unlock it
|
||
|
s2, err := b2.StateMgr(backend.DefaultStateName)
|
||
|
if err != nil {
|
||
|
t.Fatal("failed to get default state to force unlock:", err)
|
||
|
}
|
||
|
|
||
|
if err := s2.Unlock(lockID); err != nil {
|
||
|
t.Fatal("failed to force-unlock default state")
|
||
|
}
|
||
|
|
||
|
// now try the same thing with a named state
|
||
|
// first test with default
|
||
|
s1, err = b1.StateMgr("test")
|
||
|
if err != nil {
|
||
|
t.Fatal(err)
|
||
|
}
|
||
|
|
||
|
info = state.NewLockInfo()
|
||
|
info.Operation = "test"
|
||
|
info.Who = "clientA"
|
||
|
|
||
|
lockID, err = s1.Lock(info)
|
||
|
if err != nil {
|
||
|
t.Fatal("unable to get initial lock:", err)
|
||
|
}
|
||
|
|
||
|
// s1 is now locked, get the same state through s2 and unlock it
|
||
|
s2, err = b2.StateMgr("test")
|
||
|
if err != nil {
|
||
|
t.Fatal("failed to get named state to force unlock:", err)
|
||
|
}
|
||
|
|
||
|
if err = s2.Unlock(lockID); err != nil {
|
||
|
t.Fatal("failed to force-unlock named state")
|
||
|
}
|
||
|
}
|