test: test rotate by size

qazwsxedckll · Mar 5, 2024 · c51d124 · c51d124
1 parent 73e9681
commit c51d124
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/rotate_file.go b/rotate_file.go
@@ -66,7 +66,7 @@ func (r *RotateFile) logFileName() (string, time.Time) {
 		hostname = "unknownhost"
 	}
 	now := time.Now()
-	return r.filepath + "." + now.Format("20060102-150405") + "." + hostname + "." + fmt.Sprint(os.Getpid()) + ".log", now
+	return r.filepath + "." + now.Format("20060102-150405.000000000") + "." + hostname + "." + fmt.Sprint(os.Getpid()) + ".log", now
 }
 
 func (r *RotateFile) Rotate() {

diff --git a/rotate_file_test.go b/rotate_file_test.go
@@ -10,6 +10,7 @@ import (
 func TestNewRotateFile(t *testing.T) {
 	path, err := os.MkdirTemp("", "loghtest")
 	require.NoError(t, err)
+	defer os.RemoveAll(path)
 
 	path += "/TestNewRotateFile"
 	file, err := NewRotateFile(path, "test", 10)
@@ -19,7 +20,22 @@ func TestNewRotateFile(t *testing.T) {
 	dir, err := os.ReadDir(path)
 	require.NoError(t, err)
 	require.Len(t, dir, 1)
+}
+
+func TestRotateSize(t *testing.T) {
+	path, err := os.MkdirTemp("", "loghtest-TestWrite")
+	require.NoError(t, err)
+	defer os.RemoveAll(path)
+
+	file, err := NewRotateFile(path, "test", 10)
+	require.NoError(t, err)
 
-	err = os.RemoveAll(path)
+	_, err = file.Write([]byte("12345678901234567890"))
+	require.NoError(t, err)
+	_, err = file.Write([]byte("12345678901234567890"))
+	require.NoError(t, err)
+
+	dir, err := os.ReadDir(path)
 	require.NoError(t, err)
+	require.Len(t, dir, 3)
 }