wapi-lib/apps/wapi/test/wapi_identity_tests_SUITE.erl

-module(wapi_identity_tests_SUITE).

-include_lib("common_test/include/ct.hrl").

-include_lib("damsel/include/dmsl_webhooker_thrift.hrl").

-include_lib("jose/include/jose_jwk.hrl").
-include_lib("wapi_wallet_dummy_data.hrl").

-include_lib("fistful_proto/include/ff_proto_identity_thrift.hrl").
-include_lib("identdocstore_proto/include/identdocstore_identity_document_storage_thrift.hrl").

-export([all/0]).
-export([groups/0]).
-export([init_per_suite/1]).
-export([end_per_suite/1]).
-export([init_per_group/2]).
-export([end_per_group/2]).
-export([init_per_testcase/2]).
-export([end_per_testcase/2]).

-export([init/1]).

-export([
    create_identity/1,
    create_identity_thrift_name/1,
    get_identity/1,
    create_identity_challenge/1,
    get_identity_challenge/1,
    list_identity_challenges/1,
    get_identity_challenge_event/1,
    poll_identity_challenge_events/1
]).

% common-api is used since it is the domain used in production RN
% TODO: change to wallet-api (or just omit since it is the default one) when new tokens will be a thing
-define(DOMAIN, <<"common-api">>).
-define(badresp(Code), {error, {invalid_response_code, Code}}).
-define(emptyresp(Code), {error, {Code, #{}}}).

-type test_case_name()  :: atom().
-type config()          :: [{atom(), any()}].
-type group_name()      :: atom().

-behaviour(supervisor).

-spec init([]) ->
    {ok, {supervisor:sup_flags(), [supervisor:child_spec()]}}.
init([]) ->
    {ok, {#{strategy => one_for_all, intensity => 1, period => 1}, []}}.

-spec all() ->
    [test_case_name()].
all() ->
    [
        {group, base}
    ].

-spec groups() ->
    [{group_name(), list(), [test_case_name()]}].
groups() ->
    [
        {base, [],
            [
                create_identity,
                create_identity_thrift_name,
                get_identity,
                create_identity_challenge,
                get_identity_challenge,
                list_identity_challenges,
                get_identity_challenge_event,
                poll_identity_challenge_events
            ]
        }
    ].

%%
%% starting/stopping
%%
-spec init_per_suite(config()) -> config().

init_per_suite(C) ->
    wapi_ct_helper:init_suite(?MODULE, C).

-spec end_per_suite(config()) -> _.

end_per_suite(C) ->
    _ = wapi_ct_helper:stop_mocked_service_sup(?config(suite_test_sup, C)),
    _ = [application:stop(App) || App <- ?config(apps, C)],
    ok.

-spec init_per_group(group_name(), config()) ->
    config().
init_per_group(Group, Config) when Group =:= base ->
    ok = wapi_context:save(wapi_context:create(#{
        woody_context => woody_context:new(<<"init_per_group/", (atom_to_binary(Group, utf8))/binary>>)
    })),
    Party = genlib:bsuuid(),
    {ok, Token} = wapi_ct_helper:issue_token(Party, [{[party], write}], unlimited, ?DOMAIN),
    Config1 = [{party, Party} | Config],
    [{context, wapi_ct_helper:get_context(Token)} | Config1];
init_per_group(_, Config) ->
    Config.

-spec end_per_group(group_name(), config()) ->
    _.
end_per_group(_Group, _C) ->
    ok.

-spec init_per_testcase(test_case_name(), config()) ->
    config().
init_per_testcase(Name, C) ->
    C1 = wapi_ct_helper:makeup_cfg([wapi_ct_helper:test_case_name(Name), wapi_ct_helper:woody_ctx()], C),
    ok = wapi_context:save(C1),
    [{test_sup, wapi_ct_helper:start_mocked_service_sup(?MODULE)} | C1].

-spec end_per_testcase(test_case_name(), config()) ->
    config().
end_per_testcase(_Name, C) ->
    ok = wapi_context:cleanup(),
    wapi_ct_helper:stop_mocked_service_sup(?config(test_sup, C)),
    ok.

%%% Tests
-spec create_identity(config()) ->
    _.
create_identity(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {bender_thrift, fun('GenerateID', _) -> {ok, ?GENERATE_ID_RESULT} end},
        {fistful_identity, fun('Create', _) -> {ok, ?IDENTITY(PartyID)} end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:create_identity/3,
        #{
            body => #{
                <<"name">> => ?STRING,
                <<"class">> => ?STRING,
                <<"provider">> => ?STRING,
                <<"metadata">> => #{
                    <<"somedata">> => ?STRING
                }
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec create_identity_thrift_name(config()) ->
    _.
create_identity_thrift_name(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {bender_thrift, fun('GenerateID', _) -> {ok, ?GENERATE_ID_RESULT} end},
        {fistful_identity, fun('Create', _) -> {ok, ?IDENTITY(PartyID, ?DEFAULT_CONTEXT_NO_NAME(PartyID))} end}
    ], C),
    {ok, #{
        <<"name">> := ?STRING
    }} = call_api(
        fun swag_client_wallet_identities_api:create_identity/3,
        #{
            body => #{
                <<"name">> => ?STRING,
                <<"class">> => ?STRING,
                <<"provider">> => ?STRING,
                <<"metadata">> => #{
                    <<"somedata">> => ?STRING
                }
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec get_identity(config()) ->
    _.
get_identity(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {fistful_identity, fun('Get', _) -> {ok, ?IDENTITY(PartyID)} end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:get_identity/3,
        #{
            binding => #{
                <<"identityID">> => ?STRING
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec create_identity_challenge(config()) ->
    _.
create_identity_challenge(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {fistful_identity, fun
            ('GetContext', _) -> {ok, ?DEFAULT_CONTEXT(PartyID)};
            ('StartChallenge', _) -> {ok, ?IDENTITY_CHALLENGE(?IDENTITY_CHALLENGE_STATUS_COMPLETED)}
        end},
        {bender_thrift, fun('GenerateID', _) -> {ok, ?GENERATE_ID_RESULT} end},
        {identdoc_storage, fun('Get', _) -> {ok, ?IDENT_DOC} end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:start_identity_challenge/3,
        #{
            binding => #{
                <<"identityID">> => ?STRING
            },
            body => #{
                <<"type">> => <<"sword-initiation">>,
                <<"proofs">> => [
                    #{
                        <<"token">> => wapi_utils:map_to_base64url(#{
                            <<"type">> => <<"RUSRetireeInsuranceCertificate">>,
                            <<"token">> => ?STRING
                        })
                    },
                    #{
                        <<"token">> => wapi_utils:map_to_base64url(#{
                            <<"type">> => <<"RUSDomesticPassport">>,
                            <<"token">> => ?STRING
                        })
                    }
                ]
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec get_identity_challenge(config()) ->
    _.
get_identity_challenge(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {fistful_identity, fun
            ('GetContext', _) -> {ok, ?DEFAULT_CONTEXT(PartyID)};
            ('GetChallenges', _) -> {ok, [?IDENTITY_CHALLENGE(?IDENTITY_CHALLENGE_STATUS_COMPLETED)]}
        end},
        {identdoc_storage, fun('Get', _) -> {ok, ?IDENT_DOC} end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:get_identity_challenge/3,
        #{
            binding => #{
                <<"identityID">> => ?STRING,
                <<"challengeID">> => ?STRING
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec list_identity_challenges(config()) ->
    _.
list_identity_challenges(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {fistful_identity, fun
            ('GetContext', _) -> {ok, ?DEFAULT_CONTEXT(PartyID)};
            ('GetChallenges', _) -> {ok, [?IDENTITY_CHALLENGE(?IDENTITY_CHALLENGE_STATUS_COMPLETED)]}
        end},
        {identdoc_storage, fun('Get', _) -> {ok, ?IDENT_DOC} end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:list_identity_challenges/3,
        #{
            binding => #{
                <<"identityID">> => ?STRING
            },
            qs_val => #{
                <<"status">> => <<"Completed">>
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec get_identity_challenge_event(config()) ->
    _.
get_identity_challenge_event(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {fistful_identity, fun
            ('GetContext', _) -> {ok, ?DEFAULT_CONTEXT(PartyID)};
            ('GetEvents', _) -> {ok, [?IDENTITY_CHALLENGE_EVENT(?CHALLENGE_STATUS_CHANGE)]}
        end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:get_identity_challenge_event/3,
        #{
            binding => #{
                <<"identityID">> => ?STRING,
                <<"challengeID">> => ?STRING,
                <<"eventID">> => ?INTEGER
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

-spec poll_identity_challenge_events(config()) ->
    _.
poll_identity_challenge_events(C) ->
    PartyID = ?config(party, C),
    wapi_ct_helper:mock_services([
        {fistful_identity, fun
            ('GetContext', _) -> {ok, ?DEFAULT_CONTEXT(PartyID)};
            ('GetEvents', _) -> {ok, [?IDENTITY_CHALLENGE_EVENT(?CHALLENGE_STATUS_CHANGE)]}
        end}
    ], C),
    {ok, _} = call_api(
        fun swag_client_wallet_identities_api:poll_identity_challenge_events/3,
        #{
            binding => #{
                <<"identityID">> => ?STRING,
                <<"challengeID">> => ?STRING
            },
            qs_val => #{
                <<"limit">> => 551,
                <<"eventCursor">> => ?INTEGER
            }
        },
        wapi_ct_helper:cfg(context, C)
    ).

%%

-spec call_api(function(), map(), wapi_client_lib:context()) ->
    {ok, term()} | {error, term()}.
call_api(F, Params, Context) ->
    {Url, PreparedParams, Opts} = wapi_client_lib:make_request(Context, Params),
    Response = F(Url, PreparedParams, Opts),
    wapi_client_lib:handle_response(Response).