github.com/minio/mc@v0.0.0-20240507152021-646712d5e5fb/pkg/hookreader/hookreader_test.go (about)

     1  // Copyright (c) 2015-2021 MinIO, Inc.
     2  //
     3  // This file is part of MinIO Object Storage stack
     4  //
     5  // This program is free software: you can redistribute it and/or modify
     6  // it under the terms of the GNU Affero General Public License as published by
     7  // the Free Software Foundation, either version 3 of the License, or
     8  // (at your option) any later version.
     9  //
    10  // This program is distributed in the hope that it will be useful
    11  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    12  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    13  // GNU Affero General Public License for more details.
    14  //
    15  // You should have received a copy of the GNU Affero General Public License
    16  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    17  
    18  package hookreader
    19  
    20  import (
    21  	"bytes"
    22  	"testing"
    23  
    24  	check "gopkg.in/check.v1"
    25  )
    26  
    27  func Test(t *testing.T) { check.TestingT(t) }
    28  
    29  type MySuite struct{}
    30  
    31  var _ = check.Suite(&MySuite{})
    32  
    33  // customReader - implements custom progress reader.
    34  type customReader struct {
    35  	readBytes int
    36  }
    37  
    38  func (c *customReader) Read(b []byte) (n int, err error) {
    39  	c.readBytes += len(b)
    40  	return len(b), nil
    41  }
    42  
    43  // Tests hook reader implementation.
    44  func (s *MySuite) TestHookReader(c *check.C) {
    45  	var buffer bytes.Buffer
    46  	writer := &buffer
    47  	_, err := writer.Write([]byte("Hello"))
    48  	c.Assert(err, check.IsNil)
    49  	progress := &customReader{}
    50  	reader := NewHook(&buffer, progress)
    51  	b := make([]byte, 3)
    52  	n, err := reader.Read(b)
    53  	c.Assert(err, check.IsNil)
    54  	c.Assert(n, check.Equals, 3)
    55  	c.Assert(progress.readBytes, check.Equals, 3)
    56  }