gitech

1
// Copyright 2022 The Gitea Authors. All rights reserved.
2
// SPDX-License-Identifier: MIT
3

4
package filebuffer
5

6
import (
7
	"io"
8
	"strings"
9
	"testing"
10

11
	"github.com/stretchr/testify/assert"
12
)
13

14
func TestFileBackedBuffer(t *testing.T) {
15
	cases := []struct {
16
		MaxMemorySize int
17
		Data          string
18
	}{
19
		{5, "test"},
20
		{5, "testtest"},
21
	}
22

23
	for _, c := range cases {
24
		buf, err := CreateFromReader(strings.NewReader(c.Data), c.MaxMemorySize)
25
		assert.NoError(t, err)
26

27
		assert.EqualValues(t, len(c.Data), buf.Size())
28

29
		data, err := io.ReadAll(buf)
30
		assert.NoError(t, err)
31
		assert.Equal(t, c.Data, string(data))
32

33
		assert.NoError(t, buf.Close())
34
	}
35
}
36