[apel#238] Add storage record type for logger

William-Brown5515 · Sep 17, 2024 · d59807a · d59807a
1 parent ef5571c
commit d59807a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/apel/db/loader/loader.py b/apel/db/loader/loader.py
@@ -204,7 +204,9 @@ def load_msg(self, msg_text, signer):
                     apel.db.records.sync.SyncRecord: 'Sync',
                     apel.db.records.cloud.CloudRecord: 'Cloud',
                     apel.db.records.cloud_summary.CloudSummaryRecord:
-                    'Cloud Summary'}
+                    'Cloud Summary',
+                    apel.db.records.storage.StorageRecord: 'Storage',
+                    }
 
         log.info('Loading message from %s', signer)
 

diff --git a/test/test_loader.py b/test/test_loader.py
@@ -178,7 +178,7 @@ def test_load_all_other_type(self):
             self.loader.load_all_msgs()
             self.mock_db.load_records.assert_called_once()
             mock_log.assert_has_calls(
-                        [call('Message contains %i records', 1)])
+                        [call('Message contains 1 %s record', 'Storage')])
 
     def tearDown(self):
         shutil.rmtree(self.dir_path)