-
Notifications
You must be signed in to change notification settings - Fork 98
/
client_utils.go
50 lines (45 loc) · 2.28 KB
/
client_utils.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2016-present Datadog, Inc.
package testutils_test
import (
"github.com/DataDog/datadog-operator/apis/datadoghq/v1alpha1"
"github.com/DataDog/datadog-operator/apis/datadoghq/v2alpha1"
edsdatadoghqv1alpha1 "github.com/DataDog/extendeddaemonset/api/v1alpha1"
appsv1 "k8s.io/api/apps/v1"
corev1 "k8s.io/api/core/v1"
networkingv1 "k8s.io/api/networking/v1"
policyv1 "k8s.io/api/policy/v1beta1"
rbacv1 "k8s.io/api/rbac/v1"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/client-go/kubernetes/scheme"
apiregistrationv1 "k8s.io/kube-aggregator/pkg/apis/apiregistration/v1"
)
// TestScheme return a runtime.Scheme for testing purposes
func TestScheme(isV2 bool) *runtime.Scheme {
s := scheme.Scheme
s.AddKnownTypes(edsdatadoghqv1alpha1.GroupVersion, &edsdatadoghqv1alpha1.ExtendedDaemonSet{})
s.AddKnownTypes(v1alpha1.GroupVersion, &v1alpha1.DatadogAgentProfile{})
s.AddKnownTypes(v1alpha1.GroupVersion, &v1alpha1.DatadogAgentProfileList{})
s.AddKnownTypes(appsv1.SchemeGroupVersion, &appsv1.DaemonSet{})
s.AddKnownTypes(appsv1.SchemeGroupVersion, &appsv1.Deployment{})
s.AddKnownTypes(corev1.SchemeGroupVersion, &corev1.Secret{})
s.AddKnownTypes(corev1.SchemeGroupVersion, &corev1.ServiceAccount{})
s.AddKnownTypes(corev1.SchemeGroupVersion, &corev1.ConfigMap{})
s.AddKnownTypes(corev1.SchemeGroupVersion, &corev1.Node{})
s.AddKnownTypes(rbacv1.SchemeGroupVersion, &rbacv1.ClusterRoleBinding{})
s.AddKnownTypes(rbacv1.SchemeGroupVersion, &rbacv1.ClusterRole{})
s.AddKnownTypes(rbacv1.SchemeGroupVersion, &rbacv1.Role{})
s.AddKnownTypes(rbacv1.SchemeGroupVersion, &rbacv1.RoleBinding{})
s.AddKnownTypes(policyv1.SchemeGroupVersion, &policyv1.PodDisruptionBudget{})
s.AddKnownTypes(apiregistrationv1.SchemeGroupVersion, &apiregistrationv1.APIServiceList{})
s.AddKnownTypes(apiregistrationv1.SchemeGroupVersion, &apiregistrationv1.APIService{})
s.AddKnownTypes(networkingv1.SchemeGroupVersion, &networkingv1.NetworkPolicy{})
if isV2 {
s.AddKnownTypes(v2alpha1.GroupVersion, &v2alpha1.DatadogAgent{})
} else {
s.AddKnownTypes(v1alpha1.GroupVersion, &v1alpha1.DatadogAgent{})
}
return s
}