Function TestUploadFileErrorScenarios

coderd/provisionerdserver/upload_file_test.go:84–141

TestUploadFileErrorScenarios tests various error conditions in file upload

(t *testing.T)

// TestUploadFileErrorScenarios tests various error conditions in file upload
func TestUploadFileErrorScenarios(t *testing.T) {
	t.Parallel()

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

	// Generate test data
	moduleData := make([]byte, sdkproto.ChunkSize*2)
	_, err := crand.Read(moduleData)
	require.NoError(t, err)

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

	t.Run("chunk_before_upload", func(t *testing.T) {
		t.Parallel()

		stream := newMockUploadStream(nil, chunks[0])

		err := server.UploadFile(stream)
		require.ErrorContains(t, err, "unexpected chunk piece while waiting for file upload")
		require.True(t, stream.isDone(), "stream should be done after error")
	})

	t.Run("duplicate_upload", func(t *testing.T) {
		t.Parallel()

		stream := &mockUploadStream{
			done:     make(chan struct{}),
			messages: make(chan *sdkproto.FileUpload, 2),
		}

		up := &sdkproto.FileUpload{Type: &sdkproto.FileUpload_DataUpload{DataUpload: upload}}

		// Send it twice
		stream.messages <- up
		stream.messages <- up

		err := server.UploadFile(stream)
		require.ErrorContains(t, err, "unexpected file download while waiting for file completion")
		require.True(t, stream.isDone(), "stream should be done after error")
	})

	t.Run("unsupported_upload_type", func(t *testing.T) {
		t.Parallel()

		//nolint:govet // Ignore lock copy
		cpy := *upload
		cpy.UploadType = sdkproto.DataUploadType_UPLOAD_TYPE_UNKNOWN // Set to an unsupported type
		stream := newMockUploadStream(&cpy, chunks...)

		err := server.UploadFile(stream)
		require.ErrorContains(t, err, "unsupported file upload type")
		require.True(t, stream.isDone(), "stream should be done after error")
	})
}