From a7a19abccf0a9618bca336be7818f59edd134190 Mon Sep 17 00:00:00 2001
From: Ryan Zhao <ryanzhaors@qq.com>
Date: Mon, 14 Aug 2023 14:39:29 +1000
Subject: [PATCH] update unit test

---
 .../Message Handling/MessageReceiver+ExpirationTimers.swift  | 2 +-
 .../MessageReceiverDisappearingMessagesSpec.swift            | 5 ++---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift b/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift
index c5eb784c0..370ddae08 100644
--- a/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift	
+++ b/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift	
@@ -139,7 +139,7 @@ extension MessageReceiver {
         ).inserted(db)
     }
     
-    internal static func updateDisappearingMessagesConfigurationIfNeeded(
+    public static func updateDisappearingMessagesConfigurationIfNeeded(
         _ db: Database,
         threadId: String,
         threadVariant: SessionThread.Variant,
diff --git a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
index f0996cb0a..0404f8dfa 100644
--- a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
+++ b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
@@ -107,13 +107,12 @@ class MessageReceiverDisappearingMessagesSpec: QuickSpec {
                 // MARK: -- updates the local config properly
                 it("updates the local config properly") {
                     mockStorage.write { db in
-                        try MessageReceiver.handle(
+                        try MessageReceiver.updateDisappearingMessagesConfigurationIfNeeded(
                             db,
                             threadId: "TestId",
                             threadVariant: .contact,
                             message: mockMessage,
-                            serverExpirationTimestamp: nil,
-                            associatedWithProto: mockProto
+                            proto: mockProto
                         )
                     }