From 3186fb42e71e4a69b18e59233c05348172a890e1 Mon Sep 17 00:00:00 2001
From: Alex Clarke <alex.j.tusa@gmail.com>
Date: Mon, 2 Dec 2024 16:39:40 -0700
Subject: [PATCH] feat(handler): Wired in the blocklist handler to the main
 handlers

---
 src/handlers/sonarr_handlers/mod.rs            |  4 ++++
 .../sonarr_handlers/sonarr_handler_tests.rs    | 18 ++++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/src/handlers/sonarr_handlers/mod.rs b/src/handlers/sonarr_handlers/mod.rs
index fd0b895..18f36fd 100644
--- a/src/handlers/sonarr_handlers/mod.rs
+++ b/src/handlers/sonarr_handlers/mod.rs
@@ -1,3 +1,4 @@
+use blocklist::BlocklistHandler;
 use downloads::DownloadsHandler;
 use library::LibraryHandler;
 
@@ -37,6 +38,9 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SonarrHandler<'a, 'b
       _ if DownloadsHandler::accepts(self.active_sonarr_block) => {
         DownloadsHandler::with(self.key, self.app, self.active_sonarr_block, self.context).handle()
       }
+      _ if BlocklistHandler::accepts(self.active_sonarr_block) => {
+        BlocklistHandler::with(self.key, self.app, self.active_sonarr_block, self.context).handle()
+      }
       _ => self.handle_key_event(),
     }
   }
diff --git a/src/handlers/sonarr_handlers/sonarr_handler_tests.rs b/src/handlers/sonarr_handlers/sonarr_handler_tests.rs
index e9c21a4..99d438c 100644
--- a/src/handlers/sonarr_handlers/sonarr_handler_tests.rs
+++ b/src/handlers/sonarr_handlers/sonarr_handler_tests.rs
@@ -127,4 +127,22 @@ mod tests {
       active_sonarr_block
     );
   }
+
+  #[rstest]
+  fn test_delegates_blocklist_blocks_to_blocklist_handler(
+    #[values(
+      ActiveSonarrBlock::Blocklist,
+      ActiveSonarrBlock::BlocklistItemDetails,
+      ActiveSonarrBlock::DeleteBlocklistItemPrompt,
+      ActiveSonarrBlock::BlocklistClearAllItemsPrompt,
+      ActiveSonarrBlock::BlocklistSortPrompt
+    )]
+    active_sonarr_block: ActiveSonarrBlock,
+  ) {
+    test_handler_delegation!(
+      SonarrHandler,
+      ActiveSonarrBlock::Blocklist,
+      active_sonarr_block
+    );
+  }
 }