From 6a2ee998ccebf4b52fbf75657058aff3182d45f5 Mon Sep 17 00:00:00 2001
From: "Brian L. Troutwine" <brian.troutwine@datadoghq.com>
Date: Tue, 29 Oct 2024 11:47:11 -0700
Subject: [PATCH] Simulate all ticks, not just some

This commit adjusts the model to simulate every tick between the previous
moment time was advanced and the current moment. This avoids errors where
files grow well past the size they should -- demonstrated now with an
additional model property.

Signed-off-by: Brian L. Troutwine <brian.troutwine@datadoghq.com>
---
 .../generator/file_gen/logrotate_fs/model.rs  | 35 ++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/lading/src/generator/file_gen/logrotate_fs/model.rs b/lading/src/generator/file_gen/logrotate_fs/model.rs
index 3b03ddcef..040c4dc2b 100644
--- a/lading/src/generator/file_gen/logrotate_fs/model.rs
+++ b/lading/src/generator/file_gen/logrotate_fs/model.rs
@@ -521,8 +521,22 @@ impl State {
     /// advance.
     pub(crate) fn advance_time(&mut self, now: Tick) {
         assert!(now >= self.now);
-        let mut inodes: Vec<Inode> = self.nodes.keys().copied().collect();
+        // We have to simulate all ticks that happen between self.now and now,
+        // else an observer will detect a sudden shift in the model. This is
+        // still possible by careful observation of the metadata for files
+        // updating only on tick boundaries, but it's better than shifts only
+        // when observation happens.
+        while self.now < now {
+            self.now += 1;
+            self.advance_time_inner(self.now);
+        }
+    }
 
+    #[inline]
+    fn advance_time_inner(&mut self, now: Tick) {
+        assert!(now >= self.now);
+
+        let mut inodes: Vec<Inode> = self.nodes.keys().copied().collect();
         for inode in inodes.drain(..) {
             let (rotated_inode, parent_inode, bytes_per_tick, group_id, ordinal) = {
                 // If the node pointed to by inode doesn't exist, that's a
@@ -1143,6 +1157,25 @@ mod test {
                 );
             }
         }
+
+        // Property 8: max(bytes_written) <= max_bytes_per_file + bytes_per_second
+        //
+        // If just prior to a rollover the file is within bytes_per_second of
+        // max_bytes_per_file on the next tick that the rollover happens the
+        // file will be larger than max_bytes_per_file but to a limited degree.
+        for node in state.nodes.values() {
+            if let Node::File { file } = node {
+                if file.unlinked {
+                    continue;
+                }
+                let max_size = state.max_bytes_per_file + file.bytes_per_tick;
+                assert!(
+                    file.size() <= max_size,
+                    "File size {sz} exceeds max allowed size {max_size}",
+                    sz = file.size()
+                );
+            }
+        }
     }
 
     proptest! {