TD-799: Adds explicit woody dependency (#21)

2024-11-06 02:15:19 +00:00 · 2023-11-09 13:50:42 +03:00 · 2023-11-09 13:50:42 +03:00 · 4ad8fcac67
commit 4ad8fcac67
parent 211f16c46f
5 changed files with 102 additions and 92 deletions
--- a/apps/akm/test/akm_client.erl
+++ b/apps/akm/test/akm_client.erl
@ -1,5 +1,8 @@
 -module(akm_client).

+-include_lib("opentelemetry_api/include/otel_tracer.hrl").
+-include_lib("opentelemetry_api/include/opentelemetry.hrl").
+
 %% API
 -export([
    issue_key/4,
@ -12,30 +15,18 @@

 -spec issue_key(inet:hostname() | inet:ip_address(), inet:port_number(), binary(), map()) -> any().
 issue_key(Host, Port, PartyId, ApiKey) ->
+    perform_request(Host, Port, <<"issue_key">>, fun(ConnPid, Headers) ->
        Path = <<"/apikeys/v2/orgs/", PartyId/binary, "/api-keys">>,
        Body = jsx:encode(ApiKey),
-    Headers = [
-        {<<"X-Request-ID">>, <<"issue_key">>},
-        {<<"content-type">>, <<"application/json; charset=utf-8">>},
-        {<<"Authorization">>, <<"Bearer sffsdfsfsdfsdfs">>}
-    ],
-    ConnPid = connect(Host, Port),
-    Answer = post(ConnPid, Path, Headers, Body),
-    disconnect(ConnPid),
-    parse(Answer).
+        post(ConnPid, Path, Headers, Body)
+    end).

 -spec get_key(inet:hostname() | inet:ip_address(), inet:port_number(), binary(), binary()) -> any().
 get_key(Host, Port, PartyId, ApiKeyId) ->
+    perform_request(Host, Port, <<"get_key">>, fun(ConnPid, Headers) ->
        Path = <<"/apikeys/v2/orgs/", PartyId/binary, "/api-keys/", ApiKeyId/binary>>,
-    Headers = [
-        {<<"X-Request-ID">>, <<"get_key">>},
-        {<<"content-type">>, <<"application/json; charset=utf-8">>},
-        {<<"Authorization">>, <<"Bearer sffsdfsfsdfsdfs">>}
-    ],
-    ConnPid = connect(Host, Port),
-    Answer = get(ConnPid, Path, Headers),
-    disconnect(ConnPid),
-    parse(Answer).
+        get(ConnPid, Path, Headers)
+    end).

 -spec list_keys(inet:hostname() | inet:ip_address(), inet:port_number(), binary()) -> any().
 list_keys(Host, Port, PartyId) ->
@ -43,46 +34,45 @@ list_keys(Host, Port, PartyId) ->

 -spec list_keys(inet:hostname() | inet:ip_address(), inet:port_number(), binary(), list()) -> any().
 list_keys(Host, Port, PartyId, QsList) ->
+    perform_request(Host, Port, <<"list_keys">>, fun(ConnPid, Headers) ->
        Path = <<"/apikeys/v2/orgs/", PartyId/binary, "/api-keys">>,
-    Headers = [
-        {<<"X-Request-ID">>, <<"list_keys">>},
-        {<<"content-type">>, <<"application/json; charset=utf-8">>},
-        {<<"Authorization">>, <<"Bearer sffsdfsfsdfsdfs">>}
-    ],
        PathWithQuery = maybe_query(Path, QsList),
-    ConnPid = connect(Host, Port),
-    Answer = get(ConnPid, PathWithQuery, Headers),
-    disconnect(ConnPid),
-    parse(Answer).
+        get(ConnPid, PathWithQuery, Headers)
+    end).

 -spec request_revoke_key(inet:hostname() | inet:ip_address(), inet:port_number(), binary(), binary()) -> any().
 request_revoke_key(Host, Port, PartyId, ApiKeyId) ->
+    perform_request(Host, Port, <<"request_revoke">>, fun(ConnPid, Headers) ->
        Path = <<"/apikeys/v2/orgs/", PartyId/binary, "/api-keys/", ApiKeyId/binary, "/status">>,
-    Headers = [
-        {<<"X-Request-ID">>, <<"request_revoke">>},
-        {<<"content-type">>, <<"application/json; charset=utf-8">>},
-        {<<"Authorization">>, <<"Bearer sffsdfsfsdfsdfs">>}
-    ],
        Body = jsx:encode(#{<<"status">> => <<"revoked">>}),
-    ConnPid = connect(Host, Port),
-    Answer = put(ConnPid, Path, Headers, Body),
-    disconnect(ConnPid),
-    parse(Answer).
+        put(ConnPid, Path, Headers, Body)
+    end).

 -spec revoke_key(inet:hostname() | inet:ip_address(), inet:port_number(), binary()) -> any().
 revoke_key(Host, Port, PathWithQuery) ->
-    Headers = [
-        {<<"X-Request-ID">>, <<"request_revoke">>},
-        {<<"content-type">>, <<"application/json; charset=utf-8">>},
-        {<<"Authorization">>, <<"Bearer sffsdfsfsdfsdfs">>}
-    ],
-    ConnPid = connect(Host, Port),
-    Answer = get(ConnPid, PathWithQuery, Headers),
-    disconnect(ConnPid),
-    parse(Answer).
+    perform_request(Host, Port, <<"revoke_key">>, fun(ConnPid, Headers) ->
+        get(ConnPid, PathWithQuery, Headers)
+    end).

 % Internal functions

+perform_request(Host, Port, RequestID, F) ->
+    SpanName = iolist_to_binary(["client ", RequestID]),
+    ?with_span(SpanName, #{kind => ?SPAN_KIND_CLIENT}, fun(_SpanCtx) ->
+        Headers = prepare_headers(RequestID),
+        ConnPid = connect(Host, Port),
+        Answer = F(ConnPid, Headers),
+        disconnect(ConnPid),
+        parse(Answer)
+    end).
+
+prepare_headers(RequestID) ->
+    otel_propagator_text_map:inject([
+        {<<"X-Request-ID">>, RequestID},
+        {<<"content-type">>, <<"application/json; charset=utf-8">>},
+        {<<"Authorization">>, <<"Bearer sffsdfsfsdfsdfs">>}
+    ]).
+
 -spec connect(inet:hostname() | inet:ip_address(), inet:port_number()) -> any().
 connect(Host, Port) ->
    connect(Host, Port, #{}).
--- a/apps/akm/test/akm_cth.erl
+++ b/apps/akm/test/akm_cth.erl
@ -2,6 +2,8 @@

 -include_lib("common_test/include/ct.hrl").
 -include_lib("bouncer_proto/include/bouncer_ctx_thrift.hrl").
+-include_lib("bouncer_proto/include/bouncer_decision_thrift.hrl").
+-include_lib("token_keeper_proto/include/tk_token_keeper_thrift.hrl").

 -type config() :: [{atom(), term()}].
 -type test_case_name() :: atom().
@ -19,16 +21,16 @@
 ]).

 -define(ACCESS_TOKEN, <<"some.access.token">>).
-define(AUTH_CTX, #{
-    id => <<"auth_data_id">>,
-    token => ?ACCESS_TOKEN,
-    status => active,
-    context => #ctx_ContextFragment{type = 'v1_thrift_binary'},
-    metadata => #{
+-define(AUTH_CTX, #token_keeper_AuthData{
+    id = <<"auth_data_id">>,
+    token = ?ACCESS_TOKEN,
+    status = active,
+    context = #ctx_ContextFragment{type = 'v1_thrift_binary'},
+    metadata = #{
        <<"dev.vality.party.id">> => <<"some_party">>,
        <<"dev.vality.user.email">> => <<"box@mail.ru">>
    },
-    authority => <<"authority_id">>
+    authority = <<"authority_id">>
 }).

 -spec init(_, _) -> _.
@ -86,6 +88,7 @@ prepare_config(State) ->
            {transport, thrift},
            {bouncer_ruleset_id, <<"service/authz/api">>},
            {vault_token_path, WorkDir ++ "/rebar.config"},
+            {authority_id, <<"authority_id">>},
            {health_check, #{
                disk => {erl_health, disk, ["/", 99]},
                memory => {erl_health, cg_memory, [99]},
@ -111,9 +114,33 @@ prepare_config(State) ->
            }}
        ]},

