52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package docker
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/packer/packer-plugin-sdk/multistep"
|
|
"github.com/hashicorp/packer/packer-plugin-sdk/packerbuilderdata"
|
|
)
|
|
|
|
func TestStepSetGeneratedData_Run(t *testing.T) {
|
|
state := testState(t)
|
|
step := new(StepSetGeneratedData)
|
|
step.GeneratedData = &packerbuilderdata.GeneratedData{State: state}
|
|
driver := state.Get("driver").(*MockDriver)
|
|
driver.Sha256Result = "80B3BB1B1696E73A9B19DEEF92F664F8979F948DF348088B61F9A3477655AF64"
|
|
state.Put("image_id", "12345")
|
|
|
|
if action := step.Run(context.TODO(), state); action != multistep.ActionContinue {
|
|
t.Fatalf("Should not halt")
|
|
}
|
|
if !driver.Sha256Called {
|
|
t.Fatalf("driver.SHA256 should be called")
|
|
}
|
|
if driver.Sha256Id != "12345" {
|
|
t.Fatalf("driver.SHA256 got wrong image it: %s", driver.Sha256Id)
|
|
}
|
|
genData := state.Get("generated_data").(map[string]interface{})
|
|
imgSha256 := genData["ImageSha256"].(string)
|
|
if imgSha256 != driver.Sha256Result {
|
|
t.Fatalf("Expected ImageSha256 to be %s but was %s", driver.Sha256Result, imgSha256)
|
|
}
|
|
|
|
// Image ID not implement
|
|
state = testState(t)
|
|
step.GeneratedData = &packerbuilderdata.GeneratedData{State: state}
|
|
driver = state.Get("driver").(*MockDriver)
|
|
notImplementedMsg := "ERR_IMAGE_SHA256_NOT_FOUND"
|
|
|
|
if action := step.Run(context.TODO(), state); action != multistep.ActionContinue {
|
|
t.Fatalf("Should not halt")
|
|
}
|
|
if driver.Sha256Called {
|
|
t.Fatalf("driver.SHA256 should not be called")
|
|
}
|
|
genData = state.Get("generated_data").(map[string]interface{})
|
|
imgSha256 = genData["ImageSha256"].(string)
|
|
if imgSha256 != notImplementedMsg {
|
|
t.Fatalf("Expected ImageSha256 to be %s but was %s", notImplementedMsg, imgSha256)
|
|
}
|
|
}
|