/**
 * Copyright 2013-2021 Software Radio Systems Limited
 *
 * This file is part of srsRAN.
 *
 * srsRAN is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 *
 * srsRAN is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * A copy of the GNU Affero General Public License can be found in
 * the LICENSE file in the top-level directory of this distribution
 * and at http://www.gnu.org/licenses/.
 *
 */

#include "srsran/adt/fsm.h"
#include "srsran/common/test_common.h"

/////////////////////////////

// Events
struct ev1 {};
struct ev2 {};

std::vector<std::string> calls;

template <typename State>
void call_log_helper(State* state, srslog::basic_logger& logger, const char* type)
{
  std::string callname = srsran::get_type_name<State>() + "::" + type;
  logger.info("%s custom called", callname.c_str());
  calls.push_back(callname);
}

class fsm1 : public srsran::fsm_t<fsm1>
{
public:
  uint32_t idle_enter_counter = 0, state1_enter_counter = 0, inner_enter_counter = 0;
  uint32_t foo_counter = 0;

  // states
  struct idle_st {
    void enter(fsm1* f);
  };
  struct state1 {
    void enter(fsm1* f) {}
    void enter(fsm1* f, const ev1& ev);
    void enter(fsm1* f, const ev2& ev);
    void exit(fsm1* f);
  };

  explicit fsm1(srslog::basic_logger& logger) : srsran::fsm_t<fsm1>(logger) {}

  // this state is another FSM
  class fsm2 : public subfsm_t<fsm2>
  {
  public:
    // states
    struct state_inner {
      void enter(fsm2* f)
      {
        call_log_helper(this, f->get_logger(), "enter");
        f->parent_fsm()->inner_enter_counter++;
      }
    };
    struct state_inner2 {
      void enter(fsm2* f) { call_log_helper(this, f->get_logger(), "enter"); }
      void exit(fsm2* f) { call_log_helper(this, f->get_logger(), "exit"); }
    };

    explicit fsm2(fsm1* f_) : composite_fsm_t(f_) {}
    fsm2(fsm2&&)  = default;
    fsm2& operator=(fsm2&&) = default;
    ~fsm2() { get_logger().info("%s being destroyed!", get_type_name(*this).c_str()); }

    void enter(fsm1* f) { call_log_helper(this, get_logger(), "enter"); }
    void exit(fsm1* f) { call_log_helper(this, get_logger(), "exit"); }

  private:
    void inner_action1(state_inner& s, const ev1& e);
    void inner_action2(state_inner& s, const ev2& e);
    void inner_action3(state_inner2& s, const ev2& e);

  protected:
    // list of states
    state_list<state_inner, state_inner2> states{this};
    // clang-format off
    using transitions = transition_table<
      //     Start       Target       Event    Action
      //  +------------+-------------+----+----------------------+
      upd<state_inner,                ev1, &fsm2::inner_action1  >,
      row<state_inner,  state_inner2, ev2, &fsm2::inner_action2  >,
      row<state_inner2, state1,       ev2, &fsm2::inner_action3  >
      //  +------------+-------------+----+----------------------+
    >;
    // clang-format on
  };

private:
  void action1(idle_st& s, const ev1& e);
  void action2(state1& s, const ev1& e);
  void action3(state1& s, const ev2& e);

protected:
  void foo(ev1 e) { foo_counter++; }

  // list of states + transitions
  state_list<idle_st, state1, fsm2> states = {this, idle_st{}, state1{}, fsm2{this}};

  // clang-format off
  using transitions = transition_table<
    //      Start         Target    Event        Action
    //  +------------+-------------+----+------------------+
    row< idle_st,     state1,       ev1,  &fsm1::action1   >,
    row< state1,      fsm2,         ev1,  &fsm1::action2   >,
    row< state1,      idle_st,      ev2,  &fsm1::action3   >
    //  +------------+-------------+----+------------------+
  >;
  // clang-format on
};

