@@ -223,11 +223,7 @@ func TestSaveThenOverrideMetaComponent(t *testing.T) {
223223 got := run .MustLoadDataset (t , dsPath )
224224
225225 // This dataset is ds_ten.yaml, with the meta replaced by meta_override.yaml.
226- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveThenOverrideMetaComponent.json" )
227- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
228- t .Errorf ("dataset (-want +got):\n %s" , diff )
229- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveThenOverrideMetaComponent.json" , got )
230- }
226+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveThenOverrideMetaComponent.json" , got )
231227}
232228
233229// Test save with a body, then adding a meta
@@ -248,11 +244,7 @@ func TestSaveWithBodyThenAddMetaComponent(t *testing.T) {
248244 got := run .MustLoadDataset (t , dsPath )
249245
250246 // This version has a commit message about the meta being added
251- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveWithBodyThenAddMetaComponent.json" )
252- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
253- t .Errorf ("dataset (-want +got):\n %s" , diff )
254- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveWithBodyThenAddMetaComponent.json" , got )
255- }
247+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveWithBodyThenAddMetaComponent.json" , got )
256248}
257249
258250// Test save with a body, then adding a meta
@@ -273,11 +265,7 @@ func TestSaveWithBodyThenAddMetaAndSmallBodyChange(t *testing.T) {
273265 got := run .MustLoadDataset (t , dsPath )
274266
275267 // This version has a commit message about the meta being added and body changing
276- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveWithBodyThenAddMetaAndSmallBodyChange.json" )
277- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
278- t .Errorf ("dataset (-want +got):\n %s" , diff )
279- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveWithBodyThenAddMetaAndSmallBodyChange.json" , got )
280- }
268+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveWithBodyThenAddMetaAndSmallBodyChange.json" , got )
281269}
282270
283271// Test that saving with two components at once will merge them together.
@@ -299,11 +287,7 @@ func TestSaveTwoComponents(t *testing.T) {
299287
300288 // This dataset is ds_ten.yaml, with the meta replaced by meta_override ("different title") and
301289 // the structure replaced by structure_override (lazyQuotes: false && title: "name").
302- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveTwoComponents.json" )
303- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
304- t .Errorf ("dataset (-want +got):\n %s" , diff )
305- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveTwoComponents.json" , got )
306- }
290+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveTwoComponents.json" , got )
307291}
308292
309293// Test that save can override just the transform
@@ -324,11 +308,7 @@ func TestSaveThenOverrideTransform(t *testing.T) {
324308 got := run .MustLoadDataset (t , dsPath )
325309
326310 // This dataset is ds_ten.yaml, with an added transform section
327- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveThenOverrideTransform.json" )
328- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
329- t .Errorf ("dataset (-want +got):\n %s" , diff )
330- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveThenOverrideTransform.json" , got )
331- }
311+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveThenOverrideTransform.json" , got )
332312}
333313
334314// Test that save can override just the viz
@@ -349,11 +329,7 @@ func TestSaveThenOverrideViz(t *testing.T) {
349329 got := run .MustLoadDataset (t , dsPath )
350330
351331 // This dataset is ds_ten.yaml, with an added viz section
352- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveThenOverrideViz.json" )
353- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
354- t .Errorf ("dataset (-want +got):\n %s" , diff )
355- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveThenOverrideViz.json" , got )
356- }
332+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveThenOverrideViz.json" , got )
357333}
358334
359335// Test that save can combine a meta compoent, and a transform, and a viz
@@ -374,11 +350,7 @@ func TestSaveThenOverrideMetaAndTransformAndViz(t *testing.T) {
374350 got := run .MustLoadDataset (t , dsPath )
375351
376352 // This dataset is ds_ten.yaml, with an added meta component, and transform, and viz
377- expect := dstest .LoadGoldenFile (t , "testdata/expect/TestSaveThenOverrideMetaAndTransformAndViz.json" )
378- if diff := dstest .CompareDatasets (expect , got ); diff != "" {
379- t .Errorf ("dataset (-want +got):\n %s" , diff )
380- dstest .UpdateGoldenFileIfEnvVarSet ("testdata/expect/TestSaveThenOverrideMetaAndTransformAndViz.json" , got )
381- }
353+ dstest .CompareGoldenDatasetAndUpdateIfEnvVarSet (t , "testdata/expect/TestSaveThenOverrideMetaAndTransformAndViz.json" , got )
382354}
383355
384356// Test that saving a full dataset with a component at the same time is an error
0 commit comments