Golang NewWriterFromFileの例

コード例 #1

ファイル: logr_test.go プロジェクト: remeh/logr

func TestRotateMaxSize(t *testing.T) {
	f, err := ioutil.TempFile(os.TempDir(), "logr")
	require.Nil(t, err)

	rw, err := logr.NewWriterFromFile(f)
	require.Nil(t, err)

	now := time.Now()
	{
		n, err := rw.Write(makeBuf(0xFF))
		require.Nil(t, err)
		require.Equal(t, 1024, n)

		rw.MaxSize(512)

		n, err = rw.Write(makeBuf(0xFE))
		require.Nil(t, err)
		require.Equal(t, 1024, n)
	}

	newData := readFile(t, f.Name())
	require.Nil(t, checkEqual(t, newData, 0xFE))

	rotatedData := readFile(t, f.Name()+"."+now.Format(logr.TimeFormat))
	require.Nil(t, checkEqual(t, rotatedData, 0xFF))
}

コード例 #2

ファイル: logr_test.go プロジェクト: remeh/logr

func TestRotateMaxSizePrefix(t *testing.T) {
	f, err := ioutil.TempFile(os.TempDir(), "logr")
	require.Nil(t, err)

	rw, err := logr.NewWriterFromFile(f)
	require.Nil(t, err)
	rw.Prefix()

	now := time.Now()
	{
		n, err := rw.Write(makeBuf(0xFF))
		require.Nil(t, err)
		require.Equal(t, 1024, n)

		rw.MaxSize(512)

		n, err = rw.Write(makeBuf(0xFE))
		require.Nil(t, err)
		require.Equal(t, 1024, n)
	}

	newData := readFile(t, f.Name())
	require.Nil(t, checkEqual(t, newData, 0xFE))

	ext := filepath.Ext(f.Name())
	name := f.Name()[:len(f.Name())-len(ext)]
	rotatedData := readFile(t, name+"."+now.Format(logr.TimeFormat)+ext)
	require.Nil(t, checkEqual(t, rotatedData, 0xFF))
}