go/src/infra/cros/recovery/logger/logger_test.go

// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

// Package logger provides an abstract representation of logging interfaces used by recovery lib.
package logger

import (
	"bytes"
	"log"
	"os"
	"testing"

	"go.chromium.org/infra/cros/internal/assert"
)

var want = `
Line 1
Line 2
Line 3
Line 4
Line 5
`

func TestBasicLogging(t *testing.T) {
	var buf bytes.Buffer
	log.SetOutput(&buf)
	defer log.SetOutput(os.Stderr)
	flags := log.Flags()
	log.SetFlags(0)
	defer log.SetFlags(flags)

	l := NewLogger()

	// No indent
	// Empty first line to keep want string neat.
	l.Infof("")
	l.Infof("Line 1")
	l.Infof("Line 2")
	l.Infof("Line 3")
	l.Infof("Line 4")
	l.Infof("Line 5")
	assert.StringsEqual(t, buf.String(), want)
}