oss.indeed.com/go/go-opine@v1.3.0/internal/printing/log_writer_test.go (about)

     1  package printing
     2  
     3  import (
     4  	"bytes"
     5  	"testing"
     6  
     7  	"github.com/stretchr/testify/require"
     8  )
     9  
    10  func Test_LogWriter_Write(t *testing.T) {
    11  	toWrite := []byte("test")
    12  	var b bytes.Buffer
    13  	tested := NewLogWriter(&b)
    14  	n, err := tested.Write(toWrite)
    15  	require.NoError(t, err)
    16  	require.Equal(t, len(toWrite), n)
    17  	require.Equal(t, "test", b.String())
    18  }
    19  
    20  func Test_LogWriter_Write_error(t *testing.T) {
    21  	toWrite := []byte("test")
    22  	tested := NewLogWriter(errorWriter{n: 2})
    23  	n, err := tested.Write(toWrite)
    24  	require.NoError(t, err)
    25  	require.Equal(t, len(toWrite), n)
    26  }
    27  
    28  func Test_LogWriter_Log(t *testing.T) {
    29  	var b bytes.Buffer
    30  	tested := NewLogWriter(&b)
    31  	tested.Logf("a %s %s", "b", "c")
    32  	require.Equal(t, "a b c\n", b.String())
    33  }