Adjust software updates event handler tests

trento-project · Apr 18, 2024 · f930a0d · f930a0d
1 parent 9cb4ddc
commit f930a0d
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/...infrastructure/commanded/event_handlers/software_updates_discovery_event_handler_test.exs b/...infrastructure/commanded/event_handlers/software_updates_discovery_event_handler_test.exs
@@ -11,6 +11,8 @@ defmodule Trento.Infrastructure.Commanded.EventHandlers.SoftwareUpdatesDiscovery
 
   alias Trento.Infrastructure.Commanded.EventHandlers.SoftwareUpdatesDiscoveryEventHandler
 
+  require Trento.SoftwareUpdates.Enums.SoftwareUpdatesHealth, as: SoftwareUpdatesHealth
+
   setup [:set_mox_from_context, :verify_on_exit!]
 
   test "should discover software updates when a SoftwareUpdatesDiscoveryRequested is emitted" do
@@ -63,8 +65,11 @@ defmodule Trento.Infrastructure.Commanded.EventHandlers.SoftwareUpdatesDiscovery
     expect(
       Trento.Commanded.Mock,
       :dispatch,
-      0,
-      fn _ -> :ok end
+      fn %CompleteSoftwareUpdatesDiscovery{
+           health: SoftwareUpdatesHealth.unknown()
+         } ->
+        :ok
+      end
     )
 
     assert :ok = SoftwareUpdatesDiscoveryEventHandler.handle(event, %{})