Add mod_caps checks to the testing suite

diff --git a/test/ejabberd_SUITE.erl b/test/ejabberd_SUITE.erl
index f732d881a6b6b24ee15f4bc6aed1b374c2033ab4..a504acfc568d149b4a9263653276d9fc771f8af1 100644 (file)
--- a/test/ejabberd_SUITE.erl
+++ b/test/ejabberd_SUITE.erl
@@ -400,11 +400,46 @@ presence(Config) ->
     disconnect(Config).
 
 presence_broadcast(Config) ->
-    send(Config, #presence{}),
+    Feature = <<"p1:tmp:", (randoms:get_string())/binary>>,
+    Ver = crypto:sha(["client", $/, "bot", $/, "en", $/,
+                     "ejabberd_ct", $<, Feature, $<]),
+    B64Ver = base64:encode(Ver),
+    Node = <<(?EJABBERD_CT_URI)/binary, $#, B64Ver/binary>>,
+    Server = ?config(server, Config),
+    ServerJID = server_jid(Config),
+    Info = #disco_info{identities =
+                          [#identity{category = <<"client">>,
+                                     type = <<"bot">>,
+                                     lang = <<"en">>,
+                                     name = <<"ejabberd_ct">>}],
+                      node = Node, features = [Feature]},
+    Caps = #caps{hash = <<"sha-1">>, node = ?EJABBERD_CT_URI, ver = Ver},
+    send(Config, #presence{sub_els = [Caps]}),
     JID = my_jid(Config),
-    %% We receive the welcome message and the presence broadcast
-    ?recv2(#message{type = normal},
-           #presence{from = JID, to = JID}),
+    %% We receive:
+    %% 1) disco#info iq request for CAPS
+    %% 2) welcome message
+    %% 3) presence broadcast
+    {IQ, _, _} = ?recv3(#iq{type = get,
+                           from = ServerJID,
+                           sub_els = [#disco_info{node = Node}]},
+                       #message{type = normal},
+                       #presence{from = JID, to = JID}),
+    send(Config, #iq{type = result, id = IQ#iq.id,
+                    to = ServerJID, sub_els = [Info]}),
+    %% We're trying to read our feature from ejabberd database
+    %% with exponential back-off as our IQ response may be delayed.
+    [Feature] =
+       lists:foldl(
+         fun(Time, []) ->
+                 timer:sleep(Time),
+                 mod_caps:get_features(
+                   Server,
+                   mod_caps:read_caps(
+                     [xmpp_codec:encode(Caps)]));
+            (_, Acc) ->
+                 Acc
+         end, [], [0, 100, 200, 2000, 5000, 10000]),
     disconnect(Config).
 
 ping(Config) ->

diff --git a/test/suite.hrl b/test/suite.hrl
index ddd751e23f1492fb4888c61429fb60f56070e550..49bfff2d9e857ed4fec5239f5a3879f5443d98a4 100644 (file)
--- a/test/suite.hrl
+++ b/test/suite.hrl
@@ -15,6 +15,8 @@
 
 -define(PUBSUB(Node), <<(?NS_PUBSUB)/binary, "#", Node>>).
 
+-define(EJABBERD_CT_URI, <<"http://www.process-one.net/en/ejabberd_ct/">>).
+
 -define(recv2(P1, P2),
         (fun() ->
                  case {R1 = recv(), R2 = recv()} of