+        {bouncer_client, [
+            {service_clients, #{
+                bouncer => #{url => <<"http://bouncer/v1/arbiter">>},
+                org_management => #{url => <<"http://org-management/org/v1/auth-context">>}
+            }}
+        ]},
+
+        {token_keeper_client, [
+            {service_clients, #{
+                authenticator => #{url => <<"http://authenticator/v2/authenticator">>},
+                authorities => #{
+                    offline => #{
+                        <<"authority_id">> => #{
+                            url => <<"http://authenticator/v2/authority/com.empayre.apikey">>
+                        }
+                    }
+                }
+            }}
+        ]},
+
        {canal, [
            {url, "http://vault:8200"},
            {engine, kvv2}
+        ]},
+
+        {opentelemetry, [
+            {span_processor, simple}
        ]}
    ],

@ -147,37 +174,23 @@ mock_services(State) ->
            }
        end
    ),
+
    meck:expect(
-        bouncer_client,
-        judge,
-        fun(_, _, _) -> allowed end
-    ),
-    meck:expect(
-        token_keeper_authority_offline,
-        create,
-        fun(_ID, _ContextFragment, _Metadata, _Client) ->
-            {ok, #{token => ?ACCESS_TOKEN}}
+        woody_client,
+        call,
+        fun
+            (_Request = {{_, 'Arbiter'}, 'Judge', _}, _Options, _Context) ->
+                {ok, #decision_Judgement{resolution = {allowed, #decision_ResolutionAllowed{}}}};
+            (_Request = {{_, 'TokenAuthenticator'}, 'Authenticate', _}, _Options, _Context) ->
+                {ok, ?AUTH_CTX};
+            (_Request = {{_, 'TokenAuthority'}, 'Create', _}, _Options, _Context) ->
+                {ok, ?AUTH_CTX};
+            (_Request = {{_, 'TokenAuthority'}, 'Revoke', _}, _Options, _Context) ->
+                {ok, ok};
+            (Request, Options, Context) ->
+                meck:passthrough([Request, Options, Context])
        end
    ),
-    meck:expect(
-        token_keeper_authority_offline,
-        revoke,
-        fun(_ID, _Client) ->
-            {ok, ok}
-        end
-    ),
-    meck:expect(
-        token_keeper_authenticator,
-        authenticate,
-        fun(_PreAuthContext, _TokenContext, _WoodyContext) ->
-            {ok, ?AUTH_CTX}
-        end
-    ),
-    meck:expect(
-        token_keeper_client,
-        offline_authority,
-        fun(_, _) -> #{} end
-    ),
    State.

 start_akm(State) ->
--- a/compose.tracing.yaml
+++ b/compose.tracing.yaml
@ -3,6 +3,12 @@ services:
    depends_on:
      jaeger:
        condition: service_healthy
+    environment:
+      - OTEL_SERVICE_NAME=api-key-mgmt-v2
+      - OTEL_TRACES_EXPORTER=otlp
+      - OTEL_TRACES_SAMPLER=parentbased_always_on
+      - OTEL_EXPORTER_OTLP_PROTOCOL=http_protobuf
+      - OTEL_EXPORTER_OTLP_ENDPOINT=http://jaeger:4318

  jaeger:
    image: jaegertracing/all-in-one:1.47
--- a/rebar.config
+++ b/rebar.config
@ -40,6 +40,7 @@
    {bouncer_proto, {git, "https://github.com/valitydev/bouncer-proto.git", {branch, "master"}}},
    {bouncer_client, {git, "https://github.com/valitydev/bouncer-client-erlang", {branch, "master"}}},
    {token_keeper_client, {git, "https://github.com/valitydev/token-keeper-client", {branch, "master"}}},
+    {woody, {git, "https://github.com/valitydev/woody_erlang.git", {branch, master}}},

    %% Libraries generated with swagger-codegen-erlang from valitydev/swag-api-keys
    {swag_server_apikeys,
--- a/rebar.lock
+++ b/rebar.lock
@ -10,7 +10,7 @@
  0},
 {<<"bouncer_client">>,
  {git,"https://github.com/valitydev/bouncer-client-erlang",
-       {ref,"4f89634d9a009cac807cf86a0eff9e80d33318ea"}},
+       {ref,"f7ab2fdf3bd39d447d4dd039280ef98937ce7c87"}},
  0},
 {<<"bouncer_proto">>,
  {git,"https://github.com/valitydev/bouncer-proto.git",
@ -131,7 +131,7 @@
  1},
 {<<"org_management_proto">>,
  {git,"https://github.com/valitydev/org-management-proto",
-       {ref,"03a269df4805fa604e8fd2d04241619a739e2ae3"}},
+       {ref,"04de2f4ad697430c75f8efa04716d30753bd7c4b"}},
  1},
 {<<"parse_trans">>,{pkg,<<"parse_trans">>,<<"3.4.1">>},1},
 {<<"pooler">>,{pkg,<<"pooler">>,<<"1.5.3">>},1},
@ -170,11 +170,11 @@
  1},
 {<<"token_keeper_client">>,
  {git,"https://github.com/valitydev/token-keeper-client",
-       {ref,"417916aab09baf0cd7f1a2a174945446203ea0bf"}},
+       {ref,"642b20e622ef5d3ed41127bdc9b2b0fc1d43c7e5"}},
  0},
 {<<"token_keeper_proto">>,
  {git,"https://github.com/valitydev/token-keeper-proto.git",
-       {ref,"094b4f05a4e220df79911c25093feffea1cb868b"}},
+       {ref,"8b8bb4333828350301ae2fe801c0c8de61c6529c"}},
  1},
 {<<"unicode_util_compat">>,{pkg,<<"unicode_util_compat">>,<<"0.7.0">>},2},
 {<<"uuid">>,
@ -183,7 +183,7 @@
  0},
 {<<"woody">>,
  {git,"https://github.com/valitydev/woody_erlang.git",
-       {ref,"68b191ed3655dbf40d0ba687f17f75ddd74e82da"}},
+       {ref,"5d46291a6bfcee0bae2a9346a7d927603a909249"}},
  0},
 {<<"woody_user_identity">>,
  {git,"https://github.com/valitydev/woody_erlang_user_identity.git",