From e1f5f86743fc14155cf621a68484f9eaed5289ca Mon Sep 17 00:00:00 2001
From: Ryan Zhao <ryanzhaors@qq.com>
Date: Mon, 14 Aug 2023 15:46:12 +1000
Subject: [PATCH] try to fix unit test

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

diff --git a/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift b/SessionMessagingKit/Sending & Receiving/Message Handling/MessageReceiver+ExpirationTimers.swift
index 370ddae08..c5eb784c0 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)
     }
     
-    public static func updateDisappearingMessagesConfigurationIfNeeded(
+    internal static func updateDisappearingMessagesConfigurationIfNeeded(
         _ db: Database,
         threadId: String,
         threadVariant: SessionThread.Variant,
diff --git a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
index fa2bd65f8..391c78f25 100644
--- a/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
+++ b/SessionTests/MessagingKit/MessageReceiverDisappearingMessagesSpec.swift
@@ -108,12 +108,13 @@ class MessageReceiverDisappearingMessagesSpec: QuickSpec {
                 it("updates the local config properly") {
                     mockStorage.write { db in
                         do {
-                            try MessageReceiver.updateDisappearingMessagesConfigurationIfNeeded(
+                            try MessageReceiver.handle(
                                 db,
                                 threadId: "TestId",
                                 threadVariant: .contact,
                                 message: mockMessage,
-                                proto: mockProto
+                                serverExpirationTimestamp: nil,
+                                associatedWithProto: mockProto
                             )
                         }
                         catch {