Fix flaky zip archive test (#3367)

bufbuild · Oct 3, 2024 · 529f629 · 529f629
1 parent 4132d54
commit 529f629
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/private/buf/cmd/buf/workspace_test.go b/private/buf/cmd/buf/workspace_test.go
@@ -32,6 +32,7 @@ import (
  "github.com/bufbuild/buf/private/pkg/osext"
  "github.com/bufbuild/buf/private/pkg/protoencoding"
  "github.com/bufbuild/buf/private/pkg/slicesext"
+ "github.com/bufbuild/buf/private/pkg/storage"
  "github.com/bufbuild/buf/private/pkg/storage/storagearchive"
  "github.com/bufbuild/buf/private/pkg/storage/storageos"
  "github.com/stretchr/testify/require"
@@ -1686,19 +1687,12 @@ func createZipFromDir(t *testing.T, rootPath string, archiveName string) string
  storageos.ReadWriteBucketWithSymlinksIfSupported(),
  )
  require.NoError(t, err)
-
- zipCloser, err := zipBucket.Put(
+ require.NoError(t, storage.PutPath(
  context.Background(),
+ zipBucket,
  archiveName,
- )
- require.NoError(t, err)
- t.Cleanup(
- func() {
- require.NoError(t, zipCloser.Close())
- },
- )
- _, err = zipCloser.Write(buffer.Bytes())
- require.NoError(t, err)
+ buffer.Bytes(),
+ ))
  return zipDir
 }