emqx4.3/apps/emqx_dashboard/test/emqx_dashboard_haproxy_SUITE.erl

%%--------------------------------------------------------------------
%% Copyright (c) 2020-2024 EMQ Technologies Co., Ltd. All Rights Reserved.
%%
%% Licensed under the Apache License, Version 2.0 (the "License");
%% you may not use this file except in compliance with the License.
%% You may obtain a copy of the License at
%%
%%     http://www.apache.org/licenses/LICENSE-2.0
%%
%% Unless required by applicable law or agreed to in writing, software
%% distributed under the License is distributed on an "AS IS" BASIS,
%% WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
%% See the License for the specific language governing permissions and
%% limitations under the License.
%%--------------------------------------------------------------------

-module(emqx_dashboard_haproxy_SUITE).

-compile(nowarn_export_all).
-compile(export_all).

-include_lib("eunit/include/eunit.hrl").
-include("emqx_dashboard.hrl").

all() ->
    emqx_common_test_helpers:all(?MODULE).

init_per_suite(Config) ->
    emqx_mgmt_api_test_util:init_suite([emqx_management], fun set_special_configs/1),
    Config.

set_special_configs(emqx_dashboard) ->
    emqx_dashboard_api_test_helpers:set_default_config(<<"admin">>, true),
    ok;
set_special_configs(_) ->
    ok.

end_per_suite(_Config) ->
    emqx_mgmt_api_test_util:end_suite([emqx_management]).

t_status(_Config) ->
    ProxyInfo = #{
        version => 1,
        command => proxy,
        transport_family => ipv4,
        transport_protocol => stream,
        src_address => {127, 0, 0, 1},
        src_port => 444,
        dest_address => {192, 168, 0, 1},
        dest_port => 443
    },
    {ok, Socket} = gen_tcp:connect(
        "localhost",
        18083,
        [binary, {active, false}, {packet, raw}]
    ),
    ok = gen_tcp:send(Socket, ranch_proxy_header:header(ProxyInfo)),
    {ok, _Role, Token} = emqx_dashboard_admin:sign_token(<<"admin">>, <<"public">>),
    ok = gen_tcp:send(
        Socket,
        "GET /status HTTP/1.1\r\n"
        "Host: localhost\r\n"
        "Authorization: Bearer " ++ binary_to_list(Token) ++
            "\r\n"
            "\r\n"
    ),
    {_, 200, _, Rest0} = cow_http:parse_status_line(raw_recv_head(Socket)),
    {Headers, Body0} = cow_http:parse_headers(Rest0),
    {_, LenBin} = lists:keyfind(<<"content-length">>, 1, Headers),
    Len = binary_to_integer(LenBin),
    Body =
        if
            byte_size(Body0) =:= Len ->
                Body0;
            true ->
                {ok, Body1} = gen_tcp:recv(Socket, Len - byte_size(Body0), 5000),
                <<Body0/bits, Body1/bits>>
        end,
    ?assertMatch({match, _}, re:run(Body, "Node .+ is started\nemqx is running")),
    ok.

raw_recv_head(Socket) ->
    {ok, Data} = gen_tcp:recv(Socket, 0, 10000),
    raw_recv_head(Socket, Data).

raw_recv_head(Socket, Buffer) ->
    case binary:match(Buffer, <<"\r\n\r\n">>) of
        nomatch ->
            {ok, Data} = gen_tcp:recv(Socket, 0, 10000),
            raw_recv_head(Socket, <<Buffer/binary, Data/binary>>);
        {_, _} ->
            Buffer
    end.