819329228a
It is simply the best/simplest solution and trying to prevent users from passing and integer here would be like opening a can of worms. Because: * we cannot make mapstructure validate our duration string ( with an UnmarshalJSON func etc.) * we cannot make mapstructure spit a string instead of a duration and packer will decode-encode-decode config. * the hcl2 generated code asks for a string, so this will be enforced by default.
57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package sleep
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func test1sConfig() map[string]interface{} {
|
|
return map[string]interface{}{
|
|
"duration": "1s",
|
|
}
|
|
}
|
|
|
|
func TestConfigPrepare_1s(t *testing.T) {
|
|
raw := test1sConfig()
|
|
var p Provisioner
|
|
err := p.Prepare(raw)
|
|
if err != nil {
|
|
t.Fatalf("prerare failed: %v", err)
|
|
}
|
|
|
|
if p.Duration != time.Second {
|
|
t.Fatal("wrong duration")
|
|
}
|
|
}
|
|
|
|
func TestProvisioner_Provision(t *testing.T) {
|
|
ctxCancelled, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
type fields struct {
|
|
Duration time.Duration
|
|
}
|
|
type args struct {
|
|
ctx context.Context
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
fields fields
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{"valid sleep", fields{1 * time.Millisecond}, args{context.Background()}, false},
|
|
{"timeout", fields{1 * time.Millisecond}, args{ctxCancelled}, true},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
p := &Provisioner{
|
|
Duration: tt.fields.Duration,
|
|
}
|
|
if err := p.Provision(tt.args.ctx, nil, nil); (err != nil) != tt.wantErr {
|
|
t.Errorf("Provisioner.Provision() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
})
|
|
}
|
|
}
|