code.gitea.io/gitea@v1.19.3/modules/util/filebuffer/file_backed_buffer_test.go (about)

     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  }