77 lines
2.4 KiB
Go
77 lines
2.4 KiB
Go
package resources
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform-plugin-framework/types"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
)
|
|
|
|
func TestFunctionToUnstructuredAndBack(t *testing.T) {
|
|
input := functionResourceModel{
|
|
Name: types.StringValue("fn-a"),
|
|
Environment: types.StringValue("env-a"),
|
|
PackageName: types.StringValue("pkg-a"),
|
|
Entrypoint: types.StringValue("main.main"),
|
|
}
|
|
|
|
obj := functionToUnstructured(input, "default")
|
|
state := unstructuredToFunctionModel(obj, input)
|
|
|
|
if state.Name.ValueString() != "fn-a" {
|
|
t.Fatalf("unexpected name: %q", state.Name.ValueString())
|
|
}
|
|
if state.Environment.ValueString() != "env-a" {
|
|
t.Fatalf("unexpected environment: %q", state.Environment.ValueString())
|
|
}
|
|
if state.PackageName.ValueString() != "pkg-a" {
|
|
t.Fatalf("unexpected package_name: %q", state.PackageName.ValueString())
|
|
}
|
|
if state.Entrypoint.ValueString() != "main.main" {
|
|
t.Fatalf("unexpected entrypoint: %q", state.Entrypoint.ValueString())
|
|
}
|
|
|
|
invoke, found, err := unstructured.NestedMap(obj.Object, "spec", "InvokeStrategy")
|
|
if err != nil || !found || len(invoke) == 0 {
|
|
t.Fatalf("InvokeStrategy not set correctly")
|
|
}
|
|
}
|
|
|
|
func TestValidateEntrypointAgainstPackageSourcePythonOK(t *testing.T) {
|
|
source := "def main():\n return 'ok'\n"
|
|
pkg := &unstructured.Unstructured{Object: map[string]interface{}{
|
|
"spec": map[string]interface{}{
|
|
"deployment": map[string]interface{}{
|
|
"literal": base64.StdEncoding.EncodeToString([]byte(source)),
|
|
},
|
|
},
|
|
}}
|
|
|
|
if err := validateEntrypointAgainstPackageSource("main.main", pkg); err != nil {
|
|
t.Fatalf("expected valid entrypoint, got error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestValidateEntrypointAgainstPackageSourcePythonMissing(t *testing.T) {
|
|
source := "def another():\n return 'ok'\n"
|
|
pkg := &unstructured.Unstructured{Object: map[string]interface{}{
|
|
"spec": map[string]interface{}{
|
|
"deployment": map[string]interface{}{
|
|
"literal": base64.StdEncoding.EncodeToString([]byte(source)),
|
|
},
|
|
},
|
|
}}
|
|
|
|
if err := validateEntrypointAgainstPackageSource("main.main", pkg); err == nil {
|
|
t.Fatalf("expected validation error for missing python function")
|
|
}
|
|
}
|
|
|
|
func TestValidateEntrypointAgainstPackageSourceBadFormat(t *testing.T) {
|
|
pkg := &unstructured.Unstructured{}
|
|
if err := validateEntrypointAgainstPackageSource("main", pkg); err == nil {
|
|
t.Fatalf("expected validation error for bad entrypoint format")
|
|
}
|
|
}
|