Improve private and MAM tests in order to reflect

database backends with limited capabilities

diff --git a/test/mam_tests.erl b/test/mam_tests.erl
index c152e03fddb76282fc4868a3f6f2414370113227..128df2fe8775fa383356e66645498b8c8d6b3cb1 100644 (file)
--- a/test/mam_tests.erl
+++ b/test/mam_tests.erl
@@ -377,7 +377,7 @@ recv_fin(Config, I, QueryID, NS, IsComplete) when NS == ?NS_MAM_1; NS == ?NS_MAM
                            complete = Complete,
                            rsm = RSM}]} = recv_iq(Config),
     ct:comment("Checking if complete is ~s", [IsComplete]),
-    Complete = IsComplete,
+    ?match(IsComplete, Complete),
     RSM;
 recv_fin(Config, I, QueryID, ?NS_MAM_TMP = NS, _IsComplete) ->
     ct:comment("Receiving fin iq for namespace '~s'", [NS]),
@@ -394,7 +394,7 @@ recv_fin(Config, _, QueryID, ?NS_MAM_0 = NS, IsComplete) ->
             complete = Complete,
             rsm = RSM} = xmpp:get_subtag(FinMsg, #mam_fin{xmlns = NS}),
     ct:comment("Checking if complete is ~s", [IsComplete]),
-    Complete = IsComplete,
+    ?match(IsComplete, Complete),
     RSM.
 
 send_messages_to_room(Config, Range) ->
@@ -417,7 +417,6 @@ recv_messages_from_room(Config, Range) ->
     MyNickJID = jid:replace_resource(Room, MyNick),
     MyJID = my_jid(Config),
     QID = p1_rand:get_string(),
-    Count = length(Range),
     I = send(Config, #iq{type = set, to = Room,
                         sub_els = [#mam_query{xmlns = ?NS_MAM_2, id = QID}]}),
     lists:foreach(
@@ -440,8 +439,9 @@ recv_messages_from_room(Config, Range) ->
     #iq{from = Room, id = I, type = result,
        sub_els = [#mam_fin{xmlns = ?NS_MAM_2,
                            id = QID,
-                           rsm = #rsm_set{count = Count},
-                           complete = true}]} = recv_iq(Config).
+                           rsm = RSM,
+                           complete = true}]} = recv_iq(Config),
+    match_rsm_count(RSM, length(Range)).
 
 query_all(Config, From, To) ->
     lists:foreach(
@@ -454,7 +454,8 @@ query_all(Config, From, To, NS) ->
     Range = lists:seq(1, 5),
     ID = send_query(Config, #mam_query{xmlns = NS, id = QID}),
     recv_archived_messages(Config, From, To, QID, Range),
-    #rsm_set{count = 5} = recv_fin(Config, ID, QID, NS, _Complete = true).
+    RSM = recv_fin(Config, ID, QID, NS, _Complete = true),
+    match_rsm_count(RSM, 5).
 
 query_with(Config, From, To) ->
     lists:foreach(
@@ -480,7 +481,8 @@ query_with(Config, From, To, NS) ->
                      end,
              ID = send_query(Config, Query),
              recv_archived_messages(Config, From, To, QID, Range),
-             #rsm_set{count = 5} = recv_fin(Config, ID, QID, NS, true)
+             RSM = recv_fin(Config, ID, QID, NS, true),
+             match_rsm_count(RSM, 5)
       end, [Peer, BarePeer]).
 
 query_rsm_max(Config, From, To) ->
@@ -498,7 +500,8 @@ query_rsm_max(Config, From, To, NS) ->
              ID = send_query(Config, Query),
              recv_archived_messages(Config, From, To, QID, Range),
              IsComplete = Max >= 5,
-             #rsm_set{count = 5} = recv_fin(Config, ID, QID, NS, IsComplete)
+             RSM = recv_fin(Config, ID, QID, NS, IsComplete),
+             match_rsm_count(RSM, 5)
       end, lists:seq(0, 6)).
 
 query_rsm_after(Config, From, To) ->
@@ -517,8 +520,9 @@ query_rsm_after(Config, From, To, NS) ->
                                 rsm = #rsm_set{'after' = After}},
              ID = send_query(Config, Query),
              recv_archived_messages(Config, From, To, QID, Range),
-             #rsm_set{count = 5, first = First} =
+             RSM = #rsm_set{first = First} =
                  recv_fin(Config, ID, QID, NS, true),
+             match_rsm_count(RSM, 5),
              First
       end, #rsm_first{data = undefined},
       [lists:seq(N, 5) || N <- lists:seq(1, 6)]).
