@@ -16,6 +16,20 @@ import (
1616"github.com/gianlucam76/k8s_collector/pkg/utils"
1717)
1818
19+ var (
20+ data = `resources:
21+ - group: ""
22+ version: v1
23+ kind: Pod
24+ namespace: default
25+ - group: apps
26+ version: v1
27+ kind: Deployment
28+ logs:
29+ - namespace: kube-system
30+ sinceSeconds:600`
31+ )
32+
1933var _ = Describe ("Collect" , func () {
2034It ("loadConfiguration loads configuration from ConfigMap (YAML)" , func () {
2135sinceSecond := int64 (600 )
@@ -29,7 +43,7 @@ var _ = Describe("Collect", func() {
2943},
3044}
3145
32- jsonBytes , err := yaml .Marshal (collectorConfig )
46+ dataBytes , err := yaml .Marshal (collectorConfig )
3347Expect (err ).To (BeNil ())
3448
3549configMap := & corev1.ConfigMap {
@@ -38,7 +52,7 @@ var _ = Describe("Collect", func() {
3852Name : "foo" ,
3953},
4054Data : map [string ]string {
41- "config" : string (jsonBytes ),
55+ "config" : string (dataBytes ),
4256},
4357}
4458
@@ -69,7 +83,7 @@ var _ = Describe("Collect", func() {
6983},
7084}
7185
72- jsonBytes , err := json .Marshal (collectorConfig )
86+ dataBytes , err := json .Marshal (collectorConfig )
7387Expect (err ).To (BeNil ())
7488
7589configMap := & corev1.ConfigMap {
@@ -78,7 +92,7 @@ var _ = Describe("Collect", func() {
7892Name : "bar" ,
7993},
8094Data : map [string ]string {
81- "config" : string (jsonBytes ),
95+ "config" : string (dataBytes ),
8296},
8397}
8498
@@ -95,4 +109,33 @@ var _ = Describe("Collect", func() {
95109Expect (err ).To (BeNil ())
96110Expect (collectorConfig ).ToNot (BeNil ())
97111})
112+
113+ It ("loadConfiguration loads configuration from ConfigMap " , func () {
114+ dataBytes , err := yaml .Marshal (data )
115+ Expect (err ).To (BeNil ())
116+
117+ configMap := & corev1.ConfigMap {
118+ ObjectMeta : metav1.ObjectMeta {
119+ Namespace : "default" ,
120+ Name : "test" ,
121+ },
122+ Data : map [string ]string {
123+ "config" : string (dataBytes ),
124+ },
125+ }
126+
127+ collector , err := utils .GetCollectorInstance (scheme , env .Config , "" , configMap .Name )
128+ Expect (err ).To (BeNil ())
129+ Expect (k8sClient .Create (context .TODO (), configMap )).To (Succeed ())
130+
131+ waitForObject (context .TODO (), k8sClient , configMap )
132+
133+ os .Setenv ("COLLECTOR_NAMESPACE" , configMap .Namespace )
134+ config := textlogger .NewConfig (textlogger .Verbosity (1 ))
135+ logger := textlogger .NewLogger (config )
136+ collectorConfig , err := utils .LoadConfiguration (collector , context .TODO (), logger )
137+ Expect (err ).To (BeNil ())
138+ Expect (collectorConfig ).ToNot (BeNil ())
139+ Expect (len (collectorConfig .Resources )).To (Equal (2 ))
140+ })
98141})
0 commit comments