fluxer/fluxer_gateway/test/process_registry_cleanup_tests.erl

%% SPDX-License-Identifier: AGPL-3.0-or-later

-module(process_registry_cleanup_tests).
-typing([eqwalizer]).
-include_lib("eunit/include/eunit.hrl").

cleanup_on_down_preserves_loading_test() ->
    DeadPid = list_to_pid("<0.100.0>"),
    AlivePid = list_to_pid("<0.101.0>"),
    Ref1 = make_ref(),
    Ref2 = make_ref(),

    Map = #{
        guild_1 => {DeadPid, Ref1},
        guild_2 => loading,
        guild_3 => {AlivePid, Ref2}
    },

    Result = process_registry:cleanup_on_down(DeadPid, Map),

    ?assertEqual(2, maps:size(Result)),
    ?assertEqual(loading, maps:get(guild_2, Result)),
    ?assertEqual({AlivePid, Ref2}, maps:get(guild_3, Result)),
    ?assertEqual(false, maps:is_key(guild_1, Result)).

cleanup_on_down_duplicate_pids_test() ->
    DeadPid = list_to_pid("<0.100.0>"),
    Ref1 = make_ref(),
    Ref2 = make_ref(),

    Map = #{
        guild_1 => {DeadPid, Ref1},
        guild_2 => {DeadPid, Ref2}
    },

    Result = process_registry:cleanup_on_down(DeadPid, Map),
    ?assertEqual(0, maps:size(Result)).

typed_cleanup_on_down_unregisters_runtime_prefixes_test() ->
    process_registry:init(),
    DeadPid = list_to_pid("<0.100.0>"),
    RuntimeEntries = [
        {call, 9001},
        {guild, 9002},
        {presence, 9003},
        {session, <<"session-9004">>}
    ],
    try
        lists:foreach(
            fun({Prefix, Id}) ->
                Key = process_registry:build_process_key(Prefix, Id),
                ets:insert(process_registry_table, {Key, DeadPid}),
                Result = process_registry:cleanup_on_down(
                    Prefix, DeadPid, #{Id => {DeadPid, make_ref()}}
                ),
                ?assertEqual(#{}, Result),
                ?assertEqual([], ets:lookup(process_registry_table, Key))
            end,
            RuntimeEntries
        )
    after
        lists:foreach(
            fun({Prefix, Id}) ->
                process_registry:safe_unregister(process_registry:build_process_key(Prefix, Id))
            end,
            RuntimeEntries
        )
    end.

typed_cleanup_on_down_keeps_replacement_registry_pid_test() ->
    process_registry:init(),
    Id = 9101,
    Key = process_registry:build_process_key(presence, Id),
    DeadPid = list_to_pid("<0.100.0>"),
    ReplacementPid = list_to_pid("<0.101.0>"),
    try
        ets:insert(process_registry_table, {Key, ReplacementPid}),
        Result = process_registry:cleanup_on_down(
            presence, DeadPid, #{Id => {DeadPid, make_ref()}}
        ),
        ?assertEqual(#{}, Result),
        ?assertEqual([{Key, ReplacementPid}], ets:lookup(process_registry_table, Key))
    after
        process_registry:safe_unregister(Key)
    end.

typed_cleanup_on_down_ignores_unkeyable_ids_test() ->
    process_registry:init(),
    DeadPid = list_to_pid("<0.100.0>"),
    Result = process_registry:cleanup_on_down(
        guild,
        DeadPid,
        #{
            bad_id => {DeadPid, make_ref()},
            9201 => loading
        }
    ),
    ?assertEqual(#{9201 => loading}, Result).