From 56dc279794f0a4986c7cf5a5b1aee030ce0cb7af Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Matu=CC=81s=CC=8C=20Tomlein?= <matus.tomlein@gmail.com>
Date: Wed, 17 Jan 2024 17:43:05 +0100
Subject: [PATCH] Improve database test

---
 Tests/Storage/TestDatabase.swift | 31 +++++++++++++++++++------------
 1 file changed, 19 insertions(+), 12 deletions(-)

diff --git a/Tests/Storage/TestDatabase.swift b/Tests/Storage/TestDatabase.swift
index ce7eec8d4..ace9b8379 100644
--- a/Tests/Storage/TestDatabase.swift
+++ b/Tests/Storage/TestDatabase.swift
@@ -114,25 +114,32 @@ class TestDatabase: XCTestCase {
     func testRemoveOldEventsByAge() {
         let db = createDatabase("db")
         
-        db.insertRow(["test": 1])
-        Thread.sleep(forTimeInterval: 1)
-        db.insertRow(["test": 2])
-        Thread.sleep(forTimeInterval: 1)
+        for i in 1...5 {
+            db.insertRow(["test": i])
+        }
+        
+        Thread.sleep(forTimeInterval: 2)
+        
+        for i in 6...10 {
+            db.insertRow(["test": i])
+        }
+        
         db.removeOldEvents(maxSize: 5, maxAge: 1)
         
-        let rows = db.readRows(numRows: 5)
-        XCTAssertEqual(rows.count, 1)
-        XCTAssertEqual(rows.first?.data["test"] as? Int, 2)
+        let rows = db.readRows(numRows: 10)
+        XCTAssertEqual(rows.count, 5)
+        XCTAssertEqual(
+            rows.map { $0.data["test"] as! Int }.min(),
+            6
+        )
     }
     
     func testRemoveOldestEventsByMaxSize() {
         let db = createDatabase("db")
         
-        db.insertRow(["test": 1])
-        db.insertRow(["test": 2])
-        db.insertRow(["test": 3])
-        db.insertRow(["test": 4])
-        db.insertRow(["test": 5])
+        for i in 1...5 {
+            db.insertRow(["test": i])
+        }
         db.removeOldEvents(maxSize: 3, maxAge: 5)
         
         let rows = db.readRows(numRows: 5)