150 lines
3.7 KiB
Go
150 lines
3.7 KiB
Go
package logging
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hashicorp/consul/sdk/testutil"
|
|
)
|
|
|
|
const (
|
|
testFileName = "Consul.log"
|
|
testDuration = 2 * time.Second
|
|
testBytes = 10
|
|
)
|
|
|
|
func TestLogFile_timeRotation(t *testing.T) {
|
|
t.Parallel()
|
|
tempDir := testutil.TempDir(t, "LogWriterTime")
|
|
defer os.Remove(tempDir)
|
|
logFile := LogFile{
|
|
fileName: testFileName,
|
|
logPath: tempDir,
|
|
duration: testDuration,
|
|
}
|
|
logFile.Write([]byte("Hello World"))
|
|
time.Sleep(2 * time.Second)
|
|
logFile.Write([]byte("Second File"))
|
|
want := 2
|
|
if got, _ := ioutil.ReadDir(tempDir); len(got) != want {
|
|
t.Errorf("Expected %d files, got %v file(s)", want, len(got))
|
|
}
|
|
}
|
|
|
|
func TestLogFile_openNew(t *testing.T) {
|
|
t.Parallel()
|
|
tempDir := testutil.TempDir(t, "LogWriterOpen")
|
|
defer os.Remove(tempDir)
|
|
logFile := LogFile{fileName: testFileName, logPath: tempDir, duration: testDuration}
|
|
if err := logFile.openNew(); err != nil {
|
|
t.Errorf("Expected open file %s, got an error (%s)", testFileName, err)
|
|
}
|
|
|
|
if _, err := ioutil.ReadFile(logFile.FileInfo.Name()); err != nil {
|
|
t.Errorf("Expected readable file %s, got an error (%s)", logFile.FileInfo.Name(), err)
|
|
}
|
|
}
|
|
|
|
func TestLogFile_byteRotation(t *testing.T) {
|
|
t.Parallel()
|
|
tempDir := testutil.TempDir(t, "LogWriterBytes")
|
|
defer os.Remove(tempDir)
|
|
logFile := LogFile{
|
|
fileName: testFileName,
|
|
logPath: tempDir,
|
|
MaxBytes: testBytes,
|
|
duration: 24 * time.Hour,
|
|
}
|
|
logFile.Write([]byte("Hello World"))
|
|
logFile.Write([]byte("Second File"))
|
|
want := 2
|
|
tempFiles, _ := ioutil.ReadDir(tempDir)
|
|
if got := tempFiles; len(got) != want {
|
|
t.Errorf("Expected %d files, got %v file(s)", want, len(got))
|
|
}
|
|
}
|
|
|
|
func TestLogFile_deleteArchives(t *testing.T) {
|
|
t.Parallel()
|
|
tempDir := testutil.TempDir(t, "LogWriteDeleteArchives")
|
|
defer os.Remove(tempDir)
|
|
logFile := LogFile{
|
|
fileName: testFileName,
|
|
logPath: tempDir,
|
|
MaxBytes: testBytes,
|
|
duration: 24 * time.Hour,
|
|
MaxFiles: 1,
|
|
}
|
|
logFile.Write([]byte("[INFO] Hello World"))
|
|
logFile.Write([]byte("[INFO] Second File"))
|
|
logFile.Write([]byte("[INFO] Third File"))
|
|
want := 2
|
|
tempFiles, _ := ioutil.ReadDir(tempDir)
|
|
if got := tempFiles; len(got) != want {
|
|
t.Errorf("Expected %d files, got %v file(s)", want, len(got))
|
|
return
|
|
}
|
|
for _, tempFile := range tempFiles {
|
|
var bytes []byte
|
|
var err error
|
|
path := filepath.Join(tempDir, tempFile.Name())
|
|
if bytes, err = ioutil.ReadFile(path); err != nil {
|
|
t.Errorf(err.Error())
|
|
return
|
|
}
|
|
contents := string(bytes)
|
|
|
|
if contents == "[INFO] Hello World" {
|
|
t.Errorf("Should have deleted the eldest log file")
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestLogFile_deleteArchivesDisabled(t *testing.T) {
|
|
t.Parallel()
|
|
tempDir := testutil.TempDir(t, t.Name())
|
|
defer os.Remove(tempDir)
|
|
logFile := LogFile{
|
|
fileName: testFileName,
|
|
logPath: tempDir,
|
|
MaxBytes: testBytes,
|
|
duration: 24 * time.Hour,
|
|
MaxFiles: 0,
|
|
}
|
|
logFile.Write([]byte("[INFO] Hello World"))
|
|
logFile.Write([]byte("[INFO] Second File"))
|
|
logFile.Write([]byte("[INFO] Third File"))
|
|
want := 3
|
|
tempFiles, _ := ioutil.ReadDir(tempDir)
|
|
if got := tempFiles; len(got) != want {
|
|
t.Errorf("Expected %d files, got %v file(s)", want, len(got))
|
|
return
|
|
}
|
|
}
|
|
|
|
func TestLogFile_rotationDisabled(t *testing.T) {
|
|
t.Parallel()
|
|
tempDir := testutil.TempDir(t, t.Name())
|
|
defer os.Remove(tempDir)
|
|
logFile := LogFile{
|
|
fileName: testFileName,
|
|
logPath: tempDir,
|
|
MaxBytes: testBytes,
|
|
duration: 24 * time.Hour,
|
|
MaxFiles: -1,
|
|
}
|
|
logFile.Write([]byte("[INFO] Hello World"))
|
|
logFile.Write([]byte("[INFO] Second File"))
|
|
logFile.Write([]byte("[INFO] Third File"))
|
|
want := 1
|
|
tempFiles, _ := ioutil.ReadDir(tempDir)
|
|
if got := tempFiles; len(got) != want {
|
|
t.Errorf("Expected %d files, got %v file(s)", want, len(got))
|
|
return
|
|
}
|
|
}
|