40 lines
639 B
Go
40 lines
639 B
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestCreateLogger(t *testing.T) {
|
|
|
|
// flags
|
|
{
|
|
logger := loggerConfig{LogFlags: 2}.createLogger()
|
|
|
|
if logger.Flags() != 2 {
|
|
t.Errorf("flags not set")
|
|
}
|
|
}
|
|
|
|
// file
|
|
{
|
|
tmpfile, err := ioutil.TempFile("", "sshpiperlog")
|
|
if err != nil {
|
|
t.Errorf("failed to create tmp file %v", err)
|
|
}
|
|
defer os.Remove(tmpfile.Name()) // clean up
|
|
|
|
logger := loggerConfig{LogFile: tmpfile.Name()}.createLogger()
|
|
|
|
logger.Print("test123")
|
|
|
|
s, _ := ioutil.ReadFile(tmpfile.Name())
|
|
|
|
if !strings.Contains(string(s), "test123") {
|
|
|
|
t.Errorf("log failed")
|
|
}
|
|
}
|
|
}
|