91 lines
2.0 KiB
Go
91 lines
2.0 KiB
Go
package common
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/packer-plugin-sdk/template/interpolate"
|
|
)
|
|
|
|
func testShutdownConfig() *ShutdownConfig {
|
|
return &ShutdownConfig{}
|
|
}
|
|
|
|
func TestShutdownConfigPrepare_ShutdownCommand(t *testing.T) {
|
|
var c *ShutdownConfig
|
|
var errs []error
|
|
|
|
c = testShutdownConfig()
|
|
errs = c.Prepare(interpolate.NewContext())
|
|
if len(errs) > 0 {
|
|
t.Fatalf("err: %#v", errs)
|
|
}
|
|
}
|
|
|
|
func TestShutdownConfigPrepare_ShutdownTimeout(t *testing.T) {
|
|
var c *ShutdownConfig
|
|
var errs []error
|
|
|
|
// Test with a good one
|
|
c = testShutdownConfig()
|
|
c.ShutdownTimeout = 5 * time.Second
|
|
errs = c.Prepare(interpolate.NewContext())
|
|
if len(errs) > 0 {
|
|
t.Fatalf("err: %#v", errs)
|
|
}
|
|
if c.ShutdownTimeout != 5*time.Second {
|
|
t.Fatalf("bad: %s", c.ShutdownTimeout)
|
|
}
|
|
}
|
|
|
|
func TestShutdownConfigPrepare_PostShutdownDelay(t *testing.T) {
|
|
var c *ShutdownConfig
|
|
var errs []error
|
|
|
|
// Test with default value
|
|
c = testShutdownConfig()
|
|
c.PostShutdownDelay = 0
|
|
errs = c.Prepare(interpolate.NewContext())
|
|
if len(errs) > 0 {
|
|
t.Fatalf("err: %#v", errs)
|
|
}
|
|
if c.PostShutdownDelay != 2*time.Second {
|
|
t.Fatalf("bad: PostShutdownDelay should be 2 seconds but was %s", c.PostShutdownDelay)
|
|
}
|
|
|
|
// Test with a good one
|
|
c = testShutdownConfig()
|
|
c.PostShutdownDelay = 5 * time.Millisecond
|
|
errs = c.Prepare(interpolate.NewContext())
|
|
if len(errs) > 0 {
|
|
t.Fatalf("err: %#v", errs)
|
|
}
|
|
if c.PostShutdownDelay != 5*time.Millisecond {
|
|
t.Fatalf("bad: %s", c.PostShutdownDelay)
|
|
}
|
|
}
|
|
|
|
func TestShutdownConfigPrepare_DisableShutdown(t *testing.T) {
|
|
var c *ShutdownConfig
|
|
var errs []error
|
|
|
|
// Test with default value
|
|
c = testShutdownConfig()
|
|
c.DisableShutdown = false
|
|
errs = c.Prepare(interpolate.NewContext())
|
|
if len(errs) > 0 {
|
|
t.Fatalf("err: %#v", errs)
|
|
}
|
|
|
|
// Test with a good one
|
|
c = testShutdownConfig()
|
|
c.DisableShutdown = true
|
|
errs = c.Prepare(interpolate.NewContext())
|
|
if len(errs) > 0 {
|
|
t.Fatalf("err: %#v", errs)
|
|
}
|
|
if !c.DisableShutdown {
|
|
t.Fatalf("bad: %t", c.DisableShutdown)
|
|
}
|
|
}
|