Function TestPostFiles

coderd/files_test.go:26–176

(t *testing.T)

)

func TestPostFiles(t *testing.T) {
	t.Parallel()

	buildZipWithFile := func(t *testing.T, name string, writeContents func(w io.Writer) error) []byte {
		t.Helper()

		var zipBytes bytes.Buffer
		zw := zip.NewWriter(&zipBytes)
		w, err := zw.Create(name)
		require.NoError(t, err)
		require.NoError(t, writeContents(w))
		require.NoError(t, zw.Close())

		return zipBytes.Bytes()
	}

	// Single instance shared across all sub-tests. Each sub-test
	// creates independent resources with unique IDs so parallel
	// execution is safe.
	client := coderdtest.New(t, nil)
	_ = coderdtest.CreateFirstUser(t, client)
	t.Run("BadContentType", func(t *testing.T) {
		t.Parallel()
		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
		defer cancel()

		_, err := client.Upload(ctx, "bad", bytes.NewReader([]byte{'a'}))
		require.Error(t, err)
	})

	t.Run("Insert", func(t *testing.T) {
		t.Parallel()
		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
		defer cancel()

		_, err := client.Upload(ctx, codersdk.ContentTypeTar, bytes.NewReader(make([]byte, 1024)))
		require.NoError(t, err)
	})

	t.Run("InsertWindowsZip", func(t *testing.T) {
		t.Parallel()
		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
		defer cancel()

		_, err := client.Upload(ctx, "application/x-zip-compressed", bytes.NewReader(archivetest.TestZipFileBytes()))
		require.NoError(t, err)
	})

	t.Run("InsertAlreadyExists", func(t *testing.T) {
		t.Parallel()
		ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
		defer cancel()

		data := make([]byte, 1024)
		_, err := client.Upload(ctx, codersdk.ContentTypeTar, bytes.NewReader(data))
		require.NoError(t, err)
		_, err = client.Upload(ctx, codersdk.ContentTypeTar, bytes.NewReader(data))
		require.NoError(t, err)