tests: add tests for DatabaseKeyRepository

Author: josecelano

packages/tracker-core/src/authentication/key/repository/persisted.rs

@@ -46,3 +46,76 @@ impl DatabaseKeyRepository {
         Ok(keys)
     }
 }
+
+#[cfg(test)]
+mod tests {
+
+    mod the_persisted_keys_repository_should {
+
+        use std::time::Duration;
+
+        use torrust_tracker_test_helpers::configuration;
+
+        use crate::authentication::key::repository::persisted::DatabaseKeyRepository;
+        use crate::authentication::{Key, PeerKey};
+        use crate::databases::setup::initialize_database;
+
+        #[test]
+        fn persist_a_new_peer_key() {
+            let configuration = configuration::ephemeral_public();
+
+            let database = initialize_database(&configuration);
+
+            let repository = DatabaseKeyRepository::new(&database);
+
+            let peer_key = PeerKey {
+                key: Key::new("YZSl4lMZupRuOpSRC3krIKR5BPB14nrJ").unwrap(),
+                valid_until: Some(Duration::new(9999, 0)),
+            };
+
+            let result = repository.add(&peer_key);
+
+            assert!(result.is_ok());
+        }
+
+        #[test]
+        fn remove_a_persisted_peer_key() {
+            let configuration = configuration::ephemeral_public();
+
+            let database = initialize_database(&configuration);
+
+            let repository = DatabaseKeyRepository::new(&database);
+
+            let peer_key = PeerKey {
+                key: Key::new("YZSl4lMZupRuOpSRC3krIKR5BPB14nrJ").unwrap(),
+                valid_until: Some(Duration::new(9999, 0)),
+            };
+
+            let _unused = repository.add(&peer_key);
+
+            let result = repository.remove(&peer_key.key);
+
+            assert!(result.is_ok());
+        }
+
+        #[test]
+        fn load_all_persisted_peer_keys() {
+            let configuration = configuration::ephemeral_public();
+
+            let database = initialize_database(&configuration);
+
+            let repository = DatabaseKeyRepository::new(&database);
+
+            let peer_key = PeerKey {
+                key: Key::new("YZSl4lMZupRuOpSRC3krIKR5BPB14nrJ").unwrap(),
+                valid_until: Some(Duration::new(9999, 0)),
+            };
+
+            let _unused = repository.add(&peer_key);
+
+            let keys = repository.load_keys().unwrap();
+
+            assert_eq!(keys, vec!(peer_key));
+        }
+    }
+}