test: add test rotate interval

qazwsxedckll · Mar 6, 2024 · 731b5d2 · 731b5d2
1 parent ce19768
commit 731b5d2
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/rotate_file_test.go b/rotate_file_test.go
@@ -3,6 +3,7 @@ package logh
 import (
 	"os"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/require"
 )
@@ -23,7 +24,7 @@ func TestNewRotateFile(t *testing.T) {
 }
 
 func TestRotateSize(t *testing.T) {
-	path, err := os.MkdirTemp("", "loghtest-TestWrite")
+	path, err := os.MkdirTemp("", "loghtest-TestRotateSize")
 	require.NoError(t, err)
 	defer os.RemoveAll(path)
 
@@ -39,3 +40,31 @@ func TestRotateSize(t *testing.T) {
 	require.NoError(t, err)
 	require.Len(t, dir, 3)
 }
+
+func TestRotateInterval(t *testing.T) {
+	path, err := os.MkdirTemp("", "loghtest-TestRotateInterval")
+	require.NoError(t, err)
+	defer os.RemoveAll(path)
+
+	file, err := NewRotateFile(path, "test", 1024, WithCheckEveryN(2), WithRotateInterval(1*time.Second))
+	require.NoError(t, err)
+
+	_, err = file.Write([]byte("1"))
+	require.NoError(t, err)
+	_, err = file.Write([]byte("2"))
+	require.NoError(t, err)
+
+	dir, err := os.ReadDir(path)
+	require.NoError(t, err)
+	require.Len(t, dir, 1)
+
+	time.Sleep(1 * time.Second)
+	_, err = file.Write([]byte("1"))
+	require.NoError(t, err)
+	_, err = file.Write([]byte("2"))
+	require.NoError(t, err)
+
+	dir, err = os.ReadDir(path)
+	require.NoError(t, err)
+	require.Len(t, dir, 2)
+}