mirror of https://github.com/linkerd/linkerd2.git
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package profiles
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/linkerd/linkerd2/controller/gen/apis/serviceprofile/v1alpha1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"sigs.k8s.io/yaml"
|
|
)
|
|
|
|
// GenServiceProfile generates a mock ServiceProfile.
|
|
func GenServiceProfile(service, namespace string) v1alpha1.ServiceProfile {
|
|
return v1alpha1.ServiceProfile{
|
|
TypeMeta: serviceProfileMeta,
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: service + "." + namespace + ".svc.cluster.local",
|
|
Namespace: namespace,
|
|
},
|
|
Spec: v1alpha1.ServiceProfileSpec{
|
|
Routes: []*v1alpha1.RouteSpec{
|
|
{
|
|
Name: "/authors/{id}",
|
|
Condition: &v1alpha1.RequestMatch{
|
|
PathRegex: "/authors/\\d+",
|
|
Method: "POST",
|
|
},
|
|
ResponseClasses: []*v1alpha1.ResponseClass{
|
|
{
|
|
Condition: &v1alpha1.ResponseMatch{
|
|
Status: &v1alpha1.Range{
|
|
Min: 500,
|
|
Max: 599,
|
|
},
|
|
},
|
|
IsFailure: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
// ServiceProfileYamlEquals validates whether two ServiceProfiles are equal.
|
|
func ServiceProfileYamlEquals(actual, expected v1alpha1.ServiceProfile) error {
|
|
if !reflect.DeepEqual(actual, expected) {
|
|
actualYaml, err := yaml.Marshal(actual)
|
|
if err != nil {
|
|
return fmt.Errorf("Service profile mismatch but failed to marshal actual service profile: %v", err)
|
|
}
|
|
expectedYaml, err := yaml.Marshal(expected)
|
|
if err != nil {
|
|
return fmt.Errorf("Service profile mismatch but failed to marshal expected service profile: %v", err)
|
|
}
|
|
return fmt.Errorf("Expected [%s] but got [%s]", string(expectedYaml), string(actualYaml))
|
|
}
|
|
return nil
|
|
}
|