9
9
10
10
"github.com/stretchr/testify/assert"
11
11
"github.com/stretchr/testify/require"
12
+ "go.uber.org/zap"
12
13
corev1 "k8s.io/api/core/v1"
13
14
v1 "k8s.io/apimachinery/pkg/apis/meta/v1"
14
15
"k8s.io/apimachinery/pkg/types"
@@ -27,16 +28,18 @@ func TestGetMetadata(t *testing.T) {
27
28
}
28
29
29
30
tests := []struct {
30
- name string
31
- containerState corev1.ContainerState
32
- expectedStatus string
33
- expectedReason string
34
- expectedStartedAt string
35
- containerName string
36
- containerID string
37
- containerImage string
38
- podName string
39
- podUID string
31
+ name string
32
+ containerState corev1.ContainerState
33
+ expectedStatus string
34
+ expectedReason string
35
+ expectedStartedAt string
36
+ containerName string
37
+ containerID string
38
+ containerImage string
39
+ containerImageName string
40
+ containerImageTag string
41
+ podName string
42
+ podUID string
40
43
}{
41
44
{
42
45
name : "Running container" ,
@@ -45,13 +48,15 @@ func TestGetMetadata(t *testing.T) {
45
48
StartedAt : refTime ,
46
49
},
47
50
},
48
- expectedStatus : containerStatusRunning ,
49
- expectedStartedAt : refTime .Format (time .RFC3339 ),
50
- containerName : "my-test-container1" ,
51
- containerID : "f37ee861-f093-4cea-aa26-f39fff8b0998" ,
52
- containerImage : "docker/someimage1" ,
53
- podName : pod .Name ,
54
- podUID : string (pod .UID ),
51
+ expectedStatus : containerStatusRunning ,
52
+ expectedStartedAt : refTime .Format (time .RFC3339 ),
53
+ containerName : "my-test-container1" ,
54
+ containerID : "f37ee861-f093-4cea-aa26-f39fff8b0998" ,
55
+ containerImage : "docker/someimage1:v1.0" ,
56
+ containerImageName : "docker/someimage1" ,
57
+ containerImageTag : "v1.0" ,
58
+ podName : pod .Name ,
59
+ podUID : string (pod .UID ),
55
60
},
56
61
{
57
62
name : "Terminated container" ,
@@ -64,14 +69,16 @@ func TestGetMetadata(t *testing.T) {
64
69
ExitCode : 0 ,
65
70
},
66
71
},
67
- expectedStatus : containerStatusTerminated ,
68
- expectedReason : "Completed" ,
69
- expectedStartedAt : refTime .Format (time .RFC3339 ),
70
- containerName : "my-test-container2" ,
71
- containerID : "f37ee861-f093-4cea-aa26-f39fff8b0997" ,
72
- containerImage : "docker/someimage2" ,
73
- podName : pod .Name ,
74
- podUID : string (pod .UID ),
72
+ expectedStatus : containerStatusTerminated ,
73
+ expectedReason : "Completed" ,
74
+ expectedStartedAt : refTime .Format (time .RFC3339 ),
75
+ containerName : "my-test-container2" ,
76
+ containerID : "f37ee861-f093-4cea-aa26-f39fff8b0997" ,
77
+ containerImage : "docker/someimage2:v1.1" ,
78
+ containerImageName : "docker/someimage2" ,
79
+ containerImageTag : "v1.1" ,
80
+ podName : pod .Name ,
81
+ podUID : string (pod .UID ),
75
82
},
76
83
{
77
84
name : "Waiting container" ,
@@ -80,16 +87,18 @@ func TestGetMetadata(t *testing.T) {
80
87
Reason : "CrashLoopBackOff" ,
81
88
},
82
89
},
83
- expectedStatus : containerStatusWaiting ,
84
- expectedReason : "CrashLoopBackOff" ,
85
- containerName : "my-test-container3" ,
86
- containerID : "f37ee861-f093-4cea-aa26-f39fff8b0996" ,
87
- containerImage : "docker/someimage3" ,
88
- podName : pod .Name ,
89
- podUID : string (pod .UID ),
90
+ expectedStatus : containerStatusWaiting ,
91
+ expectedReason : "CrashLoopBackOff" ,
92
+ containerName : "my-test-container3" ,
93
+ containerID : "f37ee861-f093-4cea-aa26-f39fff8b0996" ,
94
+ containerImage : "docker/someimage3:latest" ,
95
+ containerImageName : "docker/someimage3" ,
96
+ containerImageTag : "latest" ,
97
+ podName : pod .Name ,
98
+ podUID : string (pod .UID ),
90
99
},
91
100
}
92
-
101
+ logger := zap . NewNop ()
93
102
for _ , tt := range tests {
94
103
t .Run (tt .name , func (t * testing.T ) {
95
104
cs := corev1.ContainerStatus {
@@ -98,7 +107,7 @@ func TestGetMetadata(t *testing.T) {
98
107
ContainerID : tt .containerID ,
99
108
Image : tt .containerImage ,
100
109
}
101
- md := GetMetadata (pod , cs )
110
+ md := GetMetadata (pod , cs , logger )
102
111
103
112
require .NotNil (t , md )
104
113
assert .Equal (t , tt .expectedStatus , md .Metadata [containerKeyStatus ])
@@ -110,7 +119,8 @@ func TestGetMetadata(t *testing.T) {
110
119
assert .Equal (t , tt .expectedStartedAt , md .Metadata [containerCreationTimestamp ])
111
120
}
112
121
assert .Equal (t , tt .containerName , md .Metadata [containerName ])
113
- assert .Equal (t , tt .containerImage , md .Metadata [containerImage ])
122
+ assert .Equal (t , tt .containerImageName , md .Metadata [containerImageName ])
123
+ assert .Equal (t , tt .containerImageTag , md .Metadata [containerImageTag ])
114
124
assert .Equal (t , tt .podName , md .Metadata [constants .K8sKeyPodName ])
115
125
assert .Equal (t , tt .podUID , md .Metadata [constants .K8sKeyPodUID ])
116
126
})
0 commit comments