github.com/khulnasoft/cli@v0.0.0-20240402070845-01bcad7beefa/cli/context/store/io_utils_test.go (about)

     1  package store
     2  
     3  import (
     4  	"io"
     5  	"strings"
     6  	"testing"
     7  
     8  	"gotest.tools/v3/assert"
     9  )
    10  
    11  func TestLimitReaderReadAll(t *testing.T) {
    12  	r := strings.NewReader("Reader")
    13  
    14  	_, err := io.ReadAll(r)
    15  	assert.NilError(t, err)
    16  
    17  	r = strings.NewReader("Test")
    18  	_, err = io.ReadAll(&LimitedReader{R: r, N: 4})
    19  	assert.NilError(t, err)
    20  
    21  	r = strings.NewReader("Test")
    22  	_, err = io.ReadAll(&LimitedReader{R: r, N: 2})
    23  	assert.Error(t, err, "read exceeds the defined limit")
    24  }