0a360f54c0
This ensures this code cannot be imported by other repositories as this is only intended for internal testing packages.
77 lines
2.0 KiB
Go
77 lines
2.0 KiB
Go
package util
|
|
|
|
import (
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/tools/record"
|
|
)
|
|
|
|
// FakeConfigMap returns a configmap with the specified namespace, name and data.
|
|
func FakeConfigMap(namespace, name string, data map[string]string) *corev1.ConfigMap {
|
|
return &corev1.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: namespace,
|
|
Name: name,
|
|
},
|
|
Data: data,
|
|
}
|
|
}
|
|
|
|
// FakeEventRecorder returns an event recorder that can be used to capture events.
|
|
func FakeEventRecorder(bufferSize int) *record.FakeRecorder {
|
|
return record.NewFakeRecorder(bufferSize)
|
|
}
|
|
|
|
// FakePodWithSingleContainer returns a pod with the specified namespace and name, and having a single container with the specified image.
|
|
func FakePodWithSingleContainer(namespace, name, image string) *corev1.Pod {
|
|
enableServiceLink := corev1.DefaultEnableServiceLinks
|
|
|
|
return &corev1.Pod{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: namespace,
|
|
Name: name,
|
|
},
|
|
Spec: corev1.PodSpec{
|
|
Containers: []corev1.Container{
|
|
{
|
|
Name: name,
|
|
Image: image,
|
|
},
|
|
},
|
|
EnableServiceLinks: &enableServiceLink,
|
|
},
|
|
}
|
|
}
|
|
|
|
// FakeSecret returns a secret with the specified namespace, name and data.
|
|
func FakeSecret(namespace, name string, data map[string]string) *corev1.Secret {
|
|
res := &corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: namespace,
|
|
Name: name,
|
|
},
|
|
Data: make(map[string][]byte),
|
|
}
|
|
for key, val := range data {
|
|
res.Data[key] = []byte(val)
|
|
}
|
|
return res
|
|
}
|
|
|
|
// FakeService returns a service with the specified namespace and name and service info.
|
|
func FakeService(namespace, name, clusterIP, protocol string, port int32) *corev1.Service {
|
|
return &corev1.Service{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: namespace,
|
|
Name: name,
|
|
},
|
|
Spec: corev1.ServiceSpec{
|
|
Ports: []corev1.ServicePort{{
|
|
Protocol: corev1.Protocol(protocol),
|
|
Port: port,
|
|
}},
|
|
ClusterIP: clusterIP,
|
|
},
|
|
}
|
|
}
|