Update onstartup.rs

src/append/rolling_file/policy/compound/trigger/onstartup.rs

@@ -42,13 +42,13 @@ impl OnStartUpTrigger {
 
 impl Trigger for OnStartUpTrigger {
     fn trigger(&self, file: &LogFile) -> anyhow::Result<bool> {
-        if !self.initial.is_completed() {
-            self.initial.call_once(|| {});
+        let mut result = false;
+        self.initial.call_once(|| {
             if file.len_estimate() >= self.min_size {
-                return Ok(true);
+                result = true;
             }
-        }
-        Ok(false)
+        });
+        Ok(result)
     }
 
     fn is_pre_process(&self) -> bool {
@@ -94,7 +94,7 @@ mod test {
     }
 
     #[test]
-    fn trigger1() {
+    fn trigger() {
         let file = tempfile::tempdir().unwrap();
         let logfile = LogFile {
             writer: &mut None,
@@ -104,23 +104,13 @@ mod test {
 
         let trigger = OnStartUpTrigger::new(1);
         let result = trigger.trigger(&logfile).unwrap();
-        assert_eq!(result, false);
-    }
-
-    #[test]
-    fn trigger2() {
-        let file = tempfile::tempdir().unwrap();
-        let logfile = LogFile {
-            writer: &mut None,
-            path: file.path(),
-            len: 0,
-        };
+        assert_eq!(result, false); // It should not be triggered with the min_size 1.
 
         let trigger = OnStartUpTrigger::new(0);
         let result = trigger.trigger(&logfile).unwrap();
-        assert_eq!(result, true);
+        assert_eq!(result, true); // First time, it should be triggered.
 
         let result = trigger.trigger(&logfile).unwrap();
-        assert_eq!(result, false);
+        assert_eq!(result, false); // Second time, it should not be triggered.
     }
 }