From 7d0d968c796ff80bee3f5ee61f4425feff9ec1eb Mon Sep 17 00:00:00 2001
From: "Evgeny @ SimpleX Chat"
 <259188159+evgeny-simplex@users.noreply.github.com>
Date: Wed, 21 Oct 2026 13:47:07 +0000
Subject: [PATCH] better fix

---
 tests/AgentTests/NotificationTests.hs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/AgentTests/NotificationTests.hs b/tests/AgentTests/NotificationTests.hs
index c083492de..accf6723f 100644
--- a/tests/AgentTests/NotificationTests.hs
+++ b/tests/AgentTests/NotificationTests.hs
@@ -332,8 +332,10 @@ testNtfTokenServerRestartReverify t apns = do
     runRight_ $ do
       verification <- ntfData .-> "verification"
       nonce <- C.unsafeCbNonce <$> ntfData .-> "nonce"
-      Left (BROKER _ _) <- tryE $ verifyNtfToken a tkn nonce verification
-      pure ()
+      tryE (verifyNtfToken a tkn nonce verification) >>= \case
+        Left (BROKER _ (NETWORK _)) -> pure ()
+        Left (BROKER _ TIMEOUT) -> pure ()
+        r -> liftIO $ expectationFailure $ "unexpected verification result: " <> show r
   threadDelay 1500000
   withAgent 2 agentCfg initAgentServers testDB $ \a' ->
     -- server stopped before token is verified, so now the attempt to verify it will return AUTH error but re-register token,