Function TestUploadFileLargeModuleFiles

coderd/provisionerdserver/upload_file_test.go:23–81

TestUploadFileLargeModuleFiles tests the UploadFile RPC with large module files

(t *testing.T)

// TestUploadFileLargeModuleFiles tests the UploadFile RPC with large module files
func TestUploadFileLargeModuleFiles(t *testing.T) {
	t.Parallel()

	// Create server
	server, db, _, _ := setup(t, false, &overrides{
		externalAuthConfigs: []*externalauth.Config{{}},
	})

	testSizes := []int{
		0,                             // Empty file
		512,                           // A small file
		drpcsdk.MaxMessageSize + 1024, // Just over the limit
		drpcsdk.MaxMessageSize * 2,    // 2x the limit
		sdkproto.ChunkSize*3 + 512,    // Multiple chunks with partial last
	}

	for _, size := range testSizes {
		t.Run(fmt.Sprintf("size_%d_bytes", size), func(t *testing.T) {
			t.Parallel()

			ctx := testutil.Context(t, testutil.WaitMedium)

			// Generate test module files data
			moduleData := make([]byte, size)
			_, err := crand.Read(moduleData)
			require.NoError(t, err)

			// Convert to upload format
			upload, chunks, err := sdkproto.BytesToDataUpload(sdkproto.DataUploadType_UPLOAD_TYPE_MODULE_FILES, moduleData)
			require.NoError(t, err)

			stream := newMockUploadStream(upload, chunks...)

			// Execute upload
			err = server.UploadFile(stream)
			require.NoError(t, err)

			// Upload should be done
			require.True(t, stream.isDone(), "stream should be done after upload")

			// Verify file was stored in database
			hashString := fmt.Sprintf("%x", upload.DataHash)
			file, err := db.GetFileByHashAndCreator(ctx, database.GetFileByHashAndCreatorParams{
				Hash:      hashString,
				CreatedBy: uuid.Nil, // Provisionerd creates with Nil UUID
			})
			require.NoError(t, err)
			require.Equal(t, hashString, file.Hash)
			require.Equal(t, moduleData, file.Data)
			require.Equal(t, "application/x-tar", file.Mimetype)

			// Try to upload it again, and it should still be successful
			stream = newMockUploadStream(upload, chunks...)
			err = server.UploadFile(stream)
			require.NoError(t, err, "re-upload should succeed without error")
			require.True(t, stream.isDone(), "stream should be done after re-upload")
		})
	}