mirror of
https://github.com/coder/coder.git
synced 2026-06-03 04:58:23 +00:00
chore: support zip filetypes in the file cache (#18750)
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
package coderdtest
|
||||
|
||||
import (
|
||||
"archive/tar"
|
||||
"bytes"
|
||||
"context"
|
||||
"crypto"
|
||||
@@ -52,6 +53,7 @@ import (
|
||||
"cdr.dev/slog"
|
||||
"cdr.dev/slog/sloggers/sloghuman"
|
||||
"cdr.dev/slog/sloggers/slogtest"
|
||||
"github.com/coder/coder/v2/archive"
|
||||
"github.com/coder/coder/v2/coderd/files"
|
||||
"github.com/coder/quartz"
|
||||
|
||||
@@ -886,14 +888,22 @@ func createAnotherUserRetry(t testing.TB, client *codersdk.Client, organizationI
|
||||
return other, user
|
||||
}
|
||||
|
||||
// CreateTemplateVersion creates a template import provisioner job
|
||||
// with the responses provided. It uses the "echo" provisioner for compatibility
|
||||
// with testing.
|
||||
func CreateTemplateVersion(t testing.TB, client *codersdk.Client, organizationID uuid.UUID, res *echo.Responses, mutators ...func(*codersdk.CreateTemplateVersionRequest)) codersdk.TemplateVersion {
|
||||
func CreateTemplateVersionMimeType(t testing.TB, client *codersdk.Client, mimeType string, organizationID uuid.UUID, res *echo.Responses, mutators ...func(*codersdk.CreateTemplateVersionRequest)) codersdk.TemplateVersion {
|
||||
t.Helper()
|
||||
data, err := echo.TarWithOptions(context.Background(), client.Logger(), res)
|
||||
require.NoError(t, err)
|
||||
file, err := client.Upload(context.Background(), codersdk.ContentTypeTar, bytes.NewReader(data))
|
||||
|
||||
switch mimeType {
|
||||
case codersdk.ContentTypeTar:
|
||||
// do nothing
|
||||
case codersdk.ContentTypeZip:
|
||||
data, err = archive.CreateZipFromTar(tar.NewReader(bytes.NewBuffer(data)), int64(len(data)))
|
||||
require.NoError(t, err, "creating zip")
|
||||
default:
|
||||
t.Fatal("unexpected mime type", mimeType)
|
||||
}
|
||||
|
||||
file, err := client.Upload(context.Background(), mimeType, bytes.NewReader(data))
|
||||
require.NoError(t, err)
|
||||
|
||||
req := codersdk.CreateTemplateVersionRequest{
|
||||
@@ -910,6 +920,13 @@ func CreateTemplateVersion(t testing.TB, client *codersdk.Client, organizationID
|
||||
return templateVersion
|
||||
}
|
||||
|
||||
// CreateTemplateVersion creates a template import provisioner job
|
||||
// with the responses provided. It uses the "echo" provisioner for compatibility
|
||||
// with testing.
|
||||
func CreateTemplateVersion(t testing.TB, client *codersdk.Client, organizationID uuid.UUID, res *echo.Responses, mutators ...func(*codersdk.CreateTemplateVersionRequest)) codersdk.TemplateVersion {
|
||||
return CreateTemplateVersionMimeType(t, client, codersdk.ContentTypeTar, organizationID, res, mutators...)
|
||||
}
|
||||
|
||||
// CreateWorkspaceBuild creates a workspace build for the given workspace and transition.
|
||||
func CreateWorkspaceBuild(
|
||||
t *testing.T,
|
||||
|
||||
Reference in New Issue
Block a user