1: -module(gdpr_SUITE).
    2: 
    3: %% Tests for features related to GDPR compliance.
    4: 
    5: -include_lib("common_test/include/ct.hrl").
    6: -include_lib("escalus/include/escalus.hrl").
    7: -include_lib("exml/include/exml.hrl").
    8: -include_lib("eunit/include/eunit.hrl").
    9: -include("muc_light.hrl").
   10: 
   11: -export([suite/0, all/0, groups/0]).
   12: -export([init_per_suite/1, end_per_suite/1]).
   13: -export([init_per_group/2, end_per_group/2]).
   14: -export([init_per_testcase/2, end_per_testcase/2]).
   15: -export([
   16:          retrieve_vcard/1,
   17:          remove_vcard/1,
   18:          remove_private/1,
   19:          remove_multiple_private_xmls/1,
   20:          dont_remove_other_user_private_xml/1,
   21:          retrieve_roster/1,
   22:          retrieve_mam_pm/1,
   23:          retrieve_mam_muc/1,
   24:          retrieve_mam_muc_private_msg/1,
   25:          retrieve_mam_muc_store_pm/1,
   26:          remove_mam_pm/1,
   27:          retrieve_mam_muc_light/1,
   28:          retrieve_mam_pm_and_muc_light_interfere/1,
   29:          retrieve_mam_pm_and_muc_light_dont_interfere/1,
   30:          remove_roster/1,
   31:          retrieve_offline/1,
   32:          remove_offline/1,
   33:          retrieve_pubsub_payloads/1,
   34:          retrieve_created_pubsub_nodes/1,
   35:          retrieve_all_pubsub_data/1,
   36:          dont_retrieve_other_user_pubsub_payload/1,
   37:          retrieve_pubsub_subscriptions/1,
   38:          retrieve_private_xml/1,
   39:          dont_retrieve_other_user_private_xml/1,
   40:          retrieve_multiple_private_xmls/1,
   41:          retrieve_inbox/1,
   42:          remove_inbox/1,
   43:          retrieve_inbox_for_multiple_messages/1,
   44:          retrieve_inbox_muclight/1,
   45:          retrieve_inbox_muc/1,
   46:          remove_inbox_muclight/1,
   47:          remove_inbox_muc/1,
   48:          retrieve_logs/1,
   49:          remove_pubsub_all_data/1,
   50:          remove_pubsub_dont_remove_node_when_only_publisher/1,
   51:          remove_pubsub_subscriptions/1,
   52:          remove_pubsub_dont_remove_flat_pubsub_node/1,
   53:          remove_pubsub_push_node/1,
   54:          remove_pubsub_pep_node/1
   55:         ]).
   56: 
   57: -import(mongooseimctl_helper, [mongooseimctl/3]).
   58: -import(distributed_helper, [mim/0, subhost_pattern/1, rpc/4]).
   59: -import(muc_light_helper, [room_bin_jid/1]).
   60: -import(domain_helper, [host_type/0]).
   61: -import(config_parser_helper, [default_mod_config/1, mod_config/2]).
   62: -import(graphql_helper, [execute_command/4, get_ok_value/2]).
   63: 
   64: -define(ROOM, <<"tt1">>).
   65: 
   66: %% -------------------------------------------------------------
   67: %% Common Test stuff
   68: %% -------------------------------------------------------------
   69: 
   70: suite() ->
   71:     escalus:suite().
   72: 
   73: all() ->
   74:     [
   75:      {group, retrieve_personal_data},
   76:      {group, remove_personal_data}
   77:     ].
   78: 
   79: groups() ->
   80:     %% **DON'T** make any of these groups parallel, because calling mongooseimctl
   81:     %% in parallel is broken!
   82:     [
   83:      {retrieve_personal_data, [], [
   84:                                    retrieve_vcard,
   85:                                    retrieve_roster,
   86:                                    retrieve_offline,
   87:                                    retrieve_inbox,
   88:                                    retrieve_logs,
   89:                                    {group, retrieve_personal_data_pubsub},
   90:                                    {group, retrieve_personal_data_private_xml},
   91:                                    {group, retrieve_personal_data_mam},
   92:                                    {group, retrieve_personal_data_inbox}
   93:                                   ]},
   94:     {retrieve_personal_data_inbox, [],[
   95:         retrieve_inbox,
   96:         retrieve_inbox_for_multiple_messages,
   97:         retrieve_inbox_muclight,
   98:         retrieve_inbox_muc
   99:     ]},
  100:      {retrieve_personal_data_pubsub, [], [
  101:                                           retrieve_pubsub_payloads,
  102:                                           dont_retrieve_other_user_pubsub_payload,
  103:                                           retrieve_pubsub_subscriptions,
  104:                                           retrieve_created_pubsub_nodes,
  105:                                           retrieve_all_pubsub_data
  106:                                          ]},
  107:      {retrieve_personal_data_private_xml, [], [
  108:                                                retrieve_private_xml,
  109:                                                dont_retrieve_other_user_private_xml,
  110:                                                retrieve_multiple_private_xmls
  111:                                               ]},
  112:      {retrieve_personal_data_mam, [], [
  113:                                        {group, retrieve_personal_data_mam_rdbms},
  114:                                        {group, retrieve_personal_data_mam_cassandra},
  115:                                        {group, retrieve_personal_data_mam_elasticsearch}
  116:                                       ]},
  117:      {retrieve_personal_data_mam_rdbms, [], all_mam_testcases()},
  118:      {retrieve_personal_data_mam_cassandra, [], all_mam_testcases()},
  119:      {retrieve_personal_data_mam_elasticsearch, [], all_mam_testcases()},
  120:      {remove_personal_data, [], removal_testcases()},
  121:      {remove_personal_data_inbox, [], [remove_inbox, remove_inbox_muclight, remove_inbox_muc]},
  122:      {remove_personal_data_mam, [], [
  123:                                      {group, remove_personal_data_mam_rdbms},
  124:                                      {group, remove_personal_data_mam_cassandra},
  125:                                      {group, remove_personal_data_mam_elasticsearch}
  126:                                     ]},
  127:      {remove_personal_data_mam_rdbms, [], mam_removal_testcases()},
  128:      {remove_personal_data_mam_cassandra, [], mam_removal_testcases()},
  129:      {remove_personal_data_mam_elasticsearch, [], mam_removal_testcases()},
  130:      {remove_personal_data_pubsub, [], [
  131:                                         remove_pubsub_subscriptions,
  132:                                         remove_pubsub_dont_remove_node_when_only_publisher,
  133:                                         remove_pubsub_dont_remove_flat_pubsub_node,
  134:                                         remove_pubsub_push_node,
  135:                                         remove_pubsub_pep_node,
  136:                                         remove_pubsub_all_data
  137:                                        ]}
  138:     ].
  139: 
  140: removal_testcases() ->
  141:     [
  142:         remove_vcard,
  143:         remove_roster,
  144:         remove_offline,
  145:         remove_private,
  146:         remove_multiple_private_xmls,
  147:         dont_remove_other_user_private_xml,
  148:         {group, remove_personal_data_inbox},
  149:         {group, remove_personal_data_pubsub},
  150:         {group, remove_personal_data_mam}
  151:     ].
  152: 
  153: mam_removal_testcases() ->
  154:     [
  155:      remove_mam_pm
  156:     ].
  157: 
  158: 
  159: mam_testcases() ->
  160:     [
  161:         retrieve_mam_pm,
  162:         retrieve_mam_muc_light,
  163:         retrieve_mam_pm_and_muc_light_interfere,
  164:         retrieve_mam_pm_and_muc_light_dont_interfere
  165:     ].
  166: 
  167: all_mam_testcases() ->
  168:     [
  169:         retrieve_mam_muc,
  170:         retrieve_mam_muc_private_msg,
  171:         retrieve_mam_muc_store_pm
  172:         | mam_testcases()
  173:     ].
  174: 
  175: init_per_suite(Config) ->
  176:     #{node := MimNode} = distributed_helper:mim(),
  177:     Config1 = [{{ejabberd_cwd, MimNode}, get_mim_cwd()} | dynamic_modules:save_modules(host_type(), Config)],
  178:     muc_helper:load_muc(),
  179:     Config2 = graphql_helper:init_admin_cli(Config1),
  180:     escalus:init_per_suite(Config2).
  181: 
  182: end_per_suite(Config) ->
  183:     delete_files(),
  184:     escalus_fresh:clean(),
  185:     graphql_helper:clean(),
  186:     dynamic_modules:restore_modules(Config),
  187:     escalus:end_per_suite(Config).
  188: 
  189: init_per_group(GN, Config) when GN =:= remove_personal_data_mam_rdbms;
  190:                                 GN =:= retrieve_personal_data_mam_rdbms ->
  191:     try_backend_for_mam(Config, rdbms);
  192: init_per_group(GN, Config) when GN =:= retrieve_personal_data_pubsub;
  193:                                 GN =:= remove_personal_data_pubsub ->
  194:     [{group, GN} | Config];
  195: init_per_group(GN, Config) when GN =:= retrieve_personal_data_mam_cassandra;
  196:                                 GN =:= remove_personal_data_mam_cassandra->
  197:     try_backend_for_mam(Config, cassandra);
  198: init_per_group(GN, Config) when GN =:= retrieve_personal_data_mam_elasticsearch;
  199:                                 GN =:= remove_personal_data_mam_elasticsearch ->
  200:     try_backend_for_mam(Config, elasticsearch);
  201: init_per_group(retrieve_personal_data_inbox = GN, Config) ->
  202:     init_inbox(GN, Config, muclight);
  203: init_per_group(remove_personal_data_inbox = GN, Config) ->
  204:     init_inbox(GN, Config, muclight);
  205: init_per_group(retrieve_personal_data_private_xml, Config) ->
  206:     private_started(),
  207:     Config;
  208: init_per_group(_GN, Config) ->
  209:     Config.
  210: 
  211: end_per_group(_GN, Config) ->
  212:     Config.
  213: 
  214: try_backend_for_mam(Config, Backend) ->
  215:     case is_backend_enabled(Backend) of
  216:         true -> [{mam_backend, Backend} | Config];
  217:         false -> {skip, backend_is_not_configured}
  218:     end.
  219: 
  220: is_backend_enabled(rdbms)         -> mongoose_helper:is_rdbms_enabled(host_type());
  221: is_backend_enabled(cassandra)     -> mam_helper:is_cassandra_enabled(host_type());
  222: is_backend_enabled(elasticsearch) -> mam_helper:is_elasticsearch_enabled(host_type()).
  223: 
  224: 
  225: init_per_testcase(retrieve_logs = CN, Config) ->
  226:     case is_mim2_started() of
  227:         false -> {skip, not_running_in_distributed};
  228:         _ -> escalus:init_per_testcase(CN, Config)
  229:     end;
  230: init_per_testcase(CN, Config) when CN =:= remove_offline;
  231:                                    CN =:= retrieve_offline ->
  232:     offline_started(),
  233:     escalus:init_per_testcase(CN, Config);
  234: init_per_testcase(CN, Config) when
  235:       CN =:= remove_inbox;
  236:       CN =:= retrieve_inbox;
  237:       CN =:= remove_inbox_muclight;
  238:       CN =:= retrieve_inbox_muclight ->
  239:     Config1 = init_inbox(CN, Config, muclight),
  240:     Config1;
  241: init_per_testcase(CN, Config) when CN =:= retrieve_inbox_muc;
  242:                                    CN =:= remove_inbox_muc ->
  243:     muc_helper:load_muc(),
  244:     Config0 = init_inbox(CN, Config, muc),
  245:     Config0;
  246: 
  247: init_per_testcase(retrieve_vcard = CN, Config) ->
  248:     case vcard_helper:is_vcard_ldap() of
  249:         true ->
  250:             {skip, skipped_for_simplicity_for_now}; % TODO: Fix the case for LDAP as well
  251:         _ ->
  252:             escalus:init_per_testcase(CN, Config)
  253:     end;
  254: init_per_testcase(remove_vcard = CN, Config) ->
  255:     case vcard_helper:is_vcard_ldap() of
  256:         true ->
  257:             {skip, skipped_for_simplicity_for_now}; % TODO: Fix the case for LDAP as well
  258:         _ ->
  259:             vcard_started(),
  260:             escalus:init_per_testcase(CN, Config)
  261:     end;
  262: init_per_testcase(CN, Config) when CN =:= remove_private;
  263:                                    CN =:= dont_remove_other_user_private_xml;
  264:                                    CN =:= remove_multiple_private_xmls ->
  265:     private_started(),
  266:     escalus:init_per_testcase(CN, Config);
  267: 
  268: init_per_testcase(CN, Config) when CN =:= retrieve_mam_muc;
  269:                                    CN =:= retrieve_mam_muc_private_msg;
  270:                                    CN =:= retrieve_mam_muc_store_pm;
  271:                                    CN =:= retrieve_mam_muc_light;
  272:                                    CN =:= retrieve_mam_pm_and_muc_light_interfere;
  273:                                    CN =:= retrieve_mam_pm_and_muc_light_dont_interfere;
  274:                                    CN =:= retrieve_mam_pm;
  275:                                    CN =:= remove_mam_pm ->
  276:     case proplists:get_value(mam_backend, Config, skip) of
  277:         skip ->
  278:             {skip, no_mam_backend_configured};
  279:         Backend ->
  280:             dynamic_modules:restore_modules(Config),
  281:             RequiredModules = mam_required_modules(CN, Backend),
  282:             dynamic_modules:ensure_modules(host_type(), RequiredModules),
  283:             ct:log("required modules: ~p~n", [RequiredModules]),
  284:             escalus:init_per_testcase(CN, [{mam_modules, RequiredModules} | Config])
  285:     end;
  286: init_per_testcase(remove_roster = CN, Config) ->
  287:     roster_started(),
  288:     escalus:init_per_testcase(CN, Config);
  289: init_per_testcase(CN, Config) ->
  290:     GN = proplists:get_value(group, Config),
  291:     IsPubSub = lists:member(GN, [retrieve_personal_data_pubsub, remove_personal_data_pubsub]),
  292:     case IsPubSub of
  293:         true ->
  294:             dynamic_modules:ensure_modules(host_type(), pubsub_required_modules());
  295:         _ ->
  296:             ok
  297:     end,
  298:     escalus:init_per_testcase(CN, Config).
  299: 
  300: 
  301: end_per_testcase(CN, Config) when CN =:= retrieve_mam_muc_light;
  302:                                   CN =:= retrieve_mam_pm_and_muc_light_interfere;
  303:                                   CN =:= retrieve_mam_pm_and_muc_light_dont_interfere ->
  304:     muc_light_helper:clear_db(host_type()),
  305:     escalus:end_per_testcase(CN, Config);
  306: %% mod_inbox
  307: end_per_testcase(CN, Config) when
  308:       CN =:= remove_inbox;
  309:       CN =:= retrieve_inbox;
  310:       CN =:= remove_inbox_muclight;
  311:       CN =:= retrieve_inbox_muclight ->
  312:     muc_light_helper:clear_db(host_type()),
  313:     escalus:end_per_testcase(CN, Config);
  314: end_per_testcase(CN, Config) when CN =:= retrieve_inbox_muc;
  315:                                   CN =:= remove_inbox_muc ->
  316:     muc_helper:unload_muc(),
  317:     escalus:end_per_testcase(CN, Config);
  318: end_per_testcase(CN, Config) ->
  319:     escalus_fresh:clean(),
  320:     escalus:end_per_testcase(CN, Config).
  321: 
  322: init_inbox(CN, Config, GroupChatType) ->
  323:     case (not ct_helper:is_ct_running())
  324:          orelse mongoose_helper:is_rdbms_enabled(host_type()) of
  325:         true ->
  326:             dynamic_modules:ensure_modules(host_type(), inbox_required_modules(GroupChatType)),
  327:             escalus:init_per_testcase(CN, Config);
  328:         false ->
  329:             {skip, require_rdbms}
  330:     end.
  331: inbox_required_modules(Type) ->
  332:     GroupChatModules = groupchat_module(Type),
  333:     InboxOpts = (inbox_helper:inbox_opts())#{groupchat => [Type]},
  334:     Inbox = {mod_inbox, InboxOpts},
  335:     GroupChatModules ++ [Inbox] .
  336: 
  337: groupchat_module(muc) ->
  338:     [];
  339: groupchat_module(muclight) ->
  340:     [{mod_muc_light, mod_config(mod_muc_light,
  341:                                 #{backend => mongoose_helper:mnesia_or_rdbms_backend(),
  342:                                   rooms_in_rosters => true})}].
  343: 
  344: mam_required_modules(CN, Backend)
  345:         when CN =:= remove_mam_pm;
  346:              CN =:= retrieve_mam_pm ->
  347:     [{mod_mam, mam_helper:config_opts(#{backend => Backend, pm => #{}})}];
  348: mam_required_modules(CN, Backend)
  349:         when CN =:= retrieve_mam_pm_and_muc_light_dont_interfere;
  350:              CN =:= retrieve_mam_muc_light ->
  351:     HostPattern = subhost_pattern(muc_light_helper:muc_host_pattern()),
  352:     MucLightOpts = #{backend => mongoose_helper:mnesia_or_rdbms_backend()},
  353:     [{mod_mam, mam_helper:config_opts(#{backend => Backend,
  354:                                         pm => #{},
  355:                                         muc => #{host => HostPattern}})},
  356:      {mod_muc_light, mod_config(mod_muc_light, MucLightOpts)}];
  357: mam_required_modules(retrieve_mam_pm_and_muc_light_interfere, Backend) ->
  358:     HostPattern = subhost_pattern(muc_light_helper:muc_host_pattern()),
  359:     MucLightOpts = #{backend => mongoose_helper:mnesia_or_rdbms_backend()},
  360:     [{mod_mam, mam_helper:config_opts(#{backend => Backend,
  361:                                         db_message_format => mam_message_xml,
  362:                                         pm => #{archive_groupchats => true},
  363:                                         muc => #{host => HostPattern}})},
  364:      {mod_muc_light, mod_config(mod_muc_light, MucLightOpts)}];
  365: mam_required_modules(CN, Backend) when CN =:= retrieve_mam_muc_private_msg;
  366:                                                CN =:= retrieve_mam_muc ->
  367:     HostPattern = subhost_pattern(muc_helper:muc_host_pattern()),
  368:     MucOpts = #{host => HostPattern,
  369:                 online_backend => ct_helper:get_internal_database(),
  370:                 backend => mongoose_helper:mnesia_or_rdbms_backend()},
  371:     [{mod_mam, mam_helper:config_opts(#{backend => Backend,
  372:                                         pm => #{},
  373:                                         muc => #{host => HostPattern}})},
  374:      {mod_muc, muc_helper:make_opts(MucOpts)}];
  375: mam_required_modules(retrieve_mam_muc_store_pm, Backend) ->
  376:     HostPattern = subhost_pattern(muc_helper:muc_host_pattern()),
  377:     MucOpts = #{host => HostPattern,
  378:                 online_backend => ct_helper:get_internal_database(),
  379:                 backend => mongoose_helper:mnesia_or_rdbms_backend()},
  380:     [{mod_mam, mam_helper:config_opts(#{backend => Backend,
  381:                                         pm => #{archive_groupchats => true},
  382:                                         muc => #{host => HostPattern}})},
  383:      {mod_muc, muc_helper:make_opts(MucOpts)}].
  384: 
  385: pick_enabled_backend() ->
  386:     BackendsList = [
  387:         {mongoose_helper:is_rdbms_enabled(host_type()), rdbms}
  388:     ],
  389:     proplists:get_value(true, BackendsList, mnesia).
  390: 
  391: roster_required_modules() ->
  392:     Backend = pick_enabled_backend(),
  393:     [{mod_roster, roster_backend_opts(Backend)}].
  394: 
  395: roster_backend_opts(Backend) ->
  396:     mod_config(mod_roster, #{backend => Backend}).
  397: 
  398: vcard_required_modules() ->
  399:     Backend = pick_enabled_backend(),
  400:     [{mod_vcard, mod_config(mod_vcard, vcard_backend_opts(Backend))}].
  401: 
  402: vcard_backend_opts(Backend) ->
  403:     #{backend => Backend}.
  404: 
  405: offline_required_modules() ->
  406:     [{mod_offline, mod_offline_config(pick_enabled_backend())}].
  407: 
  408: mod_offline_config(Backend) ->
  409:     config_parser_helper:mod_config(mod_offline, #{backend => Backend}).
  410: 
  411: pubsub_required_modules() ->
  412:     pubsub_required_modules([<<"flat">>, <<"pep">>, <<"push">>]).
  413: pubsub_required_modules(Plugins) ->
  414:     HostPattern = subhost_pattern("pubsub.@HOST@"),
  415:     PubsubConfig = mod_config(mod_pubsub, #{backend => mongoose_helper:mnesia_or_rdbms_backend(),
  416:                                             host => HostPattern,
  417:                                             nodetree => nodetree_tree,
  418:                                             plugins => Plugins}),
  419:     [{mod_caps, config_parser_helper:mod_config_with_auto_backend(mod_caps)},
  420:      {mod_pubsub, PubsubConfig}].
  421: 
  422: is_mim2_started() ->
  423:     #{node := Node} = distributed_helper:mim2(),
  424:     case net_adm:ping(Node) of
  425:         pong -> true;
  426:         _ -> false
  427:     end.
  428: 
  429: roster_started() ->
  430:     dynamic_modules:ensure_modules(host_type(), roster_required_modules()).
  431: 
  432: vcard_started() ->
  433:     dynamic_modules:ensure_modules(host_type(), vcard_required_modules()).
  434: 
  435: offline_started() ->
  436:     dynamic_modules:ensure_modules(host_type(), offline_required_modules()).
  437: 
  438: private_required_modules() ->
  439:     [{mod_private, create_private_config(pick_enabled_backend())}].
  440: 
  441: create_private_config(Backend) ->
  442:     mod_config(mod_private, #{backend => Backend}).
  443: 
  444: private_started() ->
  445:     dynamic_modules:ensure_modules(host_type(), private_required_modules()).
  446: 
  447: %% -------------------------------------------------------------
  448: %% Test cases
  449: %% -------------------------------------------------------------
  450: 
  451: %% ------------------------- Data retrieval - per type verification -------------------------
  452: 
  453: retrieve_vcard(Config) ->
  454:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
  455:             AliceFields = [{<<"FN">>, <<"Alice">>}, {<<"LN">>, <<"Ecila">>}],
  456:             AliceSetResultStanza
  457:             = escalus:send_and_wait(Alice, escalus_stanza:vcard_update(AliceFields)),
  458:             escalus:assert(is_iq_result, AliceSetResultStanza),
  459:             AliceU = escalus_utils:jid_to_lower(escalus_client:username(Alice)),
  460:             AliceS = escalus_utils:jid_to_lower(escalus_client:server(Alice)),
  461:             ExpectedHeader = ["jid", "vcard"],
  462:             ExpectedItems = [
  463:                              #{ "jid" => [{contains, AliceU},
  464:                                           {contains, AliceS}],
  465:                                 "vcard" => [{contains, "Alice"},
  466:                                             {contains, "Ecila"}] }
  467:                             ],
  468:             retrieve_and_validate_personal_data(
  469:               Alice, Config, "vcard", ExpectedHeader, ExpectedItems)
  470:         end).
  471: 
  472: remove_vcard(Config) ->
  473:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
  474:         AliceFields = [{<<"FN">>, <<"Alice">>}, {<<"LN">>, <<"Ecila">>}],
  475:         AliceSetResultStanza
  476:             = escalus:send_and_wait(Alice, escalus_stanza:vcard_update(AliceFields)),
  477:         escalus:assert(is_iq_result, AliceSetResultStanza),
  478: 
  479:         unregister(Alice, Config),
  480: 
  481:         assert_personal_data_via_rpc(Alice, [{vcard,["jid","vcard"],[]}])
  482: 
  483:         end).
  484: 
  485: remove_private(Config) ->
  486:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
  487:         %% Add some private data for Alice
  488:         Element = #xmlel{name = <<"item">>,
  489:                          attrs = [{<<"xmlns">>, <<"alice:private_remove:ns">>}],
  490:                          children = [#xmlcdata{ content = <<"Something to declare">> }]},
  491:         SetPrivateResult = escalus:send_and_wait(Alice,
  492:                              escalus_stanza:private_set(Element)),
  493:         escalus:assert(is_iq_result, SetPrivateResult),
  494: 
  495:         %% Verify the data is stored
  496:         assert_personal_data_via_rpc(Alice, [{private, ["ns","xml"],
  497:                    [{<<"alice:private_remove:ns">>,
  498:                      <<"<item xmlns='alice:private_remove:ns'>Something to declare</item>">>}]}]),
  499: 
  500:         %% Remove Alice
  501:         unregister(Alice, Config),
  502: 
  503:         %% Expect Alice's data to be gone
  504:         assert_personal_data_via_rpc(Alice, [{private, ["ns","xml"], []}])
  505: 
  506:         end).
  507: 
  508: dont_remove_other_user_private_xml(Config) ->
  509:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
  510:         %% Add some private data for Alice and Bob
  511:         AliceNS = <<"alice:private:ns">>,
  512:         AliceContent = <<"To be or not to be">>,
  513:         BobNS = <<"bob:private:ns">>,
  514:         BobContent = <<"This is the winter of our discontent">>,
  515:         send_and_assert_private_stanza(Alice, AliceNS, AliceContent),
  516:         send_and_assert_private_stanza(Bob, BobNS, BobContent),
  517: 
  518:         %% Remove Alice
  519:         unregister(Alice, Config),
  520: 
  521:         %% Expect Alice's data to be gone
  522:         assert_personal_data_via_rpc(Alice, [{private, ["ns","xml"], []}]),
  523: 
  524:         %% Verify that Bob's data is left intact
  525:         ExpectedHeader = ["ns", "xml"],
  526:         ExpectedItems = [#{ "ns" => binary_to_list(BobNS),
  527:                             "xml" => [{contains, binary_to_list(BobNS)},
  528:                                       {contains, binary_to_list(BobContent)}] }
  529:                         ],
  530:         retrieve_and_validate_personal_data(
  531:             Bob, Config, "private", ExpectedHeader, ExpectedItems)
  532: 
  533:     end).
  534: 
  535: remove_multiple_private_xmls(Config) ->
  536:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
  537:         %% Add some private data for Alice for multiple keys
  538:         NSsAndContents = [
  539:                           {<<"alice:private:ns1">>, <<"Some text">>},
  540:                           {<<"alice:private:ns2">>, <<"Other text for another key">>},
  541:                           {<<"alice:private:ns3">>, <<"Even more of text">>}
  542:                          ],
  543:         lists:foreach(
  544:             fun({NS, Content}) ->
  545:                 send_and_assert_private_stanza(Alice, NS, Content)
  546:             end, NSsAndContents),
  547:         ExpectedHeader = ["ns", "xml"],
  548:         ExpectedItems = lists:map(
  549:             fun({NS, Content}) ->
  550:                 #{ "ns" => binary_to_list(NS),
  551:                    "xml" => [{contains, binary_to_list(NS)},
  552:                          {contains, binary_to_list(Content)}]}
  553:             end, NSsAndContents),
  554: 
  555:         %% Verify the data is stored
  556:         retrieve_and_validate_personal_data(
  557:           Alice, Config, "private", ExpectedHeader, ExpectedItems),
  558: 
  559:         %% Remove Alice
  560:         unregister(Alice, Config),
  561: 
  562:         %% Expect all of Alice's data to be gone
  563:         assert_personal_data_via_rpc(Alice, [{private, ["ns","xml"], []}])
  564: 
  565:      end).
  566: 
  567: retrieve_roster(Config) ->
  568:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
  569:             escalus_story:make_all_clients_friends([Alice, Bob]),
  570:             BobU = escalus_utils:jid_to_lower(escalus_client:username(Bob)),
  571:             BobS = escalus_utils:jid_to_lower(escalus_client:server(Bob)),
  572:             ExpectedItems = [
  573:                              #{ "jid" => [{contains,  BobU}, {contains, BobS}] }
  574:                             ],
  575:             retrieve_and_validate_personal_data(
  576:                 Alice, Config, "roster", expected_header(mod_roster), ExpectedItems)
  577:         end).
  578: 
  579: remove_roster(Config) ->
  580:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
  581:         escalus_story:make_all_clients_friends([Alice, Bob]),
  582:         AliceU = escalus_utils:jid_to_lower(escalus_client:username(Alice)),
  583:         AliceS = escalus_utils:jid_to_lower(escalus_client:server(Alice)),
  584:         ExpectedItems = [
  585:                          #{ "jid" => [{contains,  AliceU}, {contains, AliceS}] }
  586:                         ],
  587: 
  588:         unregister(Alice, Config),
  589: 
  590:         assert_personal_data_via_rpc(Alice, [{roster, expected_header(mod_roster), []}]),
  591:         retrieve_and_validate_personal_data(
  592:                 Bob, Config, "roster", expected_header(mod_roster), ExpectedItems)
  593: 
  594:         end).
  595: 
  596: retrieve_mam_pm(Config) ->
  597:     F = fun(Alice, Bob) ->
  598:             Msg1 = <<"1some simple pm message">>,
  599:             Msg2 = <<"2another simple pm message">>,
  600:             Msg3 = <<"3third simple pm message">>,
  601:             escalus:send(Alice, escalus_stanza:chat_to(Bob, Msg1)),
  602:             escalus:send(Bob, escalus_stanza:chat_to(Alice, Msg2)),
  603:             escalus:send(Alice, escalus_stanza:chat_to(Bob, Msg3)),
  604:             [mam_helper:wait_for_archive_size(User, 3) || User <- [Alice, Bob]],
  605:             AliceJID = escalus_client:full_jid(Alice),
  606:             BobJID = escalus_client:full_jid(Bob),
  607: 
  608:             ExpectedHeader = ["id", "from", "message"],
  609:             ExpectedItems = [
  610:                                 #{"message" => [{contains, Msg1}], "from" => [{jid, AliceJID}]},
  611:                                 #{"message" => [{contains, Msg3}], "from" => [{jid, AliceJID}]},
  612:                                 #{"message" => [{contains, Msg2}], "from" => [{jid, BobJID}]}
  613:                             ],
  614: 
  615:             retrieve_and_validate_personal_data(
  616:                 Alice, Config, "mam_pm", ExpectedHeader, ExpectedItems, ["from", "message"]),
  617:             retrieve_and_validate_personal_data(
  618:                 Bob, Config, "mam_pm", ExpectedHeader, ExpectedItems, ["from", "message"])
  619:         end,
  620:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}], F).
  621: 
  622: retrieve_mam_muc(Config) ->
  623:     F = fun(Alice, Bob, Kate) ->
  624:         AliceUserCfg = escalus_users:get_user_by_name(alice),
  625:         RoomCfg = muc_helper:start_fresh_room([], AliceUserCfg, <<"someroom">>, []),
  626:         [Room, Domain] = [proplists:get_value(Key, RoomCfg) || Key <- [room, muc_host]],
  627:         AllRoomMembers = [Alice, Bob, Kate],
  628: 
  629:         muc_helper:enter_room(RoomCfg, [{Alice, <<"Nancy">>},
  630:                                         {Bob, <<"Sid">>},
  631:                                         {Kate, <<"Johnny">>}]),
  632: 
  633:         Body1 = <<"1some simple muc message">>,
  634:         Body2 = <<"2another one">>,
  635:         Body3 = <<"3third message">>,
  636:         muc_helper:send_to_room(RoomCfg, Alice, Body1),
  637:         muc_helper:verify_message_received(RoomCfg, AllRoomMembers, <<"Nancy">>, Body1),
  638:         muc_helper:send_to_room(RoomCfg, Alice, Body2),
  639:         muc_helper:verify_message_received(RoomCfg, AllRoomMembers, <<"Nancy">>, Body2),
  640:         muc_helper:send_to_room(RoomCfg, Bob, Body3),
  641:         muc_helper:verify_message_received(RoomCfg, AllRoomMembers, <<"Sid">>, Body3),
  642: 
  643:         mam_helper:wait_for_room_archive_size(Domain, Room, 3),
  644: 
  645:         ExpectedItemsAlice = [#{"message" => [{contains, binary_to_list(Body1)}]},
  646:                               #{"message" => [{contains, binary_to_list(Body2)}]}],
  647: 
  648:         ExpectedItemsBob = [#{"message" => [{contains, binary_to_list(Body3)}]}],
  649: 
  650:         AliceDir = retrieve_all_personal_data(Alice, Config),
  651:         BobDir = retrieve_all_personal_data(Bob, Config),
  652:         KateDir = retrieve_all_personal_data(Kate, Config),
  653: 
  654:         validate_personal_data(
  655:             AliceDir, "mam_muc", ["id", "message"], ExpectedItemsAlice, ["message"]),
  656:         validate_personal_data(
  657:             BobDir, "mam_muc", ["id", "message"], ExpectedItemsBob, ["message"]),
  658:         refute_personal_data(KateDir, "mam_muc"),
  659: 
  660:         [refute_personal_data(Dir, "mam_pm") || Dir <- [AliceDir, BobDir, KateDir]],
  661: 
  662:         muc_helper:destroy_room(RoomCfg)
  663:         end,
  664:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], F).
  665: 
  666: retrieve_mam_muc_private_msg(Config) ->
  667:     F = fun(Alice, Bob) ->
  668:             AliceUserCfg = escalus_users:get_user_by_name(alice),
  669:             RoomCfg = muc_helper:start_fresh_room([], AliceUserCfg, <<"someroom">>, []),
  670:             [Room, Domain] = [proplists:get_value(Key, RoomCfg) || Key <- [room, muc_host]],
  671: 
  672:             muc_helper:enter_room(RoomCfg, [{Alice, <<"Nancy">>}, {Bob, <<"Sid">>}]),
  673: 
  674:             PMBody = <<"Hi, Bob!">>,
  675:             {PrivAddrAlice, _} = send_receive_muc_private_message(
  676:                 Room, Domain, {Alice, <<"Nancy">>}, {Bob, <<"Sid">>}, PMBody),
  677: 
  678:             [mam_helper:wait_for_archive_size(User, 1) || User <- [Alice, Bob]],
  679: 
  680:             PMExpectedItemsAlice = [#{"message" => [{contains, binary_to_list(PMBody)}],
  681:                                       "from" => [{jid, escalus_client:full_jid(Alice)}]}],
  682:             PMExpectedItemsBob = [#{"message" => [{contains, binary_to_list(PMBody)}],
  683:                                     "from" => [{jid, PrivAddrAlice}]}],
  684: 
  685:             AliceDir = retrieve_all_personal_data(Alice, Config),
  686:             BobDir = retrieve_all_personal_data(Bob, Config),
  687: 
  688:             validate_personal_data(
  689:                 AliceDir, "mam_pm", ["id", "from", "message"], PMExpectedItemsAlice, []),
  690:             validate_personal_data(
  691:                 BobDir, "mam_pm", ["id", "from", "message"], PMExpectedItemsBob, []),
  692: 
  693:             refute_personal_data(AliceDir, "mam_muc"),
  694:             refute_personal_data(BobDir, "mam_muc"),
  695: 
  696:             muc_helper:destroy_room(RoomCfg)
  697:         end,
  698:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}], F).
  699: 
  700: 
  701: 
  702: retrieve_mam_muc_store_pm(Config) ->
  703:     F = fun(Alice, Bob, Kate) ->
  704:             AliceUserCfg = escalus_users:get_user_by_name(alice),
  705:             RoomCfg = muc_helper:start_fresh_room([], AliceUserCfg, <<"someroom">>, []),
  706:             [Room, Domain] = [proplists:get_value(Key, RoomCfg) || Key <- [room, muc_host]],
  707:             AllRoomMembers = [Alice, Bob, Kate],
  708: 
  709:             muc_helper:enter_room(RoomCfg, [{Alice, <<"Nancy">>},
  710:                                             {Bob, <<"Sid">>},
  711:                                             {Kate, <<"Johnny">>}]),
  712: 
  713:             Body1 = <<"1some simple muc message">>,
  714:             Body2 = <<"2another one">>,
  715:             Body3 = <<"3third message">>,
  716:             muc_helper:send_to_room(RoomCfg, Alice, Body1),
  717:             muc_helper:verify_message_received(RoomCfg, AllRoomMembers, <<"Nancy">>, Body1),
  718:             muc_helper:send_to_room(RoomCfg, Alice, Body2),
  719:             muc_helper:verify_message_received(RoomCfg, AllRoomMembers, <<"Nancy">>, Body2),
  720:             muc_helper:send_to_room(RoomCfg, Bob, Body3),
  721:             muc_helper:verify_message_received(RoomCfg, AllRoomMembers, <<"Sid">>, Body3),
  722: 
  723:             PMBody = <<"4Hi, Bob!">>,
  724:             {PrivAddrAlice, PrivAddrBob} = send_receive_muc_private_message(
  725:                 Room, Domain, {Alice, <<"Nancy">>}, {Bob, <<"Sid">>}, PMBody),
  726: 
  727:             mam_helper:wait_for_room_archive_size(Domain, Room, 3),
  728:             mam_helper:wait_for_archive_size(Kate, 4),
  729:             [mam_helper:wait_for_archive_size(User, 5) || User <- [Alice, Bob]],
  730: 
  731:             AliceDir = retrieve_all_personal_data(Alice, Config),
  732:             BobDir = retrieve_all_personal_data(Bob, Config),
  733:             KateDir = retrieve_all_personal_data(Kate, Config),
  734: 
  735:             ExpectedItemsAlice = [#{"message" => [{contains, binary_to_list(Body1)}]},
  736:                                   #{"message" => [{contains, binary_to_list(Body2)}]}],
  737:             ExpectedItemsBob = [#{"message" => [{contains, binary_to_list(Body3)}]}],
  738: 
  739:             validate_personal_data(
  740:                 AliceDir, "mam_muc", ["id", "message"], ExpectedItemsAlice, ["message"]),
  741:             validate_personal_data(
  742:                 BobDir, "mam_muc", ["id", "message"], ExpectedItemsBob, ["message"]),
  743:             refute_personal_data(KateDir, "mam_muc"),
  744: 
  745:             RoomJID = <<Room/binary, "@", Domain/binary>>,
  746:             MsgFromAliceToRoom = #{"message" => [{contains, "<body>[1,2]"}],
  747:                                    "from" => [{jid, PrivAddrAlice}]},
  748:             PMExpectedItemsKate = [#{"message" => [{contains, "<body/>"}],
  749:                                      "from" => [{jid, RoomJID}]},
  750:                                    MsgFromAliceToRoom, MsgFromAliceToRoom,
  751:                                    #{"message" => [{contains, binary_to_list(Body3)}],
  752:                                      "from" => [{jid, PrivAddrBob}]}
  753:                                   ],
  754:             PMExpectedItemsAlice = PMExpectedItemsKate ++
  755:                                    [#{"message" => [{contains, binary_to_list(PMBody)}],
  756:                                       "from" => [{jid, escalus_client:full_jid(Alice)}]}],
  757:             MsgFromAlice = #{"message" => [{contains, "<body>[1,2,4]"}],
  758:                              "from" => [{jid, PrivAddrAlice}]},
  759:             PMExpectedItemsBob = [#{"message" => [{contains, "<body/>"}],
  760:                                     "from" => [{jid, RoomJID}]},
  761:                                   MsgFromAlice, MsgFromAlice, MsgFromAlice,
  762:                                   #{"message" => [{contains, binary_to_list(Body3)}],
  763:                                     "from" => [{jid, PrivAddrBob}]}
  764:                                  ],
  765:             SortFn = muc_msg_first(RoomJID),
  766:             validate_personal_data(
  767:                 KateDir, "mam_pm", ["id", "from", "message"], PMExpectedItemsKate, SortFn),
  768:             validate_personal_data(
  769:                 AliceDir, "mam_pm", ["id", "from", "message"], PMExpectedItemsAlice, SortFn),
  770:             validate_personal_data(
  771:                 BobDir, "mam_pm", ["id", "from", "message"], PMExpectedItemsBob, SortFn),
  772: 
  773:             muc_helper:destroy_room(RoomCfg)
  774:         end,
  775:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], F).
  776: 
  777: remove_mam_pm(Config) ->
  778:     F = fun(Alice, Bob) ->
  779:             Msg1 = <<"1remove_mam_pm message">>,
  780:             Msg2 = <<"2remove_mam_pm message message">>,
  781:             Msg3 = <<"3remove_mam_pm message message">>,
  782:             escalus:send(Alice, escalus_stanza:chat_to(Bob, Msg1)),
  783:             escalus:send(Bob, escalus_stanza:chat_to(Alice, Msg2)),
  784:             escalus:send(Alice, escalus_stanza:chat_to(Bob, Msg3)),
  785:             [mam_helper:wait_for_archive_size(User, 3) || User <- [Alice, Bob]],
  786:             AliceJID = escalus_client:full_jid(Alice),
  787:             BobJID = escalus_client:full_jid(Bob),
  788: 
  789:             ExpectedHeader = ["id", "from", "message"],
  790:             ExpectedItems = [
  791:                                 #{"message" => [{contains, Msg1}], "from" => [{jid, AliceJID}]},
  792:                                 #{"message" => [{contains, Msg3}], "from" => [{jid, AliceJID}]},
  793:                                 #{"message" => [{contains, Msg2}], "from" => [{jid, BobJID}]}
  794:                             ],
  795: 
  796:             unregister(Alice, Config),
  797: 
  798:             assert_personal_data_via_rpc(Alice, [{mam_pm, ExpectedHeader, []}]),
  799: 
  800:             retrieve_and_validate_personal_data(
  801:                 Bob, Config, "mam_pm", ExpectedHeader, ExpectedItems, ["from", "message"])
  802:         end,
  803:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}], F).
  804: 
  805: retrieve_mam_muc_light(Config) ->
  806:     F = fun(Alice, Bob, Kate) ->
  807:             RoomJid = muc_light_helper:given_muc_light_room(undefined, Alice, [{Bob, member}, {Kate, member}]),
  808:             [Room, Domain] = binary:split(RoomJid, <<"@">>),
  809:             Body1 = <<"1some simple muc message">>,
  810:             Body2 = <<"2another one">>,
  811:             Body3 = <<"3third message">>,
  812: 
  813:             M1 = muc_light_helper:when_muc_light_message_is_sent(Alice, Room, Body1, <<"Id1">>),
  814:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M1),
  815:             M2 = muc_light_helper:when_muc_light_message_is_sent(Alice, Room, Body2, <<"Id2">>),
  816:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M2),
  817:             M3 = muc_light_helper:when_muc_light_message_is_sent(Bob, Room, Body3, <<"Id3">>),
  818:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M3),
  819: 
  820:             mam_helper:wait_for_room_archive_size(Domain, Room, 4),
  821: 
  822:             ExpectedItemsAlice = [#{"message" => [{contains, binary_to_list(Body1)}]},
  823:                                   #{"message" => [{contains, binary_to_list(Body2)}]}],
  824: 
  825:             ExpectedItemsBob = [#{"message" => [{contains, binary_to_list(Body3)}]}],
  826: 
  827:             retrieve_and_validate_personal_data(
  828:                 Alice, Config, "mam_muc", ["id", "message"], ExpectedItemsAlice, ["message"]),
  829:             retrieve_and_validate_personal_data(
  830:                 Bob, Config, "mam_muc", ["id", "message"], ExpectedItemsBob, ["message"]),
  831:             refute_personal_data(Kate, Config, "mam_muc")
  832:         end,
  833:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], F).
  834: 
  835: retrieve_mam_pm_and_muc_light_dont_interfere(Config) ->
  836:     F = fun(Alice, Bob, Kate) ->
  837:             RoomJid = muc_light_helper:given_muc_light_room(undefined, Alice,
  838:                                                             [{Bob, member}, {Kate, member}]),
  839:             [Room, Domain] = binary:split(RoomJid, <<"@">>),
  840:             BodyMucAlice = <<"some simple muc message from Alice">>,
  841:             BodyMucBob = <<"some simple muc message from Bob">>,
  842:             BodyPmAlice = <<"some simple pm message from Alice">>,
  843:             BodyPmBob = <<"some simple pm message from Bob">>,
  844: 
  845:             M1 = muc_light_helper:when_muc_light_message_is_sent(Alice, Room, BodyMucAlice, <<"Id1">>),
  846:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M1),
  847:             M2 = muc_light_helper:when_muc_light_message_is_sent(Bob, Room, BodyMucBob, <<"Id2">>),
  848:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M2),
  849: 
  850:             mam_helper:wait_for_room_archive_size(Domain, Room, 3),
  851: 
  852:             escalus:send(Alice, escalus_stanza:chat_to(Bob, BodyPmAlice)),
  853:             escalus:send(Bob, escalus_stanza:chat_to(Alice, BodyPmBob)),
  854: 
  855:             [mam_helper:wait_for_archive_size(User, 2) || User <- [Alice, Bob]],
  856: 
  857:             false = mongoose_helper:successful_rpc(gen_mod, get_module_opt,
  858:                                                    [host_type(), mod_mam_pm, archive_groupchats, undefined]),
  859: 
  860:             AliceDir = retrieve_all_personal_data(Alice, Config),
  861:             BobDir = retrieve_all_personal_data(Bob, Config),
  862:             KateDir = retrieve_all_personal_data(Kate, Config),
  863: 
  864:             validate_personal_data(
  865:                 AliceDir, "mam_muc", ["id", "message"],
  866:                 [#{"message" => [{contains, binary_to_list(BodyMucAlice)}]}], []),
  867:             validate_personal_data(
  868:                 BobDir, "mam_muc", ["id", "message"],
  869:                 [#{"message" => [{contains, binary_to_list(BodyMucBob)}]}], []),
  870: 
  871:             PM = [#{"message" => [{contains, BodyPmAlice}],
  872:                     "from" => [{jid, escalus_client:full_jid(Alice)}]},
  873:                   #{"message" => [{contains, BodyPmBob}],
  874:                     "from" => [{jid, escalus_client:full_jid(Bob)}]}],
  875:             validate_personal_data(AliceDir, "mam_pm", ["id", "from", "message"], PM, ["from", "message"]),
  876:             validate_personal_data(BobDir, "mam_pm", ["id", "from", "message"], PM, ["from", "message"]),
  877:             refute_personal_data(KateDir, "mam_pm"),
  878:             refute_personal_data(KateDir, "mam_muc")
  879:         end,
  880:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], F).
  881: 
  882: retrieve_mam_pm_and_muc_light_interfere(Config) ->
  883:     F = fun(Alice, Bob, Kate) ->
  884:             RoomJid = muc_light_helper:given_muc_light_room(undefined, Alice,
  885:                                                             [{Bob, member}, {Kate, member}]),
  886:             [Room, Domain] = binary:split(RoomJid, <<"@">>),
  887:             BodyMucAlice = <<"some simple muc message from Alice">>,
  888:             BodyMucBob = <<"some simple muc message from Bob">>,
  889:             BodyPmAlice = <<"some simple pm message from Alice">>,
  890:             BodyPmBob = <<"some simple pm message from Bob">>,
  891: 
  892:             M1 = muc_light_helper:when_muc_light_message_is_sent(Alice, Room, BodyMucAlice, <<"Id1">>),
  893:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M1),
  894:             M2 = muc_light_helper:when_muc_light_message_is_sent(Bob, Room, BodyMucBob, <<"Id2">>),
  895:             muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob, Kate], M2),
  896: 
  897:             mam_helper:wait_for_room_archive_size(Domain, Room, 3),
  898: 
  899:             escalus:send(Alice, escalus_stanza:chat_to(Bob, BodyPmAlice)),
  900:             escalus:send(Bob, escalus_stanza:chat_to(Alice, BodyPmBob)),
  901: 
  902:             [mam_helper:wait_for_archive_size(User, 5) || User <- [Alice, Bob]],
  903:             mam_helper:wait_for_archive_size(Kate, 3),
  904: 
  905:             true = mongoose_helper:successful_rpc(gen_mod, get_module_opt,
  906:                                                    [host_type(), mod_mam_pm, archive_groupchats, undefined]),
  907: 
  908:             AliceDir = retrieve_all_personal_data(Alice, Config),
  909:             BobDir = retrieve_all_personal_data(Bob, Config),
  910:             KateDir = retrieve_all_personal_data(Kate, Config),
  911: 
  912:             validate_personal_data(
  913:                 AliceDir, "mam_muc", ["id", "message"],
  914:                 [#{"message" => [{contains, binary_to_list(BodyMucAlice)}]}], []),
  915:             validate_personal_data(
  916:                 BobDir, "mam_muc", ["id", "message"],
  917:                 [#{"message" => [{contains, binary_to_list(BodyMucBob)}]}], []),
  918: 
  919:             AliceRoomJid = <<RoomJid/binary, "/", (escalus_client:short_jid(Alice))/binary>>,
  920:             BobRoomJid = <<RoomJid/binary, "/", (escalus_client:short_jid(Bob))/binary>>,
  921: 
  922:             MucPM = [#{"message" => [{contains, "urn:xmpp:muclight:0#affiliations"}],
  923:                        "from" => [{jid, RoomJid}]},
  924:                      #{"message" => [{contains, BodyMucAlice}], "from" => [{jid, AliceRoomJid}]},
  925:                      #{"message" => [{contains, BodyMucBob}], "from" => [{jid, BobRoomJid}]}],
  926:             AllPM = MucPM ++ [#{"message" => [{contains, BodyPmAlice}],
  927:                                 "from" => [{jid, escalus_client:full_jid(Alice)}]},
  928:                               #{"message" => [{contains, BodyPmBob}],
  929:                                     "from" => [{jid, escalus_client:full_jid(Bob)}]}],
  930:             SortFn = muc_msg_first(RoomJid),
  931:             validate_personal_data(AliceDir, "mam_pm", ["id", "from", "message"], AllPM, SortFn),
  932:             validate_personal_data(BobDir, "mam_pm", ["id", "from", "message"], AllPM, SortFn),
  933:             validate_personal_data(KateDir, "mam_pm", ["id", "from", "message"], MucPM, SortFn),
  934:             refute_personal_data(KateDir, "mam_muc")
  935:         end,
  936:     escalus_fresh:story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], F).
  937: 
  938: retrieve_offline(Config) ->
  939:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], fun(Alice, Bob, Kate) ->
  940:             mongoose_helper:logout_user(Config, Alice),
  941:             Body1 = <<"1Hey!">>,
  942:             Body2 = <<"2Here is Johnny!">>,
  943:             Body3 = <<"3Where is Johnny ?">>,
  944:             escalus:send(Bob, escalus_stanza:chat_to(Alice, Body1)),
  945:             escalus:send(Bob, escalus_stanza:chat_to(Alice, Body2)),
  946:             escalus:send(Kate, escalus_stanza:chat_to(Alice, Body3)),
  947:             %% Well, jid_to_lower works for any binary :)
  948:             AliceU = escalus_utils:jid_to_lower(escalus_client:username(Alice)),
  949:             AliceS = escalus_utils:jid_to_lower(escalus_client:server(Alice)),
  950:             mongoose_helper:wait_until(
  951:               fun() ->
  952:                       mongoose_helper:successful_rpc(mod_offline_backend, count_offline_messages,
  953:                                                      [host_type(), AliceU, AliceS, 10])
  954:               end, 3),
  955: 
  956:             BobJid = escalus_client:full_jid(Bob),
  957:             AliceJid = escalus_client:short_jid(Alice),
  958:             KateJid = escalus_client:full_jid(Kate),
  959:             ExpectedHeader = ["timestamp", "from", "to", "packet"],
  960:             Expected = [{Body1, BobJid, AliceJid}, {Body2, BobJid, AliceJid}, {Body3, KateJid, AliceJid}],
  961: 
  962:             ExpectedItems = lists:map(fun({Body, From ,To}) ->
  963:                 #{ "packet" => [{contains, Body}],
  964:                     "from" => binary_to_list(From),
  965:                     "to" => binary_to_list(To),
  966:                     "timestamp" => [{validate, fun validate_datetime/1}]}
  967:             end, Expected),
  968: 
  969:             retrieve_and_validate_personal_data(
  970:               Alice, Config, "offline", ExpectedHeader, ExpectedItems, ["packet"])
  971:         end).
  972: 
  973: remove_offline(Config) ->
  974:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}, {kate, 1}], fun(Alice, Bob, Kate) ->
  975:             mongoose_helper:logout_user(Config, Alice),
  976:             Body1 = <<"Hey!">>,
  977:             Body2 = <<"Here is Johnny!">>,
  978:             Body3 = <<"Where is Johnny ?">>,
  979:             escalus:send(Bob, escalus_stanza:chat_to(Alice, Body1)),
  980:             escalus:send(Bob, escalus_stanza:chat_to(Alice, Body2)),
  981:             escalus:send(Kate, escalus_stanza:chat_to(Alice, Body3)),
  982:             %% Well, jid_to_lower works for any binary :)
  983:             AliceU = escalus_utils:jid_to_lower(escalus_client:username(Alice)),
  984:             AliceS = escalus_utils:jid_to_lower(escalus_client:server(Alice)),
  985:             mongoose_helper:wait_until(
  986:               fun() ->
  987:                       mongoose_helper:successful_rpc(mod_offline_backend, count_offline_messages,
  988:                                                      [host_type(), AliceU, AliceS, 10])
  989:               end, 3),
  990: 
  991:             unregister(Alice, Config),
  992: 
  993:             assert_personal_data_via_rpc(
  994:               Alice, [{offline, ["timestamp","from", "to", "packet"],[]}])
  995:         end).
  996: 
  997: retrieve_pubsub_payloads(Config) ->
  998:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
  999:         [Node1={_,NodeName1}, Node2={_,NodeName2}] = pubsub_tools:create_node_names(2),
 1000:         {BinItem1, StringItem1} = item_content(<<"Item1Data">>),
 1001:         {BinItem2, StringItem2} = item_content(<<"Item2Data">>),
 1002:         {BinItem3, StringItem3} = item_content(<<"Item3Data">>),
 1003:         {BinOther, StringOther} = item_content(<<"OtherItemData">>),
 1004: 
 1005:         pubsub_tools:publish(Alice, <<"Item1">>, Node1, [{with_payload, BinItem1}]),
 1006:         pubsub_tools:publish(Alice, <<"Item2">>, Node1, [{with_payload, BinItem2}]),
 1007:         pubsub_tools:publish(Alice, <<"Item3">>, Node1, [{with_payload, BinItem3}]),
 1008:         pubsub_tools:publish(Alice, <<"OtherItem">>, Node2, [{with_payload, BinOther}]),
 1009: 
 1010:         ExpectedItems = [pubsub_payloads_row_map(NodeName1, "Item1", StringItem1),
 1011:                          pubsub_payloads_row_map(NodeName1, "Item2", StringItem2),
 1012:                          pubsub_payloads_row_map(NodeName1, "Item3", StringItem3),
 1013:                          pubsub_payloads_row_map(NodeName2, "OtherItem", StringOther)],
 1014: 
 1015:         retrieve_and_validate_personal_data(Alice, Config, "pubsub_payloads",
 1016:                                             ["node_name", "item_id", "payload"],
 1017:                                             ExpectedItems, ["item_id"])
 1018:                                               end).
 1019: 
 1020: dont_retrieve_other_user_pubsub_payload(Config) ->
 1021:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1022:         [Node1={_,NodeName1}] = pubsub_tools:create_node_names(1),
 1023:         pubsub_tools:create_nodes([{Alice, Node1, []}]),
 1024: 
 1025:         {BinItem1, StringItem1} = item_content(<<"Item1Data">>),
 1026:         {BinItem2, StringItem2} = item_content(<<"Item2Data">>),
 1027: 
 1028:         AffChange = [{Bob, <<"publish-only">>}],
 1029:         pubsub_tools:set_affiliations(Alice, Node1, AffChange, []),
 1030:         pubsub_tools:publish(Alice, <<"Item1">>, Node1, [{with_payload, {true, BinItem1}}]),
 1031:         pubsub_tools:publish(Bob, <<"Item2">>, Node1, [{with_payload, {true, BinItem2}}]),
 1032: 
 1033:         retrieve_and_validate_personal_data(
 1034:             Alice, Config, "pubsub_payloads", ["node_name", "item_id", "payload"],
 1035:             [pubsub_payloads_row_map(NodeName1, "Item1", StringItem1)]),
 1036: 
 1037:         retrieve_and_validate_personal_data(
 1038:             Bob, Config, "pubsub_payloads", ["node_name","item_id", "payload"],
 1039:             [pubsub_payloads_row_map(NodeName1, "Item2", StringItem2)]),
 1040: 
 1041:         pubsub_tools:delete_node(Alice, Node1, [])
 1042:                                               end).
 1043: 
 1044: retrieve_created_pubsub_nodes(Config) ->
 1045:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1046:         [Node1={_,NodeName1}, Node2={_,NodeName2}, Node3={_,NodeName3}] =
 1047:         pubsub_tools:create_node_names(3),
 1048: 
 1049:         NodeNS = <<"myns">>,
 1050:         PepNode = make_pep_node_info(Alice, NodeNS),
 1051:         AccessModel = {<<"pubsub#access_model">>, <<"authorize">>},
 1052: 
 1053:         pubsub_tools:create_nodes([
 1054:                       {Alice, Node1, []},
 1055:                       {Alice, Node2, []},
 1056:                       {Alice, PepNode, [{config, [AccessModel]}]},
 1057:                       {Bob, Node3, [{type, <<"push">>}]}
 1058:                      ]),
 1059: 
 1060:         ExpectedHeader = ["node_name", "type"],
 1061: 
 1062:         retrieve_and_validate_personal_data(
 1063:             Alice, Config, "pubsub_nodes", ExpectedHeader,
 1064:             [pubsub_nodes_row_map(NodeNS, "pep"),
 1065:              pubsub_nodes_row_map(NodeName1, "flat"),
 1066:              pubsub_nodes_row_map(NodeName2, "flat")]),
 1067: 
 1068:         retrieve_and_validate_personal_data(
 1069:             Bob, Config, "pubsub_nodes", ExpectedHeader,
 1070:             [pubsub_nodes_row_map(NodeName3, "push")]),
 1071: 
 1072: 
 1073:         Nodes = [{Alice, PepNode}, {Alice, Node1}, {Alice, Node2}, {Bob, Node3}],
 1074:         [pubsub_tools:delete_node(User, Node, []) || {User, Node} <- Nodes]
 1075:                                                         end).
 1076: 
 1077: remove_pubsub_subscriptions(Config) ->
 1078:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1079:             Node = pubsub_tools:pubsub_node(),
 1080:             pubsub_tools:create_node(Alice, Node, []),
 1081:             pubsub_tools:subscribe(Bob, Node, []),
 1082: 
 1083:             unregister(Bob, Config),
 1084: 
 1085:             assert_personal_data_via_rpc(Bob,
 1086:                                          [{pubsub_payloads,["node_name","item_id","payload"],[]},
 1087:                                           {pubsub_nodes,["node_name","type"],[]},
 1088:                                           {pubsub_subscriptions,["node_name"],[]}])
 1089:         end).
 1090: 
 1091: retrieve_pubsub_subscriptions(Config) ->
 1092:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1093:             Node = {_Domain, NodeName} = pubsub_tools:pubsub_node(),
 1094:             pubsub_tools:create_node(Alice, Node, []),
 1095:             pubsub_tools:subscribe(Bob, Node, []),
 1096:             retrieve_and_validate_personal_data(Bob, Config, "pubsub_subscriptions", ["node_name"],
 1097:                 [pubsub_subscription_row_map(NodeName)]),
 1098: 
 1099:             pubsub_tools:delete_node(Alice, Node, [])
 1100:         end).
 1101: 
 1102: remove_pubsub_dont_remove_flat_pubsub_node(Config) ->
 1103:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
 1104:         Node1 = {_,NodeName} = pubsub_tools:pubsub_node_with_num(1),
 1105:         pubsub_tools:create_nodes([{Alice, Node1, []}]),
 1106: 
 1107:         unregister(Alice, Config),
 1108: 
 1109:         assert_personal_data_via_rpc(Alice,
 1110:                                      [{pubsub_payloads,["node_name","item_id","payload"],[]},
 1111:                                       {pubsub_nodes,["node_name","type"],[[NodeName, <<"flat">>]]},
 1112:                                       {pubsub_subscriptions,["node_name"],[]}])
 1113:         end).
 1114: 
 1115: remove_pubsub_push_node(Config) ->
 1116:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1117:         [Node] = pubsub_tools:create_node_names(1),
 1118:         pubsub_tools:create_nodes([{Alice, Node, [{type, <<"push">>}]}]),
 1119: 
 1120:         Content = [
 1121:                    {<<"message-count">>, <<"1">>},
 1122:                    {<<"last-message-sender">>, <<"senderId">>},
 1123:                    {<<"last-message-body">>, <<"message body">>}
 1124:                   ],
 1125:         Options = [
 1126:                    {<<"device_id">>, <<"sometoken">>},
 1127:                    {<<"service">>, <<"apns">>}
 1128:                   ],
 1129: 
 1130:         PublishIQ = push_pubsub_SUITE:publish_iq(Bob, Node, Content, Options),
 1131:         escalus:send(Bob, PublishIQ),
 1132:         escalus:assert(is_iq_result, escalus:wait_for_stanza(Bob)),
 1133: 
 1134:         unregister(Alice, Config),
 1135: 
 1136:         assert_personal_data_via_rpc(Alice, [{pubsub_payloads,["node_name","item_id","payload"],[]},
 1137:                                              {pubsub_nodes,["node_name","type"],[]},
 1138:                                              {pubsub_subscriptions,["node_name"],[]}])
 1139:         end).
 1140: 
 1141: remove_pubsub_pep_node(Config) ->
 1142:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
 1143:         NodeName = <<"myns">>,
 1144:         PepNode = make_pep_node_info(Alice, NodeName),
 1145: 
 1146:         pubsub_tools:create_nodes([
 1147:                       {Alice, PepNode, []}
 1148:                      ]),
 1149: 
 1150:         unregister(Alice, Config),
 1151: 
 1152:         assert_personal_data_via_rpc(Alice, [{pubsub_payloads,["node_name","item_id","payload"],[]},
 1153:                                              {pubsub_nodes,["node_name","type"],[]},
 1154:                                              {pubsub_subscriptions,["node_name"],[]}])
 1155:         end).
 1156: 
 1157: remove_pubsub_dont_remove_node_when_only_publisher(Config) ->
 1158:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1159:         Node1 = {_,NodeName} = pubsub_tools:pubsub_node_with_num(1),
 1160:         pubsub_tools:create_nodes([{Alice, Node1, []}]),
 1161: 
 1162:         AffChange = [{Bob, <<"publish-only">>}],
 1163:         pubsub_tools:set_affiliations(Alice, Node1, AffChange, []),
 1164: 
 1165:         unregister(Bob, Config),
 1166: 
 1167:         assert_personal_data_via_rpc(Alice,
 1168:                                      [{pubsub_payloads,["node_name","item_id","payload"],[]},
 1169:                                       {pubsub_nodes,["node_name","type"],[[NodeName, <<"flat">>]]},
 1170:                                       {pubsub_subscriptions,["node_name"],[]}])
 1171:         end).
 1172: 
 1173: remove_pubsub_all_data(Config) ->
 1174:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1175:         [Node1={_,Name1}, Node2={_,Name2}, Node3={_,Name3}, Node4={_,Name4}]
 1176:             = pubsub_tools:create_node_names(4),
 1177:         PepNode = make_pep_node_info(Alice, <<"myns">>),
 1178:         pubsub_tools:create_nodes([
 1179:                       {Alice, Node1, []},
 1180:                       {Alice, Node2, []},
 1181:                       {Alice, PepNode, []},
 1182:                       {Bob, Node3, []},
 1183:                       {Bob, Node4, [{type, <<"push">>}]}
 1184:                      ]),
 1185: 
 1186:         AffChange = [{Bob, <<"publish-only">>}],
 1187:         pubsub_tools:set_affiliations(Alice, Node1, AffChange, []),
 1188:         pubsub_tools:subscribe(Bob, Node2, []),
 1189:         pubsub_tools:subscribe(Alice, Node3, []),
 1190: 
 1191:         {BinItem1, _} = item_content(<<"Item1Data">>),
 1192:         {BinItem2, _} = item_content(<<"Item2Data">>),
 1193:         {BinItem3, _} = item_content(<<"Item3Data">>),
 1194:         {BinItem4, _} = item_content(<<"Item4Data">>),
 1195:         AliceToNode1 = <<"Alice publishes to Node1, but nobody is subscribed">>,
 1196:         AliceToNode2 = <<"Alice published to Node2, so Bob receives it">>,
 1197:         BobToNode1 = <<"Bob publishes to Node1, but nobody is subscribed">>,
 1198:         BobToNode3 = <<"Bob publishes to Node3, so Alice receives it">>,
 1199: 
 1200:         pubsub_tools:publish(Alice, AliceToNode1, Node1, [{with_payload, {true, BinItem1}}]),
 1201: 
 1202:         pubsub_tools:publish(Alice, AliceToNode2, Node2, [{with_payload, {true, BinItem2}}]),
 1203:         pubsub_tools:receive_item_notification(Bob, AliceToNode2, Node2, []),
 1204: 
 1205:         pubsub_tools:publish(Bob, BobToNode1, Node1, [{with_payload, {true, BinItem3}}]),
 1206: 
 1207:         pubsub_tools:publish(Bob, BobToNode3, Node3, [{with_payload, {true, BinItem4}}]),
 1208:         pubsub_tools:receive_item_notification(Alice, BobToNode3, Node3, []),
 1209: 
 1210:         unregister(Alice, Config),
 1211: 
 1212:         [{pubsub_payloads,["node_name","item_id","payload"], AlicePayloads},
 1213:          {pubsub_nodes,["node_name","type"], AliceNodes},
 1214:          {pubsub_subscriptions, ["node_name"], []}]
 1215:             = get_personal_data_via_rpc(
 1216:                 Alice, [pubsub_payloads, pubsub_nodes, pubsub_subscriptions]),
 1217:         XmlBinItem1 = exml:to_binary(BinItem1),
 1218:         XmlBinItem2 = exml:to_binary(BinItem2),
 1219:         [[Name1, AliceToNode1, XmlBinItem1],
 1220:          [Name2, AliceToNode2, XmlBinItem2]] = lists:sort(AlicePayloads),
 1221:         [[Name1, <<"flat">>], [Name2, <<"flat">>]] = lists:sort(AliceNodes),
 1222: 
 1223:         [{pubsub_payloads,["node_name","item_id","payload"], Payloads},
 1224:          {pubsub_nodes,["node_name","type"], Nodes},
 1225:          {pubsub_subscriptions, ["node_name"], Subs}]
 1226:             = get_personal_data_via_rpc(
 1227:                 Bob, [pubsub_payloads, pubsub_nodes, pubsub_subscriptions]),
 1228:         XmlBinItem3 = exml:to_binary(BinItem3),
 1229:         XmlBinItem4 = exml:to_binary(BinItem4),
 1230:         [[Name1, BobToNode1, XmlBinItem3],
 1231:          [Name3, BobToNode3, XmlBinItem4]] = lists:sort(Payloads),
 1232:         [[Name3, <<"flat">>], [Name4, <<"push">>]] = lists:sort(Nodes),
 1233:         [[Name2]] = Subs
 1234:       end).
 1235: 
 1236: retrieve_all_pubsub_data(Config) ->
 1237:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1238:         [Node1={_,NodeName1}, Node2={_,NodeName2}, Node3={_,NodeName3}] =
 1239:         pubsub_tools:create_node_names(3),
 1240:         pubsub_tools:create_nodes([{Alice, Node1, []}, {Alice, Node2, []}, {Bob, Node3, []}]),
 1241: 
 1242:         AffChange = [{Bob, <<"publish-only">>}],
 1243:         pubsub_tools:set_affiliations(Alice, Node1, AffChange, []),
 1244:         pubsub_tools:subscribe(Bob, Node2, []),
 1245: 
 1246:         {BinItem1, StringItem1} = item_content(<<"Item1Data">>),
 1247:         {BinItem2, StringItem2} = item_content(<<"Item2Data">>),
 1248:         {BinItem3, StringItem3} = item_content(<<"Item3Data">>),
 1249: 
 1250:         pubsub_tools:publish(Alice, <<"Item1">>, Node1, [{with_payload, {true, BinItem1}}]),
 1251:         pubsub_tools:publish(Alice, <<"Item2">>, Node2, [{with_payload, {true, BinItem2}}]),
 1252:         pubsub_tools:receive_item_notification(Bob, <<"Item2">>, Node2, []),
 1253:         pubsub_tools:publish(Bob, <<"Item3">>, Node1, [{with_payload, {true, BinItem3}}]),
 1254: 
 1255:         %% Bob has one subscription, one node created and one payload sent
 1256:         retrieve_and_validate_personal_data(
 1257:             Bob, Config, "pubsub_subscriptions", ["node_name"],
 1258:             [pubsub_subscription_row_map(NodeName2)]),
 1259: 
 1260:         retrieve_and_validate_personal_data(
 1261:             Bob, Config, "pubsub_nodes", ["node_name", "type"],
 1262:             [pubsub_nodes_row_map(NodeName3, "flat")]),
 1263: 
 1264:         retrieve_and_validate_personal_data(
 1265:             Bob, Config, "pubsub_payloads", ["node_name", "item_id", "payload"],
 1266:             [pubsub_payloads_row_map(NodeName1, "Item3", StringItem3)]),
 1267: 
 1268:         %% Alice has two nodes created and two payloads sent
 1269:         retrieve_and_validate_personal_data(
 1270:             Alice, Config, "pubsub_nodes", ["node_name", "type"],
 1271:             [pubsub_nodes_row_map(NodeName1, "flat"),
 1272:              pubsub_nodes_row_map(NodeName2, "flat")]),
 1273:         retrieve_and_validate_personal_data(
 1274:             Alice, Config, "pubsub_payloads", ["node_name", "item_id","payload"],
 1275:             [pubsub_payloads_row_map(NodeName1, "Item1", StringItem1),
 1276:              pubsub_payloads_row_map(NodeName2, "Item2", StringItem2)]),
 1277: 
 1278:         dynamic_modules:ensure_modules(host_type(), pubsub_required_modules()),
 1279:         Nodes = [{Alice, Node1}, {Alice, Node2}, {Bob, Node3}],
 1280:         [pubsub_tools:delete_node(User, Node, []) || {User, Node} <- Nodes]
 1281:       end).
 1282: 
 1283: 
 1284: retrieve_private_xml(Config) ->
 1285:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
 1286:             NS = <<"alice:gdpr:ns">>,
 1287:             Content = <<"dGhlcmUgYmUgZHJhZ29ucw==">>,
 1288:             send_and_assert_private_stanza(Alice, NS, Content),
 1289:             ExpectedHeader = ["ns", "xml"],
 1290:             ExpectedItems = [#{ "ns" => binary_to_list(NS),
 1291:                                 "xml" => [{contains, binary_to_list(NS)},
 1292:                                           {contains, binary_to_list(Content)}] }
 1293:                             ],
 1294:             retrieve_and_validate_personal_data(
 1295:               Alice, Config, "private", ExpectedHeader, ExpectedItems)
 1296:         end).
 1297: 
 1298: dont_retrieve_other_user_private_xml(Config) ->
 1299:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1300:             AliceNS = <<"alice:gdpr:ns">>,
 1301:             AliceContent = <<"To be or not to be">>,
 1302:             BobNS = <<"bob:gdpr:ns">>,
 1303:             BobContent = <<"This is the winter of our discontent">>,
 1304:             send_and_assert_private_stanza(Alice, AliceNS, AliceContent),
 1305:             send_and_assert_private_stanza(Bob, BobNS, BobContent),
 1306:             ExpectedHeader = ["ns", "xml"],
 1307:             ExpectedItems = [#{ "ns" => binary_to_list(AliceNS),
 1308:                                 "xml" => [{contains, binary_to_list(AliceNS)},
 1309:                                           {contains, binary_to_list(AliceContent)}] }
 1310:                             ],
 1311:             retrieve_and_validate_personal_data(
 1312:               Alice, Config, "private", ExpectedHeader, ExpectedItems)
 1313:         end).
 1314: 
 1315: retrieve_multiple_private_xmls(Config) ->
 1316:     escalus:fresh_story(Config, [{alice, 1}], fun(Alice) ->
 1317:             NSsAndContents = [
 1318:                               {<<"alice:gdpr:ns1">>, <<"You do not talk about FIGHT CLUB.">>},
 1319:                               {<<"alice:gdpr:ns2">>, <<"You do not talk about FIGHT CLUB.">>},
 1320:                               {<<"alice:gdpr:ns3">>, <<"If someone says stop or goes limp,"
 1321:                                                        " taps out the fight is over.">>},
 1322:                               {<<"alice:gdpr:ns4">>, <<"Only two guys to a fight.">>},
 1323:                               {<<"alice:gdpr:ns5">>, <<"One fight at a time.">>}
 1324:                              ],
 1325:             lists:foreach(
 1326:                 fun({NS, Content}) ->
 1327:                     send_and_assert_private_stanza(Alice, NS, Content)
 1328:                 end, NSsAndContents),
 1329:             ExpectedHeader = ["ns", "xml"],
 1330:             ExpectedItems = lists:map(
 1331:                 fun({NS, Content}) ->
 1332:                     #{ "ns" => binary_to_list(NS),
 1333:                        "xml" => [{contains, binary_to_list(NS)},
 1334:                                  {contains, binary_to_list(Content)}]}
 1335:                 end, NSsAndContents),
 1336: 
 1337:             retrieve_and_validate_personal_data(
 1338:               Alice, Config, "private", ExpectedHeader, ExpectedItems)
 1339:         end).
 1340: 
 1341: retrieve_inbox_muclight(Config) ->
 1342:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1343:         muc_light_helper:given_muc_light_room(?ROOM, Alice, [{Bob, member}]),
 1344:         Domain = muc_light_helper:muc_host(),
 1345: 
 1346:         Body = <<"Are you sure?">>,
 1347:         Res = muc_light_helper:when_muc_light_message_is_sent(Alice, ?ROOM, Body, <<"9128">>),
 1348:         muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob], Res),
 1349:         ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1350:         ExpectedAliceItems = [#{ "jid" => [{contains, <<?ROOM/binary, $@, Domain/binary>>}],
 1351:                                  "unread_count" => "0" }],
 1352:         %% MUC Light affiliations are also stored in inbox
 1353:         ExpectedBobItems = [#{ "jid" => [{contains, <<?ROOM/binary, $@, Domain/binary>>}],
 1354:                                "unread_count" => "2" }],
 1355: 
 1356:         retrieve_and_validate_personal_data(
 1357:           Alice, Config, "inbox", ExpectedHeader, ExpectedAliceItems),
 1358:         retrieve_and_validate_personal_data(
 1359:           Bob, Config, "inbox", ExpectedHeader, ExpectedBobItems),
 1360: 
 1361:         StanzaDestroy = escalus_stanza:to(escalus_stanza:iq_set(?NS_MUC_LIGHT_DESTROY, []), room_bin_jid(?ROOM)),
 1362:         escalus:send(Alice, StanzaDestroy),
 1363:         ok
 1364:         end).
 1365: 
 1366: retrieve_inbox_muc(Config) ->
 1367:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1368:         {ok, Room} = given_fresh_muc_room(Alice#client.props, []),
 1369:         Users = [Alice, Bob],
 1370:         Msg = <<"Hi Room!">>,
 1371:         Id = <<"MyID">>,
 1372:         RoomAddr = muc_helper:room_address(Room),
 1373: 
 1374:         inbox_helper:enter_room(Room, Users),
 1375:         inbox_helper:make_members(Room, Alice, [Bob]),
 1376:         Stanza = escalus_stanza:set_id(
 1377:           escalus_stanza:groupchat_to(RoomAddr, Msg), Id),
 1378:         escalus:send(Bob, Stanza),
 1379:         inbox_helper:wait_for_groupchat_msg(Users),
 1380: 
 1381:         ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1382: 
 1383:         ExpectedBobItems = [#{
 1384:                                 "content" => [{contains, Msg}],
 1385:                                 "jid" => [{contains, RoomAddr}],
 1386:                                 "unread_count" => "0" }],
 1387: 
 1388:          retrieve_and_validate_personal_data(
 1389:            Bob, Config, "inbox", ExpectedHeader, ExpectedBobItems),
 1390:         ExpectedAliceItems = [#{
 1391:                                 "content" => [{contains, Msg}],
 1392:                                 "jid" => [{contains, RoomAddr}],
 1393:                                 "unread_count" => "1" }],
 1394: 
 1395:          retrieve_and_validate_personal_data(
 1396:            Alice, Config, "inbox", ExpectedHeader, ExpectedAliceItems),
 1397:         ok
 1398:       end).
 1399: 
 1400: retrieve_inbox(Config) ->
 1401:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1402:             BobU = escalus_utils:jid_to_lower(escalus_client:username(Bob)),
 1403:             BobS = escalus_utils:jid_to_lower(escalus_client:server(Bob)),
 1404:             AliceU = escalus_utils:jid_to_lower(escalus_client:username(Alice)),
 1405:             AliceS = escalus_utils:jid_to_lower(escalus_client:server(Alice)),
 1406:             Body = <<"With spam?">>,
 1407:             send_and_assert_is_chat_message(Bob, Alice, Body),
 1408:             ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1409:             ExpectedAliceItems = [#{ "content" => [{contains, Body}],
 1410:                                      "jid" => [{contains, BobS},
 1411:                                                {contains, BobU}],
 1412:                                      "unread_count" => "1" }],
 1413:             ExpectedBobItems = [#{ "content" => [{contains, Body}],
 1414:                                     "jid" => [{contains, AliceS},
 1415:                                               {contains, AliceU}],
 1416:                                     "unread_count" => "0" }],
 1417:             retrieve_and_validate_personal_data(
 1418:               Alice, Config, "inbox", ExpectedHeader, ExpectedAliceItems),
 1419:             retrieve_and_validate_personal_data(
 1420:               Bob, Config, "inbox", ExpectedHeader, ExpectedBobItems)
 1421:         end).
 1422: 
 1423: remove_inbox(Config) ->
 1424:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1425:             AliceU = escalus_utils:jid_to_lower(escalus_client:username(Alice)),
 1426:             AliceS = escalus_utils:jid_to_lower(escalus_client:server(Alice)),
 1427:             Body = <<"With spam?">>,
 1428:             send_and_assert_is_chat_message(Bob, Alice, Body),
 1429: 
 1430:             ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1431: 
 1432:             unregister(Alice, Config),
 1433: 
 1434:             assert_personal_data_via_rpc(Alice, [{inbox, ExpectedHeader, []}]),
 1435: 
 1436:             ExpectedBobItems = [
 1437:                              #{ "content" => [{contains, Body}],
 1438:                                 "jid" => [{contains, AliceS},
 1439:                                           {contains, AliceU}],
 1440:                                 "unread_count" => "0" }
 1441:                             ],
 1442:             retrieve_and_validate_personal_data(
 1443:               Bob, Config, "inbox", ExpectedHeader, ExpectedBobItems)
 1444:         end).
 1445: 
 1446: remove_inbox_muclight(Config) ->
 1447:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1448:         Domain = muc_light_helper:muc_host(),
 1449:         Room = <<"ttt2">>,
 1450:         muc_light_helper:given_muc_light_room(Room, Alice, [{Bob, member}]),
 1451: 
 1452:         Body = <<"Are you sure?">>,
 1453:         Res = muc_light_helper:when_muc_light_message_is_sent(Alice, Room , Body, <<"9128">>),
 1454:         muc_light_helper:then_muc_light_message_is_received_by([Alice, Bob], Res),
 1455: 
 1456:         ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1457: 
 1458:         unregister(Alice, Config),
 1459: 
 1460:         %% MUC Light affiliations are also stored in inbox
 1461:         %% 1. Added to the room
 1462:         %% 2. Message
 1463:         %% 3. Aff change: Alice -> none, Bob -> owner
 1464:         %% Writing aff changes to inbox is enabled by default
 1465:         ExpectedBobItems = [#{
 1466:                                 "jid" => [{contains, <<Room/binary, $@, Domain/binary>>}],
 1467:                                 "unread_count" => "3" }
 1468:                            ],
 1469: 
 1470:         retrieve_and_validate_personal_data(
 1471:            Bob, Config, "inbox", ExpectedHeader, ExpectedBobItems),
 1472: 
 1473:         assert_personal_data_via_rpc(Alice, [{inbox, ExpectedHeader, []}]),
 1474: 
 1475:         timer:sleep(5000),
 1476:         StanzaDestroy = escalus_stanza:to(escalus_stanza:iq_set(?NS_MUC_LIGHT_DESTROY, []),
 1477:                                       room_bin_jid(Room)),
 1478:         escalus:send(Alice, StanzaDestroy),
 1479:         ok
 1480:         end).
 1481: 
 1482: remove_inbox_muc(Config) ->
 1483:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1484:         {ok, Room} = given_fresh_muc_room(Alice#client.props, []),
 1485: 
 1486:         Users = [Alice, Bob],
 1487:         Msg = <<"Hi Room!">>,
 1488:         Id = <<"MyID">>,
 1489:         RoomAddr = muc_helper:room_address(Room),
 1490: 
 1491:         inbox_helper:enter_room(Room, Users),
 1492:         inbox_helper:make_members(Room, Alice, [Bob]),
 1493:         Stanza = escalus_stanza:set_id(
 1494:           escalus_stanza:groupchat_to(RoomAddr, Msg), Id),
 1495:         escalus:send(Bob, Stanza),
 1496:         inbox_helper:wait_for_groupchat_msg(Users),
 1497: 
 1498:         ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1499: 
 1500:         unregister(Alice, Config),
 1501: 
 1502:         escalus:wait_for_stanza(Bob),
 1503:         assert_personal_data_via_rpc(Alice, [{inbox, ExpectedHeader, []}]),
 1504: 
 1505:         ExpectedBobItems = [#{
 1506:                                 "content" => [{contains, Msg}],
 1507:                                 "jid" => [{contains, RoomAddr}],
 1508:                                 "unread_count" => "0" }],
 1509: 
 1510:          retrieve_and_validate_personal_data(
 1511:            Bob, Config, "inbox", ExpectedHeader, ExpectedBobItems),
 1512:         ok
 1513:       end).
 1514: 
 1515: retrieve_inbox_for_multiple_messages(Config) ->
 1516:     escalus:fresh_story(Config, [{alice, 1}, {bob, 1}], fun(Alice, Bob) ->
 1517:             Bodies = [ <<"Nobody exists on purpose.">>,
 1518:                        <<"Nobody belongs anywhere.">>,
 1519:                        <<"We're all going to die.">>,
 1520:                        <<"Come watch TV.">>],
 1521:             lists:foreach(fun(Body) -> send_and_assert_is_chat_message(Bob, Alice, Body) end, Bodies),
 1522:             BobU = escalus_utils:jid_to_lower(escalus_client:username(Bob)),
 1523:             BobS = escalus_utils:jid_to_lower(escalus_client:server(Bob)),
 1524: 
 1525:             ExpectedHeader = ["jid", "content", "unread_count", "timestamp"],
 1526:             ExpectedAliceItems = [#{ "content" => [{contains, lists:last(Bodies)}],
 1527:                                      "jid" => [{contains, BobS},
 1528:                                                {contains, BobU}],
 1529:                                      "unread_count" => integer_to_list(length(Bodies)) }],
 1530:             retrieve_and_validate_personal_data(
 1531:               Alice, Config, "inbox", ExpectedHeader, ExpectedAliceItems)
 1532:         end).
 1533: 
 1534: retrieve_logs(Config) ->
 1535:     escalus:fresh_story(Config, [{alice, 1}],
 1536:         fun(Alice) ->
 1537:             User = string:to_lower(binary_to_list(escalus_client:username(Alice))),
 1538:             Domain = string:to_lower(binary_to_list(escalus_client:server(Alice))),
 1539:             JID = string:to_upper(binary_to_list(escalus_client:short_jid(Alice))),
 1540:             #{node := MIM2NodeName} = MIM2Node = distributed_helper:mim2(),
 1541:             mongoose_helper:successful_rpc(net_kernel, connect_node, [MIM2NodeName]),
 1542:             mongoose_helper:successful_rpc(MIM2Node, error_logger, error_msg,
 1543:                                            ["event=disturbance_in_the_force, jid=~s", [JID]]),
 1544:             Dir = request_and_unzip_personal_data(list_to_binary(User), list_to_binary(Domain),
 1545:                                                   Config),
 1546:             Filename = filename:join(Dir, "logs-" ++ atom_to_list(MIM2NodeName) ++ ".txt"),
 1547:             {ok, Content} = file:read_file(Filename),
 1548:             {match, _} = re:run(Content, "disturbance_in_the_force")
 1549:         end).
 1550: 
 1551: %% -------------------------------------------------------------
 1552: %% Internal functions
 1553: %% -------------------------------------------------------------
 1554: 
 1555: assert_personal_data_via_rpc(Client, ExpectedPersonalDataEntries) ->
 1556:     ExpectedKeys = [ Key || {Key, _, _} <- ExpectedPersonalDataEntries ],
 1557: 
 1558:     %% We use wait_until here, because e.g. the deletion in ElasticSearch
 1559:     %% sometimes is applied with a delay (i.e. immediately after successful deletion
 1560:     %% the data retrieval still returned valid entries)
 1561:     mongoose_helper:wait_until(
 1562:             fun() ->
 1563:                 get_personal_data_via_rpc(Client, ExpectedKeys)
 1564:             end, ExpectedPersonalDataEntries).
 1565: 
 1566: get_personal_data_via_rpc(Client, ExpectedKeys) ->
 1567:     ClientU = escalus_utils:jid_to_lower(escalus_client:username(Client)),
 1568:     ClientS = escalus_utils:jid_to_lower(escalus_client:server(Client)),
 1569:     AllPersonalData = mongoose_helper:successful_rpc(
 1570:                         gdpr_api, get_data_from_modules, [ClientU, ClientS]),
 1571:     %% We don't use lists:filter/2 because this line also ensures order
 1572:     [ lists:keyfind(Key, 1, AllPersonalData) || Key <- ExpectedKeys ].
 1573: 
 1574: retrieve_and_validate_personal_data(User, Config, FilePrefix, ExpectedHeader, ExpectedItems) ->
 1575:     Dir = retrieve_all_personal_data(User, Config),
 1576:     validate_personal_data(Dir, FilePrefix, ExpectedHeader, ExpectedItems, ExpectedHeader).
 1577: 
 1578: retrieve_and_validate_personal_data(User, Config, FilePrefix, ExpectedHeader, ExpectedItems, SortBy) ->
 1579:     Dir = retrieve_all_personal_data(User, Config),
 1580:     validate_personal_data(Dir, FilePrefix, ExpectedHeader, ExpectedItems, SortBy).
 1581: 
 1582: validate_personal_data(Dir, FilePrefix, ExpectedHeader, ExpectedItems, SortBy) ->
 1583:     PersonalCSV = decode_personal_data(Dir, FilePrefix),
 1584:     UnsortedMaps = csv_to_maps(ExpectedHeader, PersonalCSV),
 1585:     PersonalMaps = lists:sort(get_sort_fn(SortBy), UnsortedMaps),
 1586:     try validate_personal_maps(PersonalMaps, ExpectedItems) of
 1587:         _ -> ok
 1588:     catch
 1589:         C:R:S ->
 1590:             ct:fail(#{
 1591:                         class => C,
 1592:                         reason => R,
 1593:                         stacktrace => S,
 1594:                         sorted_by => SortBy,
 1595:                         personal_maps => PersonalMaps,
 1596:                         expected_items => ExpectedItems
 1597:                     })
 1598:     end.
 1599: 
 1600: 
 1601: get_sort_fn(SortBy) when is_list(SortBy) ->
 1602:     %% if SortBy is [], than original list remains unsorted.
 1603:     fun(Map1, Map2) -> compare_maps(SortBy, Map1, Map2) end;
 1604: get_sort_fn(SortFn) when is_function(SortFn, 2) ->
 1605:     SortFn.
 1606: 
 1607: compare_maps([], _, _) -> true;
 1608: compare_maps([Key | T], Map1, Map2) ->
 1609:     #{Key:=Val1} = Map1,
 1610:     #{Key:=Val2} = Map2,
 1611:     if
 1612:         Val1 =:= Val2 -> compare_maps(T, Map1, Map2);
 1613:         Val1 > Val2 -> false;
 1614:         Val1 < Val2 -> true
 1615:     end.
 1616: 
 1617: muc_msg_first(MucJid) ->
 1618:     MucJidNormalized = escalus_utils:jid_to_lower(to_binary(MucJid)),
 1619:     N = erlang:byte_size(MucJidNormalized),
 1620:     fun(#{"from" := JID1}, #{"from" := JID2}) ->
 1621:         Jid1Normalized = escalus_utils:jid_to_lower(to_binary(JID1)),
 1622:         Jid2Normalized = escalus_utils:jid_to_lower(to_binary(JID2)),
 1623:         case {Jid1Normalized, Jid2Normalized} of
 1624:             {<<MucJidNormalized:N/binary, _/binary>>, <<MucJidNormalized:N/binary, _/binary>>} ->
 1625:                 Jid1Normalized =< Jid2Normalized;
 1626:             {<<MucJidNormalized:N/binary, _/binary>>, _} ->
 1627:                 true;
 1628:             {_, <<MucJidNormalized:N/binary, _/binary>>} ->
 1629:                 false;
 1630:             {_, _} ->
 1631:                 Jid1Normalized =< Jid2Normalized
 1632:         end
 1633:     end.
 1634: 
 1635: csv_to_maps(ExpectedHeader, [ExpectedHeader | Rows]) ->
 1636:     lists:foldl(fun(Row, Maps) -> [ csv_row_to_map(ExpectedHeader, Row) | Maps ] end, [], Rows).
 1637: 
 1638: csv_row_to_map(Header, Row) ->
 1639:     maps:from_list(lists:zip(Header, Row)).
 1640: 
 1641: validate_personal_maps(PersonalMaps, ExpectedItems) ->
 1642:     validate_sorted_personal_maps(PersonalMaps, ExpectedItems).
 1643: 
 1644: validate_sorted_personal_maps([], []) -> ok;
 1645: validate_sorted_personal_maps(UnexpectedRecords, []) ->
 1646:     erlang:error("Unexpected records left ~p", [UnexpectedRecords]);
 1647: validate_sorted_personal_maps([Map | RMaps], [Checks | RChecks]) ->
 1648:     maps:fold(fun(K, Conditions, _) ->
 1649:                       validate_personal_item(maps:get(K, Map), Conditions)
 1650:               end, ok, Checks),
 1651:     validate_sorted_personal_maps(RMaps, RChecks).
 1652: 
 1653: validate_personal_item(_Value, []) ->
 1654:     ok;
 1655: validate_personal_item(ExactValue, ExactValue) ->
 1656:     ok;
 1657: validate_personal_item(Value, [{jid, ExpectedValue} | RConditions]) ->
 1658:     JID = escalus_utils:jid_to_lower(to_binary(Value)),
 1659:     JID = escalus_utils:jid_to_lower(to_binary(ExpectedValue)),
 1660:     validate_personal_item(Value, RConditions);
 1661: validate_personal_item(Value, [{contains, String} | RConditions]) ->
 1662:     {match, _} = re:run(Value, String),
 1663:     validate_personal_item(Value, RConditions);
 1664: validate_personal_item(Value, [{validate, Validator} | RConditions]) when is_function(Validator) ->
 1665:     true = Validator(Value),
 1666:     validate_personal_item(Value, RConditions).
 1667: 
 1668: to_binary(List) when is_list(List)       -> list_to_binary(List);
 1669: to_binary(Binary) when is_binary(Binary) -> Binary.
 1670: 
 1671: decode_personal_data(Dir, FilePrefix) ->
 1672:     CSVPath = filename:join(Dir, FilePrefix ++ ".csv"),
 1673:     {ok, Content} = file:read_file(CSVPath),
 1674:     % We expect non-empty list because it must contain at least header with columns names
 1675:     [_ | _] = csv:decode_binary(Content).
 1676: 
 1677: refute_personal_data(Client, Config, FilePrefix) ->
 1678:     Dir = retrieve_all_personal_data(Client, Config),
 1679:     refute_personal_data(Dir, FilePrefix).
 1680: 
 1681: refute_personal_data(Dir, FilePrefix) ->
 1682:     CSVPath = filename:join(Dir, FilePrefix ++ ".csv"),
 1683:     false = filelib:is_regular(CSVPath).
 1684: 
 1685: retrieve_all_personal_data(Client, Config) ->
 1686:     User = escalus_client:username(Client),
 1687:     Domain = escalus_client:server(Client),
 1688:     request_and_unzip_personal_data(User, Domain, Config).
 1689: 
 1690: request_and_unzip_personal_data(User, Domain, Config) ->
 1691:     {Filename, Res} = retrieve_personal_data(User, Domain, Config),
 1692:     ParsedResult = get_ok_value([data, gdpr, retrievePersonalData], Res),
 1693:     ?assertEqual(<<"Data retrieved">>, ParsedResult),
 1694:     FullPath = get_mim_cwd() ++ "/" ++ Filename,
 1695:     Dir = make_dir_name(Filename, User),
 1696:     ct:log("extracting logs ~s", [Dir]),
 1697:     {ok, _} = zip:extract(FullPath, [{cwd, Dir}]),
 1698:     Dir.
 1699: 
 1700: make_dir_name(Filename, User) when is_binary(User) ->
 1701:     make_dir_name(Filename, binary_to_list(User));
 1702: make_dir_name(Filename, User) when is_list(User) ->
 1703:     Filename ++ "." ++ User ++ ".unzipped".
 1704: 
 1705: retrieve_personal_data(User, Domain, Config) ->
 1706:     Filename = random_filename(Config),
 1707:     Vars = #{<<"username">> => User, <<"domain">> => Domain,
 1708:              <<"resultFilepath">> => list_to_binary(Filename)},
 1709:     Result = execute_command(<<"gdpr">>, <<"retrievePersonalData">>, Vars, Config),
 1710:     {Filename, Result}.
 1711: 
 1712: unregister(Client, Config) ->
 1713:     User = escalus_client:full_jid(Client),
 1714:     Path = [data, account, removeUser, message],
 1715:     Vars = #{<<"user">> => User},
 1716:     Resp = execute_command(<<"account">>, <<"removeUser">>, Vars, Config),
 1717:     ?assertNotEqual(nomatch, binary:match(get_ok_value(Path, Resp),
 1718:                                           <<"successfully unregister">>)).
 1719: 
 1720: random_filename(Config) ->
 1721:     TCName = atom_to_list(?config(tc_name, Config)),
 1722:     TCName ++ "." ++ integer_to_list(erlang:system_time()) ++ ".zip".
 1723: 
 1724: get_mim_cwd() ->
 1725:     {ok, Cwd} = rpc(mim(), file, get_cwd, []),
 1726:     Cwd.
 1727: 
 1728: delete_files() ->
 1729:     Cwd = get_mim_cwd(),
 1730:     {ok, Filenames} = rpc(mim(), file, list_dir, [Cwd]),
 1731:     FilteredFilenames = lists:filter(
 1732:         fun is_file_to_be_deleted/1,
 1733:         Filenames),
 1734:     lists:foreach(
 1735:         fun(Filename) -> rpc(mim(), file, delete, [Cwd ++ "/" ++ Filename]) end,
 1736:         FilteredFilenames),
 1737:     ok.
 1738: 
 1739: is_file_to_be_deleted(Filename) ->
 1740:     DeletableRegexes = ["\.csv", "\.zip"],
 1741:     lists:any(
 1742:         fun(Regex) ->
 1743:             re:run(Filename, Regex) =/= nomatch
 1744:         end,
 1745:     DeletableRegexes).
 1746: 
 1747: pubsub_payloads_row_map(Node, ItemId, Payload) ->
 1748:     #{"node_name" => binary_to_list(Node), "item_id" => ItemId, "payload" => Payload}.
 1749: 
 1750: pubsub_nodes_row_map(Node, Type) ->
 1751:     #{"node_name" => binary_to_list(Node), "type" => Type}.
 1752: 
 1753: pubsub_subscription_row_map(Node) ->
 1754:     #{"node_name" => binary_to_list(Node)}.
 1755: 
 1756: make_pep_node_info(Client, NodeName) ->
 1757:     {escalus_utils:jid_to_lower(escalus_utils:get_short_jid(Client)), NodeName}.
 1758: 
 1759: item_content(Data) ->
 1760:     Bin = item_content_xml(Data),
 1761:     {Bin, binary_to_list(exml:to_binary(Bin))}.
 1762: 
 1763: item_content_xml(Data) ->
 1764:     #xmlel{name = <<"entry">>,
 1765:            attrs = [{<<"xmlns">>, <<"http://www.w3.org/2005/Atom">>}],
 1766:            children = [#xmlcdata{content = Data}]}.
 1767: 
 1768: send_and_assert_private_stanza(User, NS, Content) ->
 1769:     XML = #xmlel{ name = <<"fingerprint">>,
 1770:                   attrs = [{<<"xmlns">>, NS}],
 1771:                   children = [#xmlcdata{ content = Content }]},
 1772:     PrivateStanza = escalus_stanza:private_set(XML),
 1773:     escalus_client:send(User, PrivateStanza),
 1774:     escalus:assert(is_iq_result, [PrivateStanza], escalus_client:wait_for_stanza(User)).
 1775: 
 1776: send_and_assert_is_chat_message(UserFrom, UserTo, Body) ->
 1777:     escalus:send(UserFrom, escalus_stanza:chat_to(UserTo, Body)),
 1778:     Msg = escalus:wait_for_stanza(UserTo),
 1779:     escalus:assert(is_chat_message, [Body], Msg).
 1780: 
 1781: validate_datetime(TimeStr) ->
 1782:     [Date, Time] = string:tokens(TimeStr, "T"),
 1783:     validate_date(Date),
 1784:     validate_time(Time).
 1785: 
 1786: validate_date(Date) ->
 1787:     [Y, M, D] = string:tokens(Date, "-"),
 1788:     Date1 = {list_to_integer(Y), list_to_integer(M), list_to_integer(D)},
 1789:     calendar:valid_date(Date1).
 1790: 
 1791: validate_time(Time) ->
 1792:   [T | _] = string:tokens(Time, "Z"),
 1793:   validate_time1(T).
 1794: 
 1795: 
 1796: validate_time1(Time) ->
 1797:     [H, M, S] = string:tokens(Time, ":"),
 1798:     check_list([{H, 24}, {M, 60}, {S, 60}]).
 1799: 
 1800: check_list(List) ->
 1801:     lists:all(fun({V, L}) -> I = list_to_integer(V), I >= 0 andalso I < L end, List).
 1802: 
 1803: expected_header(mod_roster) -> ["jid", "name", "subscription",
 1804:                               "ask", "groups", "askmessage", "xs"].
 1805: 
 1806: given_fresh_muc_room(UserSpec, RoomOpts) ->
 1807:     Username = proplists:get_value(username, UserSpec),
 1808:     RoomName = muc_helper:fresh_room_name(Username),
 1809:     From = muc_helper:generate_rpc_jid({user, UserSpec}),
 1810:     muc_helper:create_instant_room(RoomName, From, Username, RoomOpts),
 1811:     {ok, RoomName}.
 1812: 
 1813: send_receive_muc_private_message(Room, Domain, {User1, Nickname1}, {User2, Nickname2}, Text) ->
 1814:     RoomPrivAddrUser1 = <<Room/binary, "@", Domain/binary, "/", Nickname1/binary>>,
 1815:     RoomPrivAddrUser2 = <<Room/binary, "@", Domain/binary, "/", Nickname2/binary>>,
 1816:     Msg = escalus_stanza:chat_to(RoomPrivAddrUser2, Text),
 1817:     escalus:send(User1, Msg),
 1818:     PMStanza = escalus:wait_for_stanza(User2),
 1819:     escalus:assert(is_chat_message_from_to,
 1820:                    [RoomPrivAddrUser1, escalus_client:full_jid(User2), Text], PMStanza),
 1821:     {RoomPrivAddrUser1, RoomPrivAddrUser2}.