void fsm1::idle_st::enter(fsm1* f)
{
  call_log_helper(this, f->logger, "enter");
  f->idle_enter_counter++;
}
void fsm1::state1::enter(fsm1* f, const ev1& ev)
{
  call_log_helper(this, f->logger, "enter");
  f->state1_enter_counter++;
}
void fsm1::state1::enter(fsm1* f, const ev2& ev)
{
  call_log_helper(this, f->logger, "enter2");
  f->state1_enter_counter++;
}
void fsm1::state1::exit(fsm1* f)
{
  call_log_helper(this, f->logger, "exit");
}

// FSM event handlers
void fsm1::fsm2::inner_action1(state_inner& s, const ev1& e)
{
  call_log_helper(this, get_logger(), "inner_action1");
}

void fsm1::fsm2::inner_action2(state_inner& s, const ev2& e)
{
  call_log_helper(this, get_logger(), "inner_action2");
}

void fsm1::fsm2::inner_action3(state_inner2& s, const ev2& e)
{
  get_logger().info("fsm2::state_inner2::react called");
}

void fsm1::action1(idle_st& s, const ev1& e)
{
  call_log_helper(this, logger, "action1");
  foo(e);
}

void fsm1::action2(state1& s, const ev1& ev)
{
  call_log_helper(this, logger, "action2");
}

void fsm1::action3(state1& s, const ev2& ev)
{
  call_log_helper(this, logger, "action3");
}

// Static Checks

namespace srsran {
namespace fsm_details {

static_assert(is_fsm<fsm1>::value, "invalid metafunction");
static_assert(is_composite_fsm<fsm1::fsm2>::value, "invalid metafunction");
static_assert(type_list_size(typename filter_transition_type<ev1, fsm1::idle_st, fsm_transitions<fsm1> >::type{}) > 0,
              "invalid filter metafunction");
static_assert(
    std::is_same<fsm_state_list_type<fsm1>, fsm1::state_list<fsm1::idle_st, fsm1::state1, fsm1::fsm2> >::value,
    "get state list failed");
static_assert(fsm1::can_hold_state<fsm1::state1>(), "failed can_hold_state check");
static_assert(std::is_same<enable_if_fsm_state<fsm1, fsm1::idle_st>, void>::value, "get state list failed");
static_assert(std::is_same<disable_if_fsm_state<fsm1, fsm1::fsm2::state_inner>, void>::value, "get state list failed");

} // namespace fsm_details
} // namespace srsran

// Runtime checks

