70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package yandex
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/packer/packer"
|
|
|
|
"github.com/yandex-cloud/go-genproto/yandex/cloud/compute/v1"
|
|
)
|
|
|
|
func TestArtifact_impl(t *testing.T) {
|
|
var _ packer.Artifact = new(Artifact)
|
|
}
|
|
|
|
func TestArtifact_Id(t *testing.T) {
|
|
i := &compute.Image{
|
|
Id: "test-id-value",
|
|
FolderId: "test-folder-id",
|
|
}
|
|
a := &Artifact{
|
|
image: i}
|
|
expected := "test-id-value"
|
|
|
|
if a.Id() != expected {
|
|
t.Fatalf("artifact ID should match: %v", expected)
|
|
}
|
|
}
|
|
|
|
func TestArtifact_String(t *testing.T) {
|
|
i := &compute.Image{
|
|
Id: "test-id-value",
|
|
FolderId: "test-folder-id",
|
|
Name: "test-name",
|
|
Family: "test-family",
|
|
}
|
|
a := &Artifact{
|
|
image: i}
|
|
expected := "A disk image was created: test-name (id: test-id-value) with family name test-family"
|
|
|
|
if a.String() != expected {
|
|
t.Fatalf("artifact string should match: %v", expected)
|
|
}
|
|
}
|
|
|
|
func TestArtifactState(t *testing.T) {
|
|
expectedData := "this is the data"
|
|
artifact := &Artifact{
|
|
StateData: map[string]interface{}{"state_data": expectedData},
|
|
}
|
|
|
|
// Valid state
|
|
result := artifact.State("state_data")
|
|
if result != expectedData {
|
|
t.Fatalf("Bad: State data was %s instead of %s", result, expectedData)
|
|
}
|
|
|
|
// Invalid state
|
|
result = artifact.State("invalid_key")
|
|
if result != nil {
|
|
t.Fatalf("Bad: State should be nil for invalid state data name")
|
|
}
|
|
|
|
// Nil StateData should not fail and should return nil
|
|
artifact = &Artifact{}
|
|
result = artifact.State("key")
|
|
if result != nil {
|
|
t.Fatalf("Bad: State should be nil for nil StateData")
|
|
}
|
|
}
|