manager_test.go

package rollingwriter

import (
	"path"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func newTimeManager() manager {
	return manager{}
}
func newSizeManager() manager {
	return manager{}
}

func TestParseVolume(t *testing.T) {
	c := &Config{}
	m := manager{}

	c.RollingVolumeSize = "1kb"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024), m.thresholdSize)
	c.RollingVolumeSize = "2k"
	m.ParseVolume(c)
	assert.Equal(t, int64(2*1024), m.thresholdSize)
	c.RollingVolumeSize = "1KB"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024), m.thresholdSize)
	c.RollingVolumeSize = "1mb"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1MB"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1Mb"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1gb"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1GB"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1g"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1tb"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024*1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1tB"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024*1024*1024), m.thresholdSize)
	c.RollingVolumeSize = "1t"
	m.ParseVolume(c)
	assert.Equal(t, int64(1024*1024*1024*1024), m.thresholdSize)
}

func TestGenLogFileName(t *testing.T) {
	m := manager{}
	c := &Config{
		LogPath:       "./",
		FileName:      "file",
		TimeTagFormat: "200601021504",
	}
	m.startAt = time.Now()

	dest := m.GenLogFileName(c)
	timetag := m.startAt.Format(c.TimeTagFormat)
	assert.Equal(t, path.Join("./", "file"+".log."+timetag), dest)

	c.Compress = true
	dest = m.GenLogFileName(c)
	timetag = m.startAt.Format(c.TimeTagFormat)
	assert.Equal(t, path.Join("./", "file"+".log.gz."+timetag), dest)
}