@@ -539,7 +543,15 @@ query_rsm_before(Config, From, To, NS) ->
                                 rsm = #rsm_set{before = Before}},
              ID = send_query(Config, Query),
              recv_archived_messages(Config, From, To, QID, Range),
-             #rsm_set{count = 5, last = Last} =
+             RSM = #rsm_set{last = Last} =
                  recv_fin(Config, ID, QID, NS, true),
+             match_rsm_count(RSM, 5),
              Last
       end, <<"">>, lists:reverse([lists:seq(1, N) || N <- lists:seq(0, 5)])).
+
+match_rsm_count(#rsm_set{count = undefined}, _) ->
+    %% The backend doesn't support counting
+    ok;
+match_rsm_count(#rsm_set{count = Count1}, Count2) ->
+    ct:comment("Checking if RSM 'count' is ~p", [Count2]),
+    ?match(Count2, Count1).

diff --git a/test/private_tests.erl b/test/private_tests.erl
index d430cb4076a2678ab2f769df30ae316b9db9cacb..5ae832b3696f5c1f687511f541a155b80a10ada0 100644 (file)
--- a/test/private_tests.erl
+++ b/test/private_tests.erl
@@ -24,7 +24,7 @@
 
 %% API
 -compile(export_all).
--import(suite, [my_jid/1, is_feature_advertised/3,
+-import(suite, [my_jid/1, server_jid/1, is_feature_advertised/3,
                send_recv/2, disconnect/1]).
 
 -include("suite.hrl").
@@ -43,9 +43,15 @@ single_cases() ->
       single_test(test_published)]}.
 
 test_features(Config) ->
+    Server = jid:encode(server_jid(Config)),
     MyJID = my_jid(Config),
-    true = is_feature_advertised(Config, ?NS_BOOKMARKS_CONVERSION_0,
-                                jid:remove_resource(MyJID)),
+    case gen_mod:is_loaded(Server, mod_pubsub) of
+       true ->
+           true = is_feature_advertised(Config, ?NS_BOOKMARKS_CONVERSION_0,
+                                        jid:remove_resource(MyJID));
+       false ->
+           ok
+    end,
     disconnect(Config).
 
 test_no_namespace(Config) ->
@@ -73,20 +79,26 @@ test_set_get(Config) ->
     disconnect(Config).
 
 test_published(Config) ->
-    Storage = bookmark_storage(),
-    Node = xmpp:get_ns(Storage),
-    #iq{type = result,
-       sub_els = [#pubsub{items = #ps_items{node = Node, items = Items}}]} =
-       send_recv(
-         Config,
-         #iq{type = get,
-             sub_els = [#pubsub{items = #ps_items{node = Node}}]}),
-    [#ps_item{sub_els = [StorageXMLIn]}] = Items,
-    Storage = xmpp:decode(StorageXMLIn),
-    #iq{type = result, sub_els = []} =
-       send_recv(Config,
-                 #iq{type = set,
-                     sub_els = [#pubsub_owner{delete = {Node, <<>>}}]}),
+    Server = jid:encode(server_jid(Config)),
+    case gen_mod:is_loaded(Server, mod_pubsub) of
+       true ->
+           Storage = bookmark_storage(),
+           Node = xmpp:get_ns(Storage),
+           #iq{type = result,
+               sub_els = [#pubsub{items = #ps_items{node = Node, items = Items}}]} =
+               send_recv(
+                 Config,
+                 #iq{type = get,
+                     sub_els = [#pubsub{items = #ps_items{node = Node}}]}),
+           [#ps_item{sub_els = [StorageXMLIn]}] = Items,
+           Storage = xmpp:decode(StorageXMLIn),
+           #iq{type = result, sub_els = []} =
+               send_recv(Config,
+                         #iq{type = set,
+                             sub_els = [#pubsub_owner{delete = {Node, <<>>}}]});
+       false ->
+           ok
+    end,
     disconnect(Config).
 
 %%%===================================================================