diff options
author | KN4CK3R <admin@oldschoolhack.me> | 2022-08-02 15:35:48 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-08-02 21:35:48 +0800 |
commit | 631539c104c790fccf31a80a92e31fcc8a16521c (patch) | |
tree | 69e4e291faee2b831f3f6ed64542a97cda5b0930 /modules/util/filebuffer/file_backed_buffer_test.go | |
parent | e43bb2b0b51e45e157df5d5cdcb4a63b2847e469 (diff) | |
download | gitea-631539c104c790fccf31a80a92e31fcc8a16521c.tar.gz gitea-631539c104c790fccf31a80a92e31fcc8a16521c.zip |
Fix package upload for files >32mb (#20622)
* Rewind file before first read.
* Added tests.
Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
Diffstat (limited to 'modules/util/filebuffer/file_backed_buffer_test.go')
-rw-r--r-- | modules/util/filebuffer/file_backed_buffer_test.go | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/modules/util/filebuffer/file_backed_buffer_test.go b/modules/util/filebuffer/file_backed_buffer_test.go new file mode 100644 index 0000000000..83ef58561d --- /dev/null +++ b/modules/util/filebuffer/file_backed_buffer_test.go @@ -0,0 +1,36 @@ +// Copyright 2022 The Gitea Authors. All rights reserved. +// Use of this source code is governed by a MIT-style +// license that can be found in the LICENSE file. + +package filebuffer + +import ( + "io" + "strings" + "testing" + + "github.com/stretchr/testify/assert" +) + +func TestFileBackedBuffer(t *testing.T) { + cases := []struct { + MaxMemorySize int + Data string + }{ + {5, "test"}, + {5, "testtest"}, + } + + for _, c := range cases { + buf, err := CreateFromReader(strings.NewReader(c.Data), c.MaxMemorySize) + assert.NoError(t, err) + + assert.EqualValues(t, len(c.Data), buf.Size()) + + data, err := io.ReadAll(buf) + assert.NoError(t, err) + assert.Equal(t, c.Data, string(data)) + + assert.NoError(t, buf.Close()) + } +} |