Fix tests 3

esl · Nov 7, 2022 · d9565e2 · d9565e2
1 parent 5e69c4c
commit d9565e2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/mam/mod_mam_riak_timed_arch_yz.erl b/src/mam/mod_mam_riak_timed_arch_yz.erl
@@ -177,6 +177,11 @@ lookup_messages_muc(Result, #{params := #{with_jid := WithJID} = Params}, #{host
     Params :: map(),
     Extra :: map().
 archive_size(_Size, #{owner := ArchiveJID}, #{host_type := Host}) ->
+    archive_size(ArchiveJID, Host);
+archive_size(_Size, #{room := ArchiveJID}, #{host_type := Host}) ->
+    archive_size(ArchiveJID, Host).
+
+archive_size(ArchiveJID, Host) ->
     OwnerJID = mod_mam_utils:bare_jid(ArchiveJID),
     RemoteJID = undefined,
     {MsgIdStartNoRSM, MsgIdEndNoRSM} =