Skip to content

Commit 405dc22

Browse files
committed
changefeedccl: clean up temporary files during parquet writer tests
Nearly every nightly failure I look at has a temporary directory filled with these files. Epic: None Release note: None
1 parent 517d0c7 commit 405dc22

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

pkg/util/parquet/writer_test.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ func TestRandomDatums(t *testing.T) {
119119
fileName := "TestRandomDatums.parquet"
120120
f, err := os.CreateTemp("", fileName)
121121
require.NoError(t, err)
122+
defer removeFileUnlessFailed(t, f)
122123

123124
schemaDef, err := NewSchema(sch.columnNames, sch.columnTypes)
124125
require.NoError(t, err)
@@ -520,6 +521,7 @@ func TestBasicDatums(t *testing.T) {
520521
fileName := "TestBasicDatums.parquet"
521522
f, err := os.CreateTemp("", fileName)
522523
require.NoError(t, err)
524+
defer removeFileUnlessFailed(t, f)
523525

524526
schemaDef, err := NewSchema(tc.sch.columnNames, tc.sch.columnTypes)
525527
require.NoError(t, err)
@@ -627,6 +629,7 @@ func optionsTest(t *testing.T, opt Option, testFn func(t *testing.T, reader *fil
627629
fileName := "OptionsTest.parquet"
628630
f, err := os.CreateTemp("", fileName)
629631
require.NoError(t, err)
632+
defer removeFileUnlessFailed(t, f)
630633

631634
writer, err := NewWriter(schemaDef, f, opt)
632635
require.NoError(t, err)
@@ -649,6 +652,16 @@ func optionsTest(t *testing.T, opt Option, testFn func(t *testing.T, reader *fil
649652
require.NoError(t, err)
650653
}
651654

655+
func removeFileUnlessFailed(t *testing.T, f *os.File) {
656+
t.Helper()
657+
if !t.Failed() {
658+
return
659+
}
660+
if err := os.Remove(f.Name()); err != nil {
661+
t.Logf("failed to remove file %s: %v", f.Name(), err)
662+
}
663+
}
664+
652665
func TestSquashTuples(t *testing.T) {
653666
datums := []tree.Datum{
654667
tree.NewDInt(1),

0 commit comments

Comments
 (0)