63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
|
package arm
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/Azure/go-autorest/autorest/azure"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestConfigRetrieverFillsTenantIDWhenEmpty(t *testing.T) {
|
||
|
|
c, _, _ := newConfig(getArmBuilderConfiguration(), getPackerConfiguration())
|
||
|
|
if expected := ""; c.TenantID != expected {
|
||
|
|
t.Errorf("Expected TenantID to be %q but got %q", expected, c.TenantID)
|
||
|
|
}
|
||
|
|
|
||
|
|
sut := newTestConfigRetriever()
|
||
|
|
retrievedTid := "my-tenant-id"
|
||
|
|
sut.findTenantID = func(azure.Environment, string) (string, error) { return retrievedTid, nil }
|
||
|
|
if err := sut.FillParameters(c); err != nil {
|
||
|
|
t.Errorf("Unexpected error when calling sut.FillParameters: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if expected := retrievedTid; c.TenantID != expected {
|
||
|
|
t.Errorf("Expected TenantID to be %q but got %q", expected, c.TenantID)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestConfigRetrieverLeavesTenantIDWhenNotEmpty(t *testing.T) {
|
||
|
|
c, _, _ := newConfig(getArmBuilderConfiguration(), getPackerConfiguration())
|
||
|
|
userSpecifiedTid := "not-empty"
|
||
|
|
c.TenantID = userSpecifiedTid
|
||
|
|
|
||
|
|
sut := newTestConfigRetriever()
|
||
|
|
sut.findTenantID = nil // assert that this not even called
|
||
|
|
if err := sut.FillParameters(c); err != nil {
|
||
|
|
t.Errorf("Unexpected error when calling sut.FillParameters: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if expected := userSpecifiedTid; c.TenantID != expected {
|
||
|
|
t.Errorf("Expected TenantID to be %q but got %q", expected, c.TenantID)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestConfigRetrieverReturnsErrorWhenTenantIDEmptyAndRetrievalFails(t *testing.T) {
|
||
|
|
c, _, _ := newConfig(getArmBuilderConfiguration(), getPackerConfiguration())
|
||
|
|
if expected := ""; c.TenantID != expected {
|
||
|
|
t.Errorf("Expected TenantID to be %q but got %q", expected, c.TenantID)
|
||
|
|
}
|
||
|
|
|
||
|
|
sut := newTestConfigRetriever()
|
||
|
|
errorString := "sorry, I failed"
|
||
|
|
sut.findTenantID = func(azure.Environment, string) (string, error) { return "", errors.New(errorString) }
|
||
|
|
if err := sut.FillParameters(c); err != nil && err.Error() != errorString {
|
||
|
|
t.Errorf("Unexpected error when calling sut.FillParameters: %v", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func newTestConfigRetriever() configRetriever {
|
||
|
|
return configRetriever{
|
||
|
|
findTenantID: func(azure.Environment, string) (string, error) { return "findTenantID is mocked", nil },
|
||
|
|
}
|
||
|
|
}
|