int test_hsm()
{
  srslog::basic_logger& logger = srslog::fetch_basic_logger("HSM", false);
  logger.set_level(srslog::basic_levels::info);

  fsm1 f{logger};
  TESTASSERT(f.idle_enter_counter == 1);
  TESTASSERT(get_type_name(f) == "fsm1");
  TESTASSERT(f.current_state_name() == "idle_st");
  TESTASSERT(f.is_in_state<fsm1::idle_st>());
  TESTASSERT(f.foo_counter == 0);

  // Moving Idle -> State1
  ev1 e;
  f.trigger(e);
  TESTASSERT(f.current_state_name() == "state1");
  TESTASSERT(f.is_in_state<fsm1::state1>());

  // Moving State1 -> fsm2
  f.trigger(e);
  TESTASSERT(f.current_state_name() == "fsm2");
  TESTASSERT(f.is_in_state<fsm1::fsm2>());
  TESTASSERT(f.get_if_current_state<fsm1::fsm2>()->current_state_name() == "state_inner");
  TESTASSERT(f.inner_enter_counter == 1);

  // Fsm2 does not listen to ev1
  f.trigger(e);
  TESTASSERT(f.current_state_name() == "fsm2");
  TESTASSERT(f.is_in_state<fsm1::fsm2>());
  TESTASSERT(f.get_if_current_state<fsm1::fsm2>()->current_state_name() == "state_inner");

  // Fsm2 state_inner -> state_inner2
  f.trigger(ev2{});
  TESTASSERT(f.current_state_name() == "fsm2");
  TESTASSERT(f.is_in_state<fsm1::fsm2>());
  TESTASSERT(f.get_if_current_state<fsm1::fsm2>()->current_state_name() == "state_inner2");

  // Moving fsm2 -> state1
  f.trigger(ev2{});
  TESTASSERT(f.current_state_name() == "state1");
  TESTASSERT(f.is_in_state<fsm1::state1>());
  TESTASSERT(f.state1_enter_counter == 2);

  // Moving state1 -> idle
  f.trigger(ev2{});
  TESTASSERT(std::string{f.current_state_name()} == "idle_st");
  TESTASSERT(f.is_in_state<fsm1::idle_st>());
  TESTASSERT(f.foo_counter == 1);
  TESTASSERT(f.idle_enter_counter == 2);

  // Call unhandled event
  f.trigger(ev2{});
  TESTASSERT(f.current_state_name() == "idle_st");

  // Enter fsm2 again
  f.trigger(ev1{});
  f.trigger(ev1{});
  TESTASSERT(f.current_state_name() == "fsm2");
  TESTASSERT(f.is_in_state<fsm1::fsm2>());
  TESTASSERT(f.get_if_current_state<fsm1::fsm2>()->current_state_name() == "state_inner");

  // Ensure correct call order
  TESTASSERT(calls[0] == srsran::get_type_name<fsm1::idle_st>() + "::enter"); // enter for init state called
  TESTASSERT(calls[1] == srsran::get_type_name<fsm1>() + "::action1");
  TESTASSERT(calls[2] == srsran::get_type_name<fsm1::state1>() + "::enter");
  TESTASSERT(calls[3] == srsran::get_type_name<fsm1>() + "::action2");
  TESTASSERT(calls[4] == srsran::get_type_name<fsm1::state1>() + "::exit");
  TESTASSERT(calls[5] == srsran::get_type_name<fsm1::fsm2>() + "::enter"); // entry is recursive
  TESTASSERT(calls[6] == srsran::get_type_name<fsm1::fsm2::state_inner>() + "::enter");
  TESTASSERT(calls[7] == srsran::get_type_name<fsm1::fsm2>() + "::inner_action1");
  TESTASSERT(calls[8] == srsran::get_type_name<fsm1::fsm2>() + "::inner_action2");
  TESTASSERT(calls[9] == srsran::get_type_name<fsm1::fsm2::state_inner2>() + "::enter");
  TESTASSERT(calls[10] == srsran::get_type_name<fsm1::fsm2::state_inner2>() + "::exit");
  TESTASSERT(calls[11] == srsran::get_type_name<fsm1::fsm2>() + "::exit");     // exit is recursive
  TESTASSERT(calls[12] == srsran::get_type_name<fsm1::state1>() + "::enter2"); // differentiates different entry funcs

  return SRSRAN_SUCCESS;
}

/////////////////////////////

struct procevent1 {
  bool is_success;
};

struct proc1 : public srsran::proc_fsm_t<proc1, int> {
public:
  struct procstate1 {
    void enter(proc1* f, const srsran::proc_launch_ev<int>& ev);
  };

  explicit proc1(srslog::basic_logger& logger) : base_t(logger) {}

protected:
  // Transitions
  void handle_success(procstate1& s, const procevent1& ev);
  void handle_failure(procstate1& s, const procevent1& ev);

  bool is_success(procstate1& s, const procevent1& ev) { return ev.is_success; }

  bool is_failure(procstate1& s, const procevent1& ev) { return not ev.is_success; }

  state_list<idle_st, procstate1> states{this, idle_st{}, procstate1{}};
  // clang-format off
  using transitions = transition_table<
    //      Start         Target      Event              Action                Guard (optional)
    //  +------------+-------------+----------------+------------------------+--------------------+
    row< idle_st,      procstate1,  launch_ev<int>                                                >,
    upd< procstate1,                procevent1,       &proc1::handle_success, &proc1::is_success  >,
    upd< procstate1,                procevent1,       &proc1::handle_failure, &proc1::is_failure  >,
    to_state<          idle_st,     complete_ev                                                   >
    //  +------------+-------------+----------------+------------------------+--------------------+
  >;
  // clang-format on
};

void proc1::procstate1::enter(proc1* f, const launch_ev<int>& ev)
{
  f->logger.info("started!");
}

void proc1::handle_success(procstate1& s, const procevent1& ev)
{
  logger.info("success!");
  trigger(complete_ev{5});
}

