github.com/ali-iotechsys/cli@v20.10.0+incompatible/cli/context/store/io_utils_test.go (about)

     1  package store
     2  
     3  import (
     4  	"io/ioutil"
     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 := ioutil.ReadAll(r)
    15  	assert.NilError(t, err)
    16  
    17  	r = strings.NewReader("Test")
    18  	_, err = ioutil.ReadAll(&LimitedReader{R: r, N: 4})
    19  	assert.NilError(t, err)
    20  
    21  	r = strings.NewReader("Test")
    22  	_, err = ioutil.ReadAll(&LimitedReader{R: r, N: 2})
    23  	assert.Error(t, err, "read exceeds the defined limit")
    24  }