@@ -319,36 +319,29 @@ var _ = Describe("Repo Config", func() {
319319 Expect (repoConfig .OnlineStore ).To (Equal (expectedOnlineConfig ))
320320 Expect (repoConfig .Registry ).To (Equal (expectedRegistryConfig ))
321321
322- By ("Having DQM config with initial distribution disabled" )
322+ By ("Having DQM config with auto_baseline disabled" )
323323 featureStore = minimalFeatureStore ()
324- dqmEnabled := false
324+ dqmAutoBaseline := false
325325 featureStore .Spec .Dqm = & feastdevv1.DqmConfig {
326- Distribution : & feastdevv1.DqmDistributionConfig {
327- Initial : & feastdevv1.DqmInitialDistributionConfig {
328- Enabled : & dqmEnabled ,
329- },
330- },
326+ AutoBaseline : & dqmAutoBaseline ,
331327 }
332328 ApplyDefaultsToStatus (featureStore )
333329 repoConfig , err = getServiceRepoConfig (featureStore , emptyMockExtractConfigFromSecret , emptyMockExtractConfigFromConfigMap , false )
334330 Expect (err ).NotTo (HaveOccurred ())
335- Expect (repoConfig .FeatureServer ).NotTo (BeNil ())
336- Expect (repoConfig .FeatureServer .Dqm ).NotTo (BeNil ())
337- Expect (repoConfig .FeatureServer .Dqm .Distribution ).NotTo (BeNil ())
338- Expect (repoConfig .FeatureServer .Dqm .Distribution .Initial ).NotTo (BeNil ())
339- Expect (repoConfig .FeatureServer .Dqm .Distribution .Initial .Enabled ).To (BeFalse ())
331+ Expect (repoConfig .Dqm ).NotTo (BeNil ())
332+ Expect (repoConfig .Dqm .AutoBaseline ).To (BeFalse ())
340333
341334 fsYaml , marshalErr := yaml .Marshal (repoConfig )
342335 Expect (marshalErr ).NotTo (HaveOccurred ())
343- Expect (string (fsYaml )).To (ContainSubstring ("feature_server :" ))
344- Expect (string (fsYaml )).To (ContainSubstring ("enabled : false" ))
336+ Expect (string (fsYaml )).To (ContainSubstring ("dqm :" ))
337+ Expect (string (fsYaml )).To (ContainSubstring ("auto_baseline : false" ))
345338
346- By ("Having no DQM config — feature_server should be nil" )
339+ By ("Having no DQM config — dqm should be nil" )
347340 featureStore = minimalFeatureStore ()
348341 ApplyDefaultsToStatus (featureStore )
349342 repoConfig , err = getServiceRepoConfig (featureStore , emptyMockExtractConfigFromSecret , emptyMockExtractConfigFromConfigMap , false )
350343 Expect (err ).NotTo (HaveOccurred ())
351- Expect (repoConfig .FeatureServer ).To (BeNil ())
344+ Expect (repoConfig .Dqm ).To (BeNil ())
352345 })
353346
354347 It ("should set feature_server block with type local and all options" , func () {
0 commit comments