-module(admin_handler_node_metrics_tests).
-include_lib("eunit/include/eunit.hrl").
-include("records.hrl").

-define(TABLES, [admin, admin_audit, node_metric, ticket]).
-define(ADMIN_ID, <<"adm_test_1">>).

setup() ->
  eh_test_support:start_mnesia(),
  eh_test_support:ensure_tables(?TABLES),
  eh_test_support:ensure_jwt(),
  eh_test_support:seed_admin(#{id => ?ADMIN_ID}),
  ok.

cleanup(_) ->
  eh_test_support:unload_cowboy(),
  eh_test_support:delete_tables(?TABLES),
  eh_test_support:stop_mnesia(),
  ok.

admin_node_metrics_test_() ->
  {foreach, fun setup/0, fun cleanup/1, [
    {"GET – empty range", {timeout, 60, fun test_empty/0}},
    {"GET – with seeded metric", {timeout, 60, fun test_with_data/0}},
    {"GET – missing from", {timeout, 60, fun test_missing_from/0}},
    {"GET – unauthorized", {timeout, 60, fun test_unauthorized/0}},
    {"POST – method not allowed", {timeout, 60, fun test_wrong_method/0}}
  ]}.

range_qs() ->
  From = handler_utils:datetime_to_iso8601(eh_test_support:days_from_now(-1)),
  To = handler_utils:datetime_to_iso8601(eh_test_support:days_from_now(1)),
  [{<<"from">>, From}, {<<"to">>, To}].

seed_metric() ->
  Now = calendar:universal_time(),
  Metric = #node_metric{
    timestamp = Now,
    node = node(),
    memory_total = 1000,
    memory_processes = 100,
    memory_atom = 10,
    memory_binary = 20,
    memory_ets = 30,
    process_count = 5,
    run_queue = 0,
    mnesia_commits = 1,
    mnesia_failures = 0,
    table_sizes = #{},
    active_sessions = 0,
    ws_connections = 0,
    io_in = 0,
    io_out = 0,
    memory_available = 500,
    cpu_utilization = 1.5
  },
  ok = core_node_metric:save(Metric),
  Metric.

test_empty() ->
  {Status, _, Body} = eh_test_support:call(admin_handler_node_metrics, #{
    method => <<"GET">>,
    path => <<"/v1/admin/nodes/metrics">>,
    qs => range_qs(),
    auth => ?ADMIN_ID
  }),
  ?assertEqual(200, Status),
  Result = jsx:decode(Body, [return_maps]),
  ?assertEqual([], Result).

test_with_data() ->
  _ = seed_metric(),
  {Status, _, Body} = eh_test_support:call(admin_handler_node_metrics, #{
    method => <<"GET">>,
    path => <<"/v1/admin/nodes/metrics">>,
    qs => range_qs(),
    auth => ?ADMIN_ID
  }),
  ?assertEqual(200, Status),
  Result = jsx:decode(Body, [return_maps]),
  ?assertEqual(1, length(Result)),
  [M] = Result,
  ?assertEqual(1000, maps:get(<<"memory_total">>, M)),
  ?assertEqual(5, maps:get(<<"process_count">>, M)).

test_missing_from() ->
  To = handler_utils:datetime_to_iso8601(eh_test_support:days_from_now(1)),
  {Status, _, _} = eh_test_support:call(admin_handler_node_metrics, #{
    method => <<"GET">>,
    path => <<"/v1/admin/nodes/metrics">>,
    qs => [{<<"to">>, To}],
    auth => ?ADMIN_ID
  }),
  ?assertEqual(400, Status).

test_unauthorized() ->
  {Status, _, _} = eh_test_support:call(admin_handler_node_metrics, #{
    method => <<"GET">>,
    path => <<"/v1/admin/nodes/metrics">>,
    qs => range_qs(),
    auth => none
  }),
  ?assertEqual(401, Status).

test_wrong_method() ->
  {Status, _, _} = eh_test_support:call(admin_handler_node_metrics, #{
    method => <<"POST">>,
    path => <<"/v1/admin/nodes/metrics">>,
    qs => range_qs(),
    auth => ?ADMIN_ID
  }),
  ?assertEqual(405, Status).