From c4d9ca97c6f8a4bc1653b72dad2e50fa48bece88 Mon Sep 17 00:00:00 2001
From: michaelvlach <vlach.michael@gmail.com>
Date: Sat, 27 Aug 2022 18:26:36 +0200
Subject: [PATCH 1/2] Update file_records.rs

---
 src/storage/file_records.rs | 67 ++++++++++++++++++++++++-------------
 1 file changed, 43 insertions(+), 24 deletions(-)

diff --git a/src/storage/file_records.rs b/src/storage/file_records.rs
index edde2b572..e24e5c6b3 100644
--- a/src/storage/file_records.rs
+++ b/src/storage/file_records.rs
@@ -63,30 +63,6 @@ impl From<Vec<FileRecord>> for FileRecords {
 mod tests {
     use super::*;
 
-    #[test]
-    fn default_constructed() {
-        let _records = FileRecords::default();
-    }
-
-    #[test]
-    fn get() {
-        let mut file_records = FileRecords::default();
-        let position = 32_u64;
-        let size = 64_u64;
-
-        file_records.create(position, size);
-        let expected_record = FileRecord {
-            index: 0,
-            position,
-            size,
-        };
-
-        assert_eq!(
-            file_records.get(expected_record.index),
-            Some(&expected_record)
-        );
-    }
-
     #[test]
     fn create() {
         let mut file_records = FileRecords::default();
@@ -103,6 +79,11 @@ mod tests {
         assert_eq!(actual_record, expected_record);
     }
 
+    #[test]
+    fn default_constructed() {
+        let _records = FileRecords::default();
+    }
+
     #[test]
     fn from_records() {
         let record1 = FileRecord {
@@ -158,6 +139,44 @@ mod tests {
         assert_eq!(new_record3.index, 5);
     }
 
+    #[test]
+    fn get() {
+        let mut file_records = FileRecords::default();
+        let position = 32_u64;
+        let size = 64_u64;
+
+        file_records.create(position, size);
+        let expected_record = FileRecord {
+            index: 0,
+            position,
+            size,
+        };
+
+        assert_eq!(
+            file_records.get(expected_record.index),
+            Some(&expected_record)
+        );
+    }
+
+    #[test]
+    fn get_mut() {
+        let mut file_records = FileRecords::default();
+        let position = 32_u64;
+        let size = 64_u64;
+
+        file_records.create(position, size);
+        let mut expected_record = FileRecord {
+            index: 0,
+            position,
+            size,
+        };
+
+        assert_eq!(
+            file_records.get_mut(expected_record.index),
+            Some(&mut expected_record)
+        );
+    }
+
     #[test]
     fn remove() {
         let mut file_records = FileRecords::default();

From 7a9fcf64526fa069cfe040f088ae78452877a22a Mon Sep 17 00:00:00 2001
From: michaelvlach <vlach.michael@gmail.com>
Date: Sat, 27 Aug 2022 18:26:38 +0200
Subject: [PATCH 2/2] Update file_records.rs

---
 src/storage/file_records.rs | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/storage/file_records.rs b/src/storage/file_records.rs
index e24e5c6b3..d747f4218 100644
--- a/src/storage/file_records.rs
+++ b/src/storage/file_records.rs
@@ -32,6 +32,10 @@ impl FileRecords {
         self.records.get(&index)
     }
 
+    pub(crate) fn get_mut(&mut self, index: i64) -> Option<&mut FileRecord> {
+        self.records.get_mut(&index)
+    }
+
     pub(crate) fn remove(&mut self, index: i64) {
         self.records.remove(&index);
         self.free_list.push(index);