pkg_test.go

package file_rotator

import (
	"log"
	"os"
	"testing"
	"time"
)

var (
	filePath = "./app.log"
)

func TestDoRotate(t *testing.T) {
	fw, err := NewFileRotator(filePath)
	if err != nil {
		log.Printf("new rotator failed failed: %s", err)
		t.FailNow()
	}
	err = fw.doRotate()
	if err != nil {
		log.Printf("rotate failed: %s", err)
		t.FailNow()
	}
	time.Sleep(2 * time.Second)

}

func TestDoRotateFileNotFound(t *testing.T) {
	fw, err := NewFileRotator(filePath)
	if err != nil {
		log.Printf("new rotator failed: %s", err)
		t.FailNow()
	}
	os.Remove(filePath)
	err = fw.doRotate()
	if err != nil {
		log.Printf("rotate failed: %s", err)
		t.FailNow()
	}
	time.Sleep(2 * time.Second)

}

func TestDeleteOldFile(t *testing.T) {
	fw, err := NewFileRotator(filePath)
	if err != nil {
		log.Printf("new rotator failed: %s", err)
		t.FailNow()
	}
	fw.deleteOldFiles()

}