cu_cp,e1ap: Add unittest for cu-cp e1ap ue context

Author: FabianEckermann

tests/unittests/e1ap/cu_cp/CMakeLists.txt

@@ -11,6 +11,7 @@ target_include_directories(e1ap_cu_cp_test_helpers PRIVATE ${CMAKE_SOURCE_DIR})
 target_link_libraries(e1ap_cu_cp_test_helpers e1ap_test_helpers srsran_e1ap srsran_pcap srsran_support e1ap_asn1 srslog)
 
 set(SOURCES
+        e1ap_cu_cp_ue_context_test.cpp
         e1ap_cu_cp_setup_procedure_test.cpp
         e1ap_cu_cp_bearer_context_setup_procedure_test.cpp
         e1ap_cu_cp_bearer_context_modification_procedure_test.cpp

tests/unittests/e1ap/cu_cp/e1ap_cu_cp_ue_context_test.cpp

@@ -0,0 +1,130 @@
+/*
+ *
+ * Copyright 2021-2023 Software Radio Systems Limited
+ *
+ * By using this file, you agree to the terms and conditions set
+ * forth in the LICENSE file which can be found at the top level of
+ * the distribution.
+ *
+ */
+
+#include "../common/test_helpers.h"
+#include "lib/e1ap/cu_cp/ue_context/e1ap_cu_cp_ue_context.h"
+#include "srsran/support/executors/manual_task_worker.h"
+#include "srsran/support/test_utils.h"
+
+#include <gtest/gtest.h>
+
+using namespace srsran;
+using namespace srs_cu_cp;
+
+/// Fixture class for E1AP UE context
+class e1ap_cu_cp_ue_context_test : public ::testing::Test
+{
+protected:
+  e1ap_cu_cp_ue_context_test() { srslog::init(); };
+  ~e1ap_cu_cp_ue_context_test()
+  {
+    // flush logger after each test
+    srslog::flush();
+  };
+
+  ue_index_t generate_random_ue_index()
+  {
+    return uint_to_ue_index(
+        test_rgen::uniform_int<uint64_t>(ue_index_to_uint(ue_index_t::min), ue_index_to_uint(ue_index_t::max) - 1));
+  };
+
+  timer_manager        timer_mng;
+  manual_task_worker   ctrl_worker{128};
+  timer_factory        timers{timer_mng, ctrl_worker};
+  e1ap_ue_context_list ue_ctxt_list{timers};
+};
+
+TEST_F(e1ap_cu_cp_ue_context_test, when_ue_added_then_ue_exists)
+{
+  ue_index_t             ue_index         = generate_random_ue_index();
+  gnb_cu_cp_ue_e1ap_id_t cu_cp_ue_e1ap_id = generate_random_gnb_cu_cp_ue_e1ap_id();
+
+  ue_ctxt_list.add_ue(ue_index, cu_cp_ue_e1ap_id);
+
+  ASSERT_TRUE(ue_ctxt_list.contains(cu_cp_ue_e1ap_id));
+  ASSERT_TRUE(ue_ctxt_list.contains(ue_index));
+
+  ASSERT_EQ(ue_ctxt_list[cu_cp_ue_e1ap_id].cu_cp_ue_e1ap_id, cu_cp_ue_e1ap_id);
+  ASSERT_EQ(ue_ctxt_list[cu_cp_ue_e1ap_id].ue_index, ue_index);
+  ASSERT_EQ(ue_ctxt_list[ue_index].cu_cp_ue_e1ap_id, cu_cp_ue_e1ap_id);
+  ASSERT_EQ(ue_ctxt_list[ue_index].ue_index, ue_index);
+}
+
+TEST_F(e1ap_cu_cp_ue_context_test, when_ue_not_added_then_ue_doesnt_exist)
+{
+  ue_index_t             ue_index         = generate_random_ue_index();
+  gnb_cu_cp_ue_e1ap_id_t cu_cp_ue_e1ap_id = generate_random_gnb_cu_cp_ue_e1ap_id();
+
+  ASSERT_FALSE(ue_ctxt_list.contains(cu_cp_ue_e1ap_id));
+  ASSERT_FALSE(ue_ctxt_list.contains(ue_index));
+}
+
+TEST_F(e1ap_cu_cp_ue_context_test, when_unsupported_number_of_ues_addeded_then_ue_not_added)
+{
+  // Add maximum number of supported UEs
+  for (unsigned it = 0; it < MAX_NOF_CU_UES; ++it) {
+    gnb_cu_cp_ue_e1ap_id_t cu_cp_ue_e1ap_id = ue_ctxt_list.next_gnb_cu_cp_ue_e1ap_id();
+    ASSERT_NE(cu_cp_ue_e1ap_id, gnb_cu_cp_ue_e1ap_id_t::invalid);
+    ue_index_t ue_index = uint_to_ue_index(it);
+
+    ue_ctxt_list.add_ue(ue_index, cu_cp_ue_e1ap_id);
+
+    ASSERT_TRUE(ue_ctxt_list.contains(cu_cp_ue_e1ap_id));
+    ASSERT_TRUE(ue_ctxt_list.contains(ue_index));
+  }
+
+  // Try to get another cu_cp_ue_e1ap_id (should fail)
+  ASSERT_EQ(ue_ctxt_list.next_gnb_cu_cp_ue_e1ap_id(), gnb_cu_cp_ue_e1ap_id_t::invalid);
+}
+
+TEST_F(e1ap_cu_cp_ue_context_test, when_ue_exists_then_removal_succeeds)
+{
+  ue_index_t             ue_index         = generate_random_ue_index();
+  gnb_cu_cp_ue_e1ap_id_t cu_cp_ue_e1ap_id = generate_random_gnb_cu_cp_ue_e1ap_id();
+
+  ue_ctxt_list.add_ue(ue_index, cu_cp_ue_e1ap_id);
+
+  // test removal by gnb_cu_cp_ue_e1ap_id_t
+  ue_ctxt_list.remove_ue(cu_cp_ue_e1ap_id);
+
+  ASSERT_FALSE(ue_ctxt_list.contains(cu_cp_ue_e1ap_id));
+  ASSERT_FALSE(ue_ctxt_list.contains(ue_index));
+
+  ue_ctxt_list.add_ue(ue_index, cu_cp_ue_e1ap_id);
+
+  // test removal by ue_index
+  ue_ctxt_list.remove_ue(ue_index);
+
+  ASSERT_FALSE(ue_ctxt_list.contains(cu_cp_ue_e1ap_id));
+  ASSERT_FALSE(ue_ctxt_list.contains(ue_index));
+}
+
+TEST_F(e1ap_cu_cp_ue_context_test, when_ue_exists_then_ue_index_update_succeeds)
+{
+  ue_index_t             old_ue_index     = generate_random_ue_index();
+  gnb_cu_cp_ue_e1ap_id_t cu_cp_ue_e1ap_id = generate_random_gnb_cu_cp_ue_e1ap_id();
+
+  ue_ctxt_list.add_ue(old_ue_index, cu_cp_ue_e1ap_id);
+
+  // new ue index to update
+  ue_index_t ue_index = generate_random_ue_index();
+
+  // test ue index update
+  ue_ctxt_list.update_ue_index(ue_index, old_ue_index);
+
+  ASSERT_TRUE(ue_ctxt_list.contains(ue_index));
+  ASSERT_TRUE(ue_ctxt_list.contains(cu_cp_ue_e1ap_id));
+  ASSERT_FALSE(ue_ctxt_list.contains(old_ue_index));
+
+  ASSERT_EQ(ue_ctxt_list[cu_cp_ue_e1ap_id].cu_cp_ue_e1ap_id, cu_cp_ue_e1ap_id);
+  ASSERT_EQ(ue_ctxt_list[cu_cp_ue_e1ap_id].ue_index, ue_index);
+  ASSERT_EQ(ue_ctxt_list[ue_index].cu_cp_ue_e1ap_id, cu_cp_ue_e1ap_id);
+  ASSERT_EQ(ue_ctxt_list[ue_index].ue_index, ue_index);
+}