@@ -155,101 +155,70 @@ func TestIsValidDatadogAgentProfile(t *testing.T) {
155155 },
156156 },
157157 }
158- invalidFeaturesNoDDAI := & DatadogAgentProfileSpec {
159- ProfileAffinity : basicProfileAffinity ,
160- Config : & v2alpha1.DatadogAgentSpec {
161- Features : & v2alpha1.DatadogFeatures {
162- GPU : & v2alpha1.GPUFeatureConfig {
163- Enabled : ptr .To (true ),
164- },
165- },
166- },
167- }
168-
169158 testCases := []struct {
170- name string
171- spec * DatadogAgentProfileSpec
172- datadogAgentInternalEnabled bool
173- wantErr string
159+ name string
160+ spec * DatadogAgentProfileSpec
161+ wantErr string
174162 }{
175163 {
176- name : "valid dap" ,
177- spec : valid ,
178- datadogAgentInternalEnabled : true ,
179- },
180- {
181- name : "valid dap, resources specified in one container only" ,
182- spec : validResourceOverrideInOneContainerOnly ,
183- datadogAgentInternalEnabled : true ,
164+ name : "valid dap" ,
165+ spec : valid ,
184166 },
185167 {
186- name : "invalid component override" ,
187- spec : invalidComponentOverride ,
188- datadogAgentInternalEnabled : true ,
189- wantErr : "component node selector override is not supported" ,
168+ name : "valid dap, resources specified in one container only" ,
169+ spec : validResourceOverrideInOneContainerOnly ,
190170 },
191171 {
192- name : "invalid container override" ,
193- spec : invalidContainerOverride ,
194- datadogAgentInternalEnabled : true ,
195- wantErr : "container command override is not supported" ,
172+ name : "invalid component override" ,
173+ spec : invalidComponentOverride ,
174+ wantErr : "component node selector override is not supported" ,
196175 },
197176 {
198- name : "missing override when ddai disabled" ,
199- spec : missingOverride ,
200- datadogAgentInternalEnabled : false ,
201- wantErr : "config override must be defined" ,
177+ name : "invalid container override" ,
178+ spec : invalidContainerOverride ,
179+ wantErr : "container command override is not supported" ,
202180 },
203181 {
204- name : "missing config" ,
205- spec : missingConfig ,
206- datadogAgentInternalEnabled : true ,
207- wantErr : "config must be defined" ,
182+ name : "missing override is valid" ,
183+ spec : missingOverride ,
208184 },
209185 {
210- name : "missing node selector requirement" ,
211- spec : missingNSR ,
212- datadogAgentInternalEnabled : true ,
213- wantErr : "profileNodeAffinity must have at least 1 requirement" ,
186+ name : "missing config" ,
187+ spec : missingConfig ,
188+ wantErr : "config must be defined" ,
214189 },
215190 {
216- name : "missing profile node affinity" ,
217- spec : missingNodeAffinity ,
218- datadogAgentInternalEnabled : true ,
219- wantErr : "profileNodeAffinity must be defined" ,
191+ name : "missing node selector requirement" ,
192+ spec : missingNSR ,
193+ wantErr : "profileNodeAffinity must have at least 1 requirement" ,
220194 },
221195 {
222- name : "missing profile affinity" ,
223- spec : missingProfileAffinity ,
224- datadogAgentInternalEnabled : true ,
225- wantErr : "profileAffinity must be defined" ,
196+ name : "missing profile node affinity" ,
197+ spec : missingNodeAffinity ,
198+ wantErr : "profileNodeAffinity must be defined" ,
226199 },
227200 {
228- name : "gpu feature override " ,
229- spec : validGPUFeature ,
230- datadogAgentInternalEnabled : true ,
201+ name : "missing profile affinity " ,
202+ spec : missingProfileAffinity ,
203+ wantErr : "profileAffinity must be defined" ,
231204 },
232205 {
233- name : "valid dap with features only when ddai enabled" ,
234- spec : validFeaturesNoOverride ,
235- datadogAgentInternalEnabled : true ,
206+ name : "gpu feature override" ,
207+ spec : validGPUFeature ,
236208 },
237209 {
238- name : "dap with unsupported feature when ddai enabled" ,
239- spec : invalidFeatures ,
240- datadogAgentInternalEnabled : true ,
241- wantErr : "npm override is not supported" ,
210+ name : "valid dap with features only, no override" ,
211+ spec : validFeaturesNoOverride ,
242212 },
243213 {
244- name : "features not supported when ddai disabled" ,
245- spec : invalidFeaturesNoDDAI ,
246- datadogAgentInternalEnabled : false ,
247- wantErr : "the 'features' field is only supported when DatadogAgentInternal is enabled" ,
214+ name : "dap with unsupported feature" ,
215+ spec : invalidFeatures ,
216+ wantErr : "npm override is not supported" ,
248217 },
249218 }
250219 for _ , test := range testCases {
251220 t .Run (test .name , func (t * testing.T ) {
252- result := ValidateDatadogAgentProfileSpec (test .spec , test . datadogAgentInternalEnabled )
221+ result := ValidateDatadogAgentProfileSpec (test .spec )
253222 if test .wantErr != "" {
254223 assert .EqualError (t , result , test .wantErr )
255224 } else {
0 commit comments