void proc1::handle_failure(procstate1& s, const procevent1& ev)
{
  logger.info("failure!");
  trigger(complete_ev{3});
}

struct proc_listener_fsm : public srsran::fsm_t<proc_listener_fsm> {
public:
  struct st1 {};
  struct st2 {};
  using proc1_st = srsran::proc_wait_st<proc1>;

  proc_listener_fsm(srslog::basic_logger& logger, proc1* proc_ptr_) :
    base_t(logger), states(this, st1{}, st2{}, proc1_st{proc_ptr_})
  {}

protected:
  bool is_success(proc1_st& s, const proc1::complete_ev& ev) { return ev.result; }

  // clang-format off
  state_list<st1, st2, proc1_st > states;
  using f = proc_listener_fsm;
  using transitions = transition_table<
    //      Start         Target             Event             Action                 Guard (optional)
    //  +--------------+--------------+-----------------------+------------------------+-------------------+
    row<  st1,           proc1_st,      int                                                                >,
    row<  proc1_st,      st2,           proc1::complete_ev,    nullptr,                 &f::is_success     >,
    row<  proc1_st,      st1,           proc1::complete_ev                                                 >
    //  +--------------+--------------+-----------------------+------------------------+-------------------+
  >;
  // clang-format on
};

int test_fsm_proc()
{
  proc1 proc{srslog::fetch_basic_logger("PROC", false)};
  proc.get_logger().set_level(srslog::basic_levels::info);
  proc.set_fsm_event_log_level(srslog::basic_levels::info);

  int v = 2;
  TESTASSERT(proc.current_state_name() == "idle_st");
  proc.trigger(srsran::proc_launch_ev<int>{v});
  TESTASSERT(proc.current_state_name() == "procstate1");
  proc.trigger(srsran::proc_launch_ev<int>{v});
  TESTASSERT(proc.current_state_name() == "procstate1");
  proc.trigger(srsran::proc_launch_ev<int>{5});
  TESTASSERT(proc.current_state_name() == "procstate1");
  proc.trigger(procevent1{true});
  TESTASSERT(proc.current_state_name() == "idle_st");
  TESTASSERT(proc.get_result() == 5);
  proc.trigger(srsran::proc_launch_ev<int>{v});
  TESTASSERT(proc.current_state_name() == "procstate1");
  proc.trigger(procevent1{false});
  TESTASSERT(proc.current_state_name() == "idle_st");
  TESTASSERT(proc.get_result() == 3);

  {
    proc_listener_fsm outer_fsm{srslog::fetch_basic_logger("TEST"), &proc};
    TESTASSERT(outer_fsm.is_in_state<proc_listener_fsm::st1>());
    outer_fsm.trigger(6);
    TESTASSERT(outer_fsm.is_in_state<proc_listener_fsm::proc1_st>());
    TESTASSERT(proc.is_running());
    proc.trigger(procevent1{true});
    TESTASSERT(not proc.is_running());
    TESTASSERT(outer_fsm.is_in_state<proc_listener_fsm::st2>());
  }

  {
    proc_listener_fsm outer_fsm{srslog::fetch_basic_logger("TEST"), &proc};
    TESTASSERT(outer_fsm.is_in_state<proc_listener_fsm::st1>());
    proc.trigger(srsran::proc_launch_ev<int>{v});
    TESTASSERT(proc.is_running());
    outer_fsm.trigger(7);
    TESTASSERT(outer_fsm.is_in_state<proc_listener_fsm::st1>());
    TESTASSERT(proc.is_running());
  }

  return SRSRAN_SUCCESS;
}

///////////////////////////

class nas_fsm : public srsran::fsm_t<nas_fsm>
{
public:
  // states
  struct emm_null_st {};
  struct emm_deregistered {};
  struct emm_deregistered_initiated {};
  struct emm_ta_updating_initiated {};
  struct emm_registered {};
  struct emm_service_req_initiated {};
  struct emm_registered_initiated {};

