fluxer/fluxer_gateway/test/voice_utils_tests.erl

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

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

build_force_disconnect_rpc_request_test() ->
    Req = voice_utils:build_force_disconnect_rpc_request(123, 456, 789, <<"conn">>),
    ?assertEqual(<<"voice_force_disconnect_participant">>, maps:get(<<"type">>, Req)),
    ?assertEqual(<<"123">>, maps:get(<<"guild_id">>, Req)),
    ?assertEqual(<<"conn">>, maps:get(<<"connection_id">>, Req)).

build_update_participant_rpc_request_with_permissions_test() ->
    VoicePerms = #{
        can_speak => true,
        can_stream => false,
        can_video => false
    },
    Req = voice_utils:build_update_participant_rpc_request(
        123, 456, 789, true, false, VoicePerms
    ),
    ?assertEqual(true, maps:get(<<"can_speak">>, Req)),
    ?assertEqual(false, maps:get(<<"can_stream">>, Req)),
    ?assertEqual(false, maps:get(<<"can_video">>, Req)).

apply_voice_permissions_to_flags_suppresses_without_speak_test() ->
    Flags = #{self_mute => false, suppress => false},
    VoicePerms = #{
        can_speak => false,
        can_stream => true,
        can_video => true
    },
    Result = voice_utils:apply_voice_permissions_to_flags(Flags, VoicePerms),
    ?assertEqual(false, maps:get(self_mute, Result)),
    ?assertEqual(true, maps:get(suppress, Result)).

apply_voice_permissions_to_flags_clears_self_stream_without_stream_test() ->
    Flags = #{self_stream => true, self_video => true, suppress => false},
    VoicePerms = #{
        can_speak => true,
        can_stream => false,
        can_video => false
    },
    Result = voice_utils:apply_voice_permissions_to_flags(Flags, VoicePerms),
    ?assertEqual(false, maps:get(self_stream, Result)),
    ?assertEqual(false, maps:get(self_video, Result)),
    ?assertEqual(false, maps:get(suppress, Result)).

apply_voice_permissions_to_flags_keeps_self_stream_with_stream_test() ->
    Flags = #{self_stream => true, self_video => true, suppress => false},
    VoicePerms = #{
        can_speak => true,
        can_stream => true,
        can_video => true
    },
    Result = voice_utils:apply_voice_permissions_to_flags(Flags, VoicePerms),
    ?assertEqual(true, maps:get(self_stream, Result)),
    ?assertEqual(true, maps:get(self_video, Result)).

apply_voice_permissions_to_flags_clears_self_stream_denied_by_overwrite_test() ->
    UserId = 10,
    ChannelId = 500,
    State = stream_denied_overwrite_state(UserId, ChannelId),
    VoicePerms = voice_utils:compute_voice_permissions(UserId, ChannelId, State),
    ?assertEqual(true, maps:get(can_speak, VoicePerms)),
    ?assertEqual(false, maps:get(can_stream, VoicePerms)),
    Result = voice_utils:apply_voice_permissions_to_flags(
        #{self_stream => true, self_video => true, suppress => false}, VoicePerms
    ),
    ?assertEqual(false, maps:get(self_stream, Result)),
    ?assertEqual(false, maps:get(self_video, Result)),
    ?assertEqual(false, maps:get(suppress, Result)).

stream_denied_overwrite_state(UserId, ChannelId) ->
    GuildId = 90,
    RoleId = 300,
    RolePerms =
        constants:view_channel_permission() bor
            constants:connect_permission() bor
            constants:speak_permission() bor
            constants:stream_permission(),
    #{
        id => GuildId,
        data => #{
            <<"guild">> => #{<<"owner_id">> => <<"999">>},
            <<"roles">> => [
                #{<<"id">> => integer_to_binary(GuildId), <<"permissions">> => <<"0">>},
                #{
                    <<"id">> => integer_to_binary(RoleId),
                    <<"permissions">> => integer_to_binary(RolePerms)
                }
            ],
            <<"members">> => [
                #{
                    <<"user">> => #{<<"id">> => integer_to_binary(UserId)},
                    <<"roles">> => [integer_to_binary(RoleId)]
                }
            ],
            <<"channels">> => [
                #{
                    <<"id">> => integer_to_binary(ChannelId),
                    <<"permission_overwrites">> => [
                        #{
                            <<"id">> => integer_to_binary(RoleId),
                            <<"type">> => 0,
                            <<"allow">> => <<"0">>,
                            <<"deny">> => integer_to_binary(constants:stream_permission())
                        }
                    ]
                }
            ]
        }
    }.

build_voice_token_rpc_request_with_nonce_test() ->
    VoicePerms = #{
        can_speak => true,
        can_stream => false,
        can_video => false
    },
    Req = voice_utils:build_voice_token_rpc_request(
        123, 456, 789, null, null, null, VoicePerms, <<"test-nonce-123">>
    ),
    ?assertEqual(<<"test-nonce-123">>, maps:get(<<"token_nonce">>, Req)),
    ?assertEqual(true, maps:get(<<"can_speak">>, Req)),
    ?assertEqual(false, maps:get(<<"can_stream">>, Req)).