  // events
  struct enable_s1_ev {};
  struct disable_s1_ev {};
  struct attach_request_ev {};
  struct emm_registr_fail_ev {}; ///< attach rejected, network init DETACH request, lower layer failure
  struct attach_accept_ev {};    ///< attach accept and default EPS bearer context activated
  struct sr_initiated_ev {};
  struct sr_outcome_ev {};
  struct tau_request_ev {};
  struct tau_outcome_ev {};
  struct tau_reject_other_cause_ev {};
  struct power_off_ev {};
  struct detach_request_ev {};
  struct detach_accept_ev {};

  explicit nas_fsm(srslog::basic_logger& logger) : fsm_t<nas_fsm>(logger) {}

protected:
  state_list<emm_null_st,
             emm_deregistered,
             emm_registered_initiated,
             emm_registered,
             emm_service_req_initiated,
             emm_ta_updating_initiated,
             emm_deregistered_initiated>
      states{nullptr};
  // clang-format off
  using transitions = transition_table<
  //              Start                       Target                     Event
  //  +-----------------------------+-------------------------+-----------------------------+
    row< emm_null_st,                emm_deregistered,           enable_s1_ev              >,
    row< emm_deregistered,           emm_null_st,                disable_s1_ev             >,
    row< emm_deregistered,           emm_registered_initiated,   attach_request_ev         >,
    row< emm_registered_initiated,   emm_deregistered,           emm_registr_fail_ev       >,
    row< emm_registered_initiated,   emm_registered,             attach_accept_ev          >,
    row< emm_registered,             emm_service_req_initiated,  sr_initiated_ev           >,
    row< emm_service_req_initiated,  emm_registered,             sr_outcome_ev             >,
    row< emm_registered,             emm_ta_updating_initiated,  tau_request_ev            >,
    row< emm_registered,             emm_deregistered_initiated, detach_request_ev         >,
    row< emm_ta_updating_initiated,  emm_registered,             tau_outcome_ev            >,
    row< emm_ta_updating_initiated,  emm_deregistered,           tau_reject_other_cause_ev >,
    row< emm_deregistered_initiated, emm_deregistered,           detach_accept_ev          >,
    to_state<                        emm_deregistered,           power_off_ev              >
  //  +-----------------------------+-------------------------+-----------------------------+
  >;
  // clang-format on
};

int test_nas_fsm()
{
  auto& logger = srslog::fetch_basic_logger("NAS", false);
  logger.set_level(srslog::basic_levels::info);

  nas_fsm fsm{logger};

  TESTASSERT(fsm.is_in_state<nas_fsm::emm_null_st>());

  // NULL -> EMM-DEREGISTERED
  fsm.trigger(nas_fsm::enable_s1_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered>());

  // EMM-DEREGISTERED -> EMM-NULL -> EMM-DEREGISTERED
  fsm.trigger(nas_fsm::disable_s1_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_null_st>());
  fsm.trigger(nas_fsm::enable_s1_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered>());

  // EMM-DEREGISTERED -> EMM-REGISTERED-INITIATED -> EMM-DEREGISTERED
  fsm.trigger(nas_fsm::attach_request_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered_initiated>());
  fsm.trigger(nas_fsm::emm_registr_fail_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered>());

  // EMM-DEREGISTERED -> EMM-REGISTERED-INITIATED -> EMM-REGISTERED
  fsm.trigger(nas_fsm::attach_request_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered_initiated>());
  fsm.trigger(nas_fsm::attach_accept_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered>());

  // EMM-REGISTERED -> EMM-SERVICE-REQUEST-INITIATED -> EMM-REGISTERED
  fsm.trigger(nas_fsm::sr_initiated_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_service_req_initiated>());
  fsm.trigger(nas_fsm::sr_outcome_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered>());

  // EMM-REGISTERED -> EMM-TRACKING-AREA-UPDATING-INITIATED -> EMM-REGISTERED
  fsm.trigger(nas_fsm::tau_request_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_ta_updating_initiated>());
  fsm.trigger(nas_fsm::tau_outcome_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered>());

  // EMM-REGISTERED -> EMM-DEREGISTED-INITIATED -> EMM-DEREGISTERED
  fsm.trigger(nas_fsm::detach_request_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered_initiated>());
  fsm.trigger(nas_fsm::detach_accept_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered>());

  // EMM-DEREGISTERED -> EMM-REGISTERED-INITIATED -> EMM-REGISTERED -> EMM-TRACKING-AREA-UPDATING-INITIATED ->
  // EMM-DEREGISTERED
  fsm.trigger(nas_fsm::attach_request_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered_initiated>());
  fsm.trigger(nas_fsm::attach_accept_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_registered>());
  fsm.trigger(nas_fsm::tau_request_ev{});
  fsm.trigger(nas_fsm::tau_reject_other_cause_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered>());

  // EMM-DEREGISTERED -> EMM-REGISTERED-INITIATED -> EMM-REGISTERED -> EMM-SERVICE-REQUEST-INITIATED -> EMM-DEREGISTERED
  // (power-off)
  fsm.trigger(nas_fsm::attach_request_ev{});
  fsm.trigger(nas_fsm::attach_accept_ev{});
  fsm.trigger(nas_fsm::sr_initiated_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_service_req_initiated>());
  fsm.trigger(nas_fsm::power_off_ev{});
  TESTASSERT(fsm.is_in_state<nas_fsm::emm_deregistered>());

  return SRSRAN_SUCCESS;
}

struct fsm3 : public srsran::fsm_t<fsm3> {
  struct st1 {};
  struct st2 {
    int  counter = 0;
    void enter(fsm3* fsm)
    {
      counter++;
      fsm->events.push_back(fsm->current_state_name());
    }
  };

  fsm3() : base_t(srslog::fetch_basic_logger("TEST")) {}

  std::vector<std::string> events;

protected:
  void handle_ev1(st1& s, const ev1& ev)
  {
    trigger(ev2{});
    events.push_back(current_state_name() + "::action"); // still in st1
  }
  void handle_ev2(st2& s, const ev2& ev)
  {
    if (s.counter < 2) {
      trigger(ev1{});
    }
    events.push_back(current_state_name() + "::action"); // still in st2
  }

  state_list<st1, st2> states{this};
  // clang-format off
  using transitions = transition_table<
  //     Start                    Target                   Event               Action
  //  +------------------------+-------------------------+-------------------+--------------------+
  row<  st1,                      st2,                     ev1,                &fsm3::handle_ev1  >,
  row<  st2,                      st1,                     ev2,                &fsm3::handle_ev2  >
  //  +------------------------+-------------------------+-------------------+--------------------+
  >;
  // clang-format on
};

int test_fsm_self_trigger()
{
  fsm3 fsm;
  TESTASSERT(fsm.events.empty());

  TESTASSERT(fsm.is_in_state<fsm3::st1>());
  fsm.trigger(ev1{});
  TESTASSERT(fsm.is_in_state<fsm3::st1>());

  TESTASSERT(fsm.events.size() == 6);
  TESTASSERT(fsm.events[0] == "st1::action");
  TESTASSERT(fsm.events[1] == "st2");
  TESTASSERT(fsm.events[2] == "st2::action");
  TESTASSERT(fsm.events[3] == "st1::action");
  TESTASSERT(fsm.events[4] == "st2");
  TESTASSERT(fsm.events[5] == "st2::action");

  return SRSRAN_SUCCESS;
}

int main()
{
  srslog::basic_logger& logger = srslog::fetch_basic_logger("TEST", false);
  logger.set_level(srslog::basic_levels::info);
  srslog::init();

  TESTASSERT(test_hsm() == SRSRAN_SUCCESS);
  logger.info("TEST \"hsm\" finished successfully\n");
  TESTASSERT(test_fsm_proc() == SRSRAN_SUCCESS);
  logger.info("TEST \"proc\" finished successfully\n");
  TESTASSERT(test_nas_fsm() == SRSRAN_SUCCESS);
  logger.info("TEST \"nas fsm\" finished successfully\n");
  TESTASSERT(test_fsm_self_trigger() == SRSRAN_SUCCESS);
  logger.info("TEST \"fsm self trigger\" finished successfully\n");

  return SRSRAN_SUCCESS;
}