/*
 * Copyright (c) 2011 Centre Tecnologic de Telecomunicacions de Catalunya (CTTC)
 *
 * SPDX-License-Identifier: GPL-2.0-only
 *
 * Author: Manuel Requena <manuel.requena@cttc.es>
 *         Nicola Baldo <nbaldo@cttc.es>
 *         Marco Miozzo <mmiozzo@cttc.es>
 *         Budiarto Herman <budiarto.herman@magister.fi>
 */

#include "lte-test-ue-measurements.h"

#include "ns3/boolean.h"
#include "ns3/callback.h"
#include "ns3/config.h"
#include "ns3/double.h"
#include "ns3/enum.h"
#include "ns3/ff-mac-scheduler.h"
#include "ns3/internet-stack-helper.h"
#include "ns3/ipv4-address-helper.h"
#include "ns3/ipv4-interface-container.h"
#include "ns3/ipv4-static-routing-helper.h"
#include "ns3/log.h"
#include "ns3/lte-common.h"
#include "ns3/lte-enb-net-device.h"
#include "ns3/lte-enb-phy.h"
#include "ns3/lte-enb-rrc.h"
#include "ns3/lte-helper.h"
#include "ns3/lte-ue-net-device.h"
#include "ns3/lte-ue-phy.h"
#include "ns3/lte-ue-rrc.h"
#include "ns3/mobility-helper.h"
#include "ns3/net-device-container.h"
#include "ns3/node-container.h"
#include "ns3/point-to-point-epc-helper.h"
#include "ns3/point-to-point-helper.h"
#include "ns3/simulator.h"
#include "ns3/string.h"

using namespace ns3;

NS_LOG_COMPONENT_DEFINE("LteUeMeasurementsTest");

const Time UE_MEASUREMENT_REPORT_DELAY = MicroSeconds(1);

// ===== LTE-UE-MEASUREMENTS TEST SUITE ==================================== //

void
ReportUeMeasurementsCallback(LteUeMeasurementsTestCase* testcase,
                             std::string path,
                             uint16_t rnti,
                             uint16_t cellId,
                             double rsrp,
                             double rsrq,
                             bool servingCell,
                             uint8_t componentCarrierId)
{
    testcase->ReportUeMeasurements(rnti, cellId, rsrp, rsrq, servingCell);
}

void
RecvMeasurementReportCallback(LteUeMeasurementsTestCase* testcase,
                              std::string path,
                              uint64_t imsi,
                              uint16_t cellId,
                              uint16_t rnti,
                              LteRrcSap::MeasurementReport meas)
{
    testcase->RecvMeasurementReport(imsi, cellId, rnti, meas);
}

/*
 * Test Suite
 */

LteUeMeasurementsTestSuite::LteUeMeasurementsTestSuite()
    : TestSuite("lte-ue-measurements", Type::SYSTEM)
{
    AddTestCase(new LteUeMeasurementsTestCase("d1=10, d2=10000",
                                              10.000000,
                                              10000.000000,
                                              -53.739702,
                                              -113.739702,
                                              -3.010305,
                                              -63.010305),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=20, d2=10000",
                                              20.000000,
                                              10000.000000,
                                              -59.760302,
                                              -113.739702,
                                              -3.010319,
                                              -56.989719),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=50, d2=10000",
                                              50.000000,
                                              10000.000000,
                                              -67.719102,
                                              -113.739702,
                                              -3.010421,
                                              -49.031021),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=100, d2=10000",
                                              100.000000,
                                              10000.000000,
                                              -73.739702,
                                              -113.739702,
                                              -3.010783,
                                              -43.010783),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=200, d2=10000",
                                              200.000000,
                                              10000.000000,
                                              -79.760302,
                                              -113.739702,
                                              -3.012232,
                                              -36.991632),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=100, d2=10000",
                                              100.000000,
                                              10000.000000,
                                              -73.739702,
                                              -113.739702,
                                              -3.010783,
                                              -43.010783),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=200, d2=10000",
                                              200.000000,
                                              10000.000000,
                                              -79.760302,
                                              -113.739702,
                                              -3.012232,
                                              -36.991632),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=500, d2=10000",
                                              500.000000,
                                              10000.000000,
                                              -87.719102,
                                              -113.739702,
                                              -3.022359,
                                              -29.042959),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=1000, d2=10000",
                                              1000.000000,
                                              10000.000000,
                                              -93.739702,
                                              -113.739702,
                                              -3.058336,
                                              -23.058336),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=2000, d2=10000",
                                              2000.000000,
                                              10000.000000,
                                              -99.760302,
                                              -113.739702,
                                              -3.199337,
                                              -17.178738),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=5000, d2=10000",
                                              5000.000000,
                                              10000.000000,
                                              -107.719102,
                                              -113.739702,
                                              -4.075793,
                                              -10.096393),
                TestCase::Duration::QUICK);
    AddTestCase(new LteUeMeasurementsTestCase("d1=10000, d2=10000",
                                              10000.000000,
                                              10000.000000,
                                              -113.739702,
                                              -113.739702,
                                              -6.257687,
                                              -6.257687),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=20000, d2=10000",
                                              20000.000000,
                                              10000.000000,
                                              -119.760302,
                                              -113.739702,
                                              -10.373365,
                                              -4.352765),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=50000, d2=10000",
                                              50000.000000,
                                              10000.000000,
                                              -127.719102,
                                              -113.739702,
                                              -17.605046,
                                              -3.625645),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=100000, d2=10000",
                                              100000.000000,
                                              10000.000000,
                                              -133.739702,
                                              -113.739702,
                                              -23.511071,
                                              -3.511071),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=200000, d2=10000",
                                              200000.000000,
                                              10000.000000,
                                              -139.760302,
                                              -113.739702,
                                              -29.502549,
                                              -3.481949),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=500000, d2=10000",
                                              500000.000000,
                                              10000.000000,
                                              -147.719102,
                                              -113.739702,
                                              -37.453160,
                                              -3.473760),
                TestCase::Duration::EXTENSIVE);
    AddTestCase(new LteUeMeasurementsTestCase("d1=1000000, d2=10000",
                                              1000000.000000,
                                              10000.000000,
                                              -153.739702,
                                              -113.739702,
                                              -43.472589,
                                              -3.472589),
                TestCase::Duration::EXTENSIVE);
}

/**
 * @ingroup lte-test
 * Static variable for test initialization
 */
static LteUeMeasurementsTestSuite lteUeMeasurementsTestSuite;

/*
 * Test Case
 */

LteUeMeasurementsTestCase::LteUeMeasurementsTestCase(std::string name,
                                                     double d1,
                                                     double d2,
                                                     double rsrpDbmUe1,
                                                     double rsrpDbmUe2,
                                                     double rsrqDbUe1,
                                                     double rsrqDbUe2)
    : TestCase(name),
      m_d1(d1),
      m_d2(d2),
      m_rsrpDbmUeServingCell(rsrpDbmUe1),
      m_rsrpDbmUeNeighborCell(rsrpDbmUe2),
      m_rsrqDbUeServingCell(rsrqDbUe1),
      m_rsrqDbUeNeighborCell(rsrqDbUe2)
{
    NS_LOG_INFO("Test UE Measurements d1 = " << d1 << " m. and d2 = " << d2 << " m.");
}

LteUeMeasurementsTestCase::~LteUeMeasurementsTestCase()
{
}

void
LteUeMeasurementsTestCase::DoRun()
{
    NS_LOG_INFO(this << " " << GetName());

    Config::SetDefault("ns3::LteSpectrumPhy::CtrlErrorModelEnabled", BooleanValue(false));
    Config::SetDefault("ns3::LteSpectrumPhy::DataErrorModelEnabled", BooleanValue(false));
    Config::SetDefault("ns3::LteAmc::AmcModel", EnumValue(LteAmc::PiroEW2010));
    Config::SetDefault("ns3::LteAmc::Ber", DoubleValue(0.00005));
    Ptr<LteHelper> lteHelper = CreateObject<LteHelper>();
    lteHelper->SetAttribute("PathlossModel", StringValue("ns3::FriisSpectrumPropagationLossModel"));
    lteHelper->SetAttribute("UseIdealRrc", BooleanValue(false));

    // Disable Uplink Power Control
    Config::SetDefault("ns3::LteUePhy::EnableUplinkPowerControl", BooleanValue(false));

    // LogComponentEnable ("LteUeMeasurementsTest", LOG_LEVEL_ALL);

    // Create Nodes: eNodeB and UE
    NodeContainer enbNodes;
    NodeContainer ueNodes1;
    NodeContainer ueNodes2;
    enbNodes.Create(2);
    ueNodes1.Create(1);
    ueNodes2.Create(1);
    NodeContainer allNodes = NodeContainer(enbNodes, ueNodes1, ueNodes2);

    // the topology is the following:
    //         d2
    //  UE1-----------eNB2
    //   |             |
    // d1|             |d1
    //   |     d2      |
    //  eNB1----------UE2
    //
    Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator>();
    positionAlloc->Add(Vector(0.0, 0.0, 0.0));   // eNB1
    positionAlloc->Add(Vector(m_d2, m_d1, 0.0)); // eNB2
    positionAlloc->Add(Vector(0.0, m_d1, 0.0));  // UE1
    positionAlloc->Add(Vector(m_d2, 0.0, 0.0));  // UE2
    MobilityHelper mobility;
    mobility.SetMobilityModel("ns3::ConstantPositionMobilityModel");
    mobility.SetPositionAllocator(positionAlloc);
    mobility.Install(allNodes);

    // Create Devices and install them in the Nodes (eNB and UE)
    NetDeviceContainer enbDevs;
    NetDeviceContainer ueDevs1;
    NetDeviceContainer ueDevs2;
    lteHelper->SetSchedulerType("ns3::RrFfMacScheduler");
    lteHelper->SetSchedulerAttribute("UlCqiFilter", EnumValue(FfMacScheduler::PUSCH_UL_CQI));
    enbDevs = lteHelper->InstallEnbDevice(enbNodes);
    ueDevs1 = lteHelper->InstallUeDevice(ueNodes1);
    ueDevs2 = lteHelper->InstallUeDevice(ueNodes2);

    // Attach UEs to eNodeBs
    lteHelper->Attach(ueDevs1, enbDevs.Get(0));
    lteHelper->Attach(ueDevs2, enbDevs.Get(1));

    // Activate an EPS bearer
    EpsBearer::Qci q = EpsBearer::GBR_CONV_VOICE;
    EpsBearer bearer(q);
    lteHelper->ActivateDataRadioBearer(ueDevs1, bearer);
    lteHelper->ActivateDataRadioBearer(ueDevs2, bearer);

    Config::Connect(
        "/NodeList/2/DeviceList/0/ComponentCarrierMapUe/0/LteUePhy/ReportUeMeasurements",
        MakeBoundCallback(&ReportUeMeasurementsCallback, this));
    Config::Connect("/NodeList/0/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
                    MakeBoundCallback(&RecvMeasurementReportCallback, this));

    Config::Connect(
        "/NodeList/3/DeviceList/0/ComponentCarrierMapUe/0/LteUePhy/ReportUeMeasurements",
        MakeBoundCallback(&ReportUeMeasurementsCallback, this));
    Config::Connect("/NodeList/1/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
                    MakeBoundCallback(&RecvMeasurementReportCallback, this));

    // need to allow for RRC connection establishment + SRS
    Simulator::Stop(Seconds(0.800));
    Simulator::Run();

    Simulator::Destroy();
}

void
LteUeMeasurementsTestCase::ReportUeMeasurements(uint16_t rnti,
                                                uint16_t cellId,
                                                double rsrp,
                                                double rsrq,
                                                bool servingCell)
{
    // need to allow for RRC connection establishment + CQI feedback reception + UE measurements
    // filtering (200 ms)
    if (Simulator::Now() > MilliSeconds(400))
    {
        if (servingCell)
        {
            NS_LOG_DEBUG("UE serving cellId " << cellId << " Rxed RSRP " << rsrp << " thr "
                                              << m_rsrpDbmUeServingCell << " RSRQ " << rsrq
                                              << " thr " << m_rsrqDbUeServingCell);
            NS_TEST_ASSERT_MSG_EQ_TOL(m_rsrpDbmUeServingCell, rsrp, 0.2, "Wrong RSRP UE 1");
            NS_TEST_ASSERT_MSG_EQ_TOL(m_rsrqDbUeServingCell, rsrq, 0.2, "Wrong RSRQ UE 1");
        }
        else
        {
            NS_LOG_DEBUG("UE neighbor cellId " << cellId << " Rxed RSRP " << rsrp << " thr "
                                               << m_rsrpDbmUeNeighborCell << " RSRQ " << rsrq
                                               << " thr " << m_rsrqDbUeNeighborCell);
            NS_TEST_ASSERT_MSG_EQ_TOL(m_rsrpDbmUeNeighborCell, rsrp, 0.2, "Wrong RSRP UE 2");
            NS_TEST_ASSERT_MSG_EQ_TOL(m_rsrqDbUeNeighborCell, rsrq, 0.2, "Wrong RSRQ UE ");
        }
    }
}

void
LteUeMeasurementsTestCase::RecvMeasurementReport(uint64_t imsi,
                                                 uint16_t cellId,
                                                 uint16_t rnti,
                                                 LteRrcSap::MeasurementReport meas)
{
    // need to allow for RRC connection establishment + CQI feedback reception + UE measurements
    // filtering (200 ms)
    if (Simulator::Now() > MilliSeconds(400))
    {
        if (cellId == imsi)
        {
            NS_LOG_DEBUG(
                this << "Serving Cell: received IMSI " << imsi << " CellId " << cellId << " RNTI "
                     << rnti << " thr "
                     << (uint16_t)EutranMeasurementMapping::Dbm2RsrpRange(m_rsrpDbmUeServingCell)
                     << " RSRP " << (uint16_t)meas.measResults.measResultPCell.rsrpResult
                     << " RSRQ " << (uint16_t)meas.measResults.measResultPCell.rsrqResult << " thr "
                     << (uint16_t)EutranMeasurementMapping::Db2RsrqRange(m_rsrqDbUeServingCell));
            NS_TEST_ASSERT_MSG_EQ(meas.measResults.measResultPCell.rsrpResult,
                                  EutranMeasurementMapping::Dbm2RsrpRange(m_rsrpDbmUeServingCell),
                                  "Wrong RSRP ");
            NS_TEST_ASSERT_MSG_EQ(meas.measResults.measResultPCell.rsrqResult,
                                  EutranMeasurementMapping::Db2RsrqRange(m_rsrqDbUeServingCell),
                                  "Wrong RSRQ ");
        }
        else
        {
            NS_LOG_DEBUG(
                this << "Neighbor cell: received IMSI " << imsi << " CellId " << cellId << " RNTI "
                     << rnti << " thr "
                     << (uint16_t)EutranMeasurementMapping::Dbm2RsrpRange(m_rsrpDbmUeNeighborCell)
                     << " RSRP " << (uint16_t)meas.measResults.measResultPCell.rsrpResult
                     << " RSRQ " << (uint16_t)meas.measResults.measResultPCell.rsrqResult << " thr "
                     << (uint16_t)EutranMeasurementMapping::Db2RsrqRange(m_rsrqDbUeNeighborCell));
            NS_TEST_ASSERT_MSG_EQ(meas.measResults.measResultPCell.rsrpResult,
                                  EutranMeasurementMapping::Dbm2RsrpRange(m_rsrpDbmUeNeighborCell),
                                  "Wrong RSRP ");
            NS_TEST_ASSERT_MSG_EQ(meas.measResults.measResultPCell.rsrqResult,
                                  EutranMeasurementMapping::Db2RsrqRange(m_rsrqDbUeNeighborCell),
                                  "Wrong RSRQ ");
        }
    }
}

// ===== LTE-UE-MEASUREMENTS-PIECEWISE-1 TEST SUITE ======================== //

/*
 * Overloaded operators, for the convenience of defining test cases
 */

std::vector<Time>&
operator<<(std::vector<Time>& v, const uint64_t& ms)
{
    /*
     * Prior attempt to use seconds as unit of choice resulted in precision lost.
     * Therefore milliseconds are used now instead.
     */
    v.push_back(MilliSeconds(ms) + UE_MEASUREMENT_REPORT_DELAY);
    return v;
}

std::vector<uint8_t>&
operator<<(std::vector<uint8_t>& v, const uint8_t& range)
{
    v.push_back(range);
    return v;
}

/*
 * Test Suite
 */

LteUeMeasurementsPiecewiseTestSuite1::LteUeMeasurementsPiecewiseTestSuite1()
    : TestSuite("lte-ue-measurements-piecewise-1", Type::SYSTEM)
{
    std::vector<Time> expectedTime;
    std::vector<uint8_t> expectedRsrp;

    // === Event A1 (serving cell becomes better than threshold) ===

    // With very low threshold
    LteRrcSap::ReportConfigEutra config;
    config.triggerType = LteRrcSap::ReportConfigEutra::EVENT;
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    config.threshold1.choice = LteRrcSap::ThresholdEutra::THRESHOLD_RSRP;
    config.threshold1.range = 0;
    config.triggerQuantity = LteRrcSap::ReportConfigEutra::RSRP;
    config.reportInterval = LteRrcSap::ReportConfigEutra::MS120;
    expectedTime.clear();
    expectedTime << 200 << 320 << 440 << 560 << 680 << 800 << 920 << 1040 << 1160 << 1280 << 1400
                 << 1520 << 1640 << 1760 << 1880 << 2000 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 67 << 67 << 57 << 57 << 66 << 47 << 47 << 66 << 66 << 57 << 51 << 51 << 47 << 47
                 << 51 << 57 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A1 with very low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With normal threshold
    config.threshold1.range = 54;
    expectedTime.clear();
    expectedTime << 200 << 320 << 440 << 560 << 680 << 1000 << 1120 << 1240 << 1360 << 2000 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 67 << 67 << 57 << 57 << 66 << 66 << 66 << 57 << 57 << 57 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A1 with normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With short time-to-trigger
    config.timeToTrigger = 64;
    expectedTime.clear();
    expectedTime << 264 << 384 << 504 << 624 << 744 << 1064 << 1184 << 1304 << 1424 << 2064 << 2184;
    expectedRsrp.clear();
    expectedRsrp << 67 << 67 << 57 << 66 << 66 << 66 << 66 << 57 << 51 << 57 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A1 with short time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With long time-to-trigger
    config.timeToTrigger = 128;
    expectedTime.clear();
    expectedTime << 328 << 448 << 568 << 688 << 808 << 1128 << 1248 << 1368 << 1488 << 2128;
    expectedRsrp.clear();
    expectedRsrp << 67 << 57 << 57 << 66 << 47 << 66 << 57 << 57 << 51 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A1 with long time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With super time-to-trigger
    config.timeToTrigger = 256;
    expectedTime.clear();
    expectedTime << 456 << 576 << 696 << 816 << 936 << 1056 << 1176 << 1296 << 1416 << 1536;
    expectedRsrp.clear();
    expectedRsrp << 57 << 57 << 66 << 47 << 47 << 66 << 66 << 57 << 51 << 51;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A1 with super time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With hysteresis
    config.hysteresis = 8;
    config.timeToTrigger = 0;
    expectedTime.clear();
    expectedTime << 200 << 320 << 440 << 560 << 680 << 1000 << 1120 << 1240 << 1360 << 1480 << 2200;
    expectedRsrp.clear();
    expectedRsrp << 67 << 67 << 57 << 57 << 66 << 66 << 66 << 57 << 57 << 51 << 67;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase1("Piecewise test case 1 - Event A1 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::QUICK);

    // With very high threshold
    config.threshold1.range = 97;
    config.hysteresis = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A1 with very high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // === Event A2 (serving cell becomes worse than threshold) ===

    // With very low threshold
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    config.threshold1.range = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A2 with very low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With normal threshold
    config.threshold1.range = 54;
    expectedTime.clear();
    expectedTime << 800 << 920 << 1400 << 1520 << 1640 << 1760 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 47 << 47 << 51 << 51 << 47 << 47 << 51;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A2 with normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With short time-to-trigger
    config.timeToTrigger = 64;
    expectedTime.clear();
    expectedTime << 864 << 984 << 1464 << 1584 << 1704 << 1824 << 1944;
    expectedRsrp.clear();
    expectedRsrp << 47 << 47 << 51 << 51 << 47 << 51 << 51;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A2 with short time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With long time-to-trigger
    config.timeToTrigger = 128;
    expectedTime.clear();
    expectedTime << 928 << 1048 << 1528 << 1648 << 1768 << 1888 << 2008;
    expectedRsrp.clear();
    expectedRsrp << 47 << 66 << 51 << 47 << 47 << 51 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A2 with long time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With super time-to-trigger
    config.timeToTrigger = 256;
    expectedTime.clear();
    expectedTime << 1656 << 1776 << 1896 << 2016 << 2136;
    expectedRsrp.clear();
    expectedRsrp << 47 << 47 << 51 << 57 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A2 with super time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With hysteresis
    config.hysteresis = 8;
    config.timeToTrigger = 0;
    expectedTime.clear();
    expectedTime << 800 << 920 << 1600 << 1720 << 1840 << 1960 << 2080;
    expectedRsrp.clear();
    expectedRsrp << 47 << 47 << 47 << 47 << 51 << 51 << 57;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase1("Piecewise test case 1 - Event A2 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::EXTENSIVE);

    // With very high threshold
    config.threshold1.range = 97;
    config.hysteresis = 0;
    expectedTime.clear();
    expectedTime << 200 << 320 << 440 << 560 << 680 << 800 << 920 << 1040 << 1160 << 1280 << 1400
                 << 1520 << 1640 << 1760 << 1880 << 2000 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 67 << 67 << 57 << 57 << 66 << 47 << 47 << 66 << 66 << 57 << 51 << 51 << 47 << 47
                 << 51 << 57 << 57;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1(
                    "Piecewise test case 1 - Event A2 with very high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    /*
     * Event A3, A4, and A5 are not tested intensively here because they depend on
     * the existence of at least one neighbouring cell, which is not available in
     * this configuration. Piecewise configuration #2 includes a neighbouring
     * cell, hence more thorough tests on these events are performed there.
     */

    expectedTime.clear();
    expectedRsrp.clear();

    // === Event A3 (neighbour becomes offset better than PCell) ===

    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    config.a3Offset = 0;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1("Piecewise test case 1 - Event A3",
                                                        config,
                                                        expectedTime,
                                                        expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // === Event A4 (neighbour becomes better than threshold) ===

    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    config.threshold1.range = 54;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1("Piecewise test case 1 - Event A4",
                                                        config,
                                                        expectedTime,
                                                        expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // === Event A5 (PCell becomes worse than absolute threshold1 AND neighbour becomes better than
    // another absolute threshold2) ===

    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A5;
    config.threshold2.range = 58;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase1("Piecewise test case 1 - Event A5",
                                                        config,
                                                        expectedTime,
                                                        expectedRsrp),
                TestCase::Duration::EXTENSIVE);

} // end of LteUeMeasurementsPiecewiseTestSuite1::LteUeMeasurementsPiecewiseTestSuite1

/**
 * @ingroup lte-test
 * Static variable for test initialization
 */
static LteUeMeasurementsPiecewiseTestSuite1 lteUeMeasurementsPiecewiseTestSuite1;

/*
 * Test Case
 */

LteUeMeasurementsPiecewiseTestCase1::LteUeMeasurementsPiecewiseTestCase1(
    std::string name,
    LteRrcSap::ReportConfigEutra config,
    std::vector<Time> expectedTime,
    std::vector<uint8_t> expectedRsrp)
    : TestCase(name),
      m_config(config),
      m_expectedTime(expectedTime),
      m_expectedRsrp(expectedRsrp)
{
    // input sanity check
    uint16_t size = m_expectedTime.size();

    if (size != m_expectedRsrp.size())
    {
        NS_FATAL_ERROR("Vectors of expected results are not of the same size");
    }

    m_itExpectedTime = m_expectedTime.begin();
    m_itExpectedRsrp = m_expectedRsrp.begin();

    NS_LOG_INFO(this << " name=" << name);
}

LteUeMeasurementsPiecewiseTestCase1::~LteUeMeasurementsPiecewiseTestCase1()
{
    NS_LOG_FUNCTION(this);
}

void
LteUeMeasurementsPiecewiseTestCase1::DoRun()
{
    NS_LOG_INFO(this << " " << GetName());

    Ptr<LteHelper> lteHelper = CreateObject<LteHelper>();
    lteHelper->SetAttribute("PathlossModel", StringValue("ns3::FriisSpectrumPropagationLossModel"));
    lteHelper->SetAttribute("UseIdealRrc", BooleanValue(true));

    // Disable Uplink Power Control
    Config::SetDefault("ns3::LteUePhy::EnableUplinkPowerControl", BooleanValue(false));

    // Create Nodes: eNodeB and UE
    NodeContainer enbNodes;
    NodeContainer ueNodes;
    enbNodes.Create(1);
    ueNodes.Create(1);

    /*
     * The topology is the following:
     *
     * eNodeB     UE
     *    |       |
     *    x ----- x --------- x --------------- x ------------------- x
     *      100 m |   200 m   |      300 m      |        400 m        |
     *            |           |                 |                     |
     *         VeryNear      Near              Far                 VeryFar
     */

    Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator>();
    positionAlloc->Add(Vector(0.0, 0.0, 0.0));   // eNodeB
    positionAlloc->Add(Vector(100.0, 0.0, 0.0)); // UE
    MobilityHelper mobility;
    mobility.SetMobilityModel("ns3::ConstantPositionMobilityModel");
    mobility.SetPositionAllocator(positionAlloc);
    mobility.Install(enbNodes);
    mobility.Install(ueNodes);
    m_ueMobility = ueNodes.Get(0)->GetObject<MobilityModel>();

    // Disable layer-3 filtering
    Config::SetDefault("ns3::LteEnbRrc::RsrpFilterCoefficient", UintegerValue(0));

    // Create Devices and install them in the Nodes (eNB and UE)
    NetDeviceContainer enbDevs;
    NetDeviceContainer ueDevs;
    lteHelper->SetSchedulerType("ns3::RrFfMacScheduler");
    lteHelper->SetSchedulerAttribute("UlCqiFilter", EnumValue(FfMacScheduler::PUSCH_UL_CQI));
    enbDevs = lteHelper->InstallEnbDevice(enbNodes);
    ueDevs = lteHelper->InstallUeDevice(ueNodes);

    // Setup UE measurement configuration
    Ptr<LteEnbRrc> enbRrc = enbDevs.Get(0)->GetObject<LteEnbNetDevice>()->GetRrc();
    m_expectedMeasId = enbRrc->AddUeMeasReportConfig(m_config).at(0);

    // Attach UE to eNodeB
    lteHelper->Attach(ueDevs.Get(0), enbDevs.Get(0));

    // Activate an EPS bearer
    EpsBearer::Qci q = EpsBearer::GBR_CONV_VOICE;
    EpsBearer bearer(q);
    lteHelper->ActivateDataRadioBearer(ueDevs, bearer);

    // Connect to trace sources
    Config::Connect(
        "/NodeList/0/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
        MakeCallback(&LteUeMeasurementsPiecewiseTestCase1::RecvMeasurementReportCallback, this));

    /*
     * Schedule "teleports"
     *          0                   1                   2
     *          +-------------------+-------------------+---------> time
     * VeryNear |------  ----    ----                    --------
     *     Near |                    ----            ----
     *      Far |                        ----    ----
     *  VeryFar |      --    ----            ----
     */
    Simulator::Schedule(MilliSeconds(301),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportVeryFar,
                        this);
    Simulator::Schedule(MilliSeconds(401),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportVeryNear,
                        this);
    Simulator::Schedule(MilliSeconds(601),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportVeryFar,
                        this);
    Simulator::Schedule(MilliSeconds(801),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportVeryNear,
                        this);
    Simulator::Schedule(MilliSeconds(1001),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportNear,
                        this);
    Simulator::Schedule(MilliSeconds(1201),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportFar,
                        this);
    Simulator::Schedule(MilliSeconds(1401),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportVeryFar,
                        this);
    Simulator::Schedule(MilliSeconds(1601),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportFar,
                        this);
    Simulator::Schedule(MilliSeconds(1801),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportNear,
                        this);
    Simulator::Schedule(MilliSeconds(2001),
                        &LteUeMeasurementsPiecewiseTestCase1::TeleportVeryNear,
                        this);

    // Run simulation
    Simulator::Stop(Seconds(2.201));
    Simulator::Run();
    Simulator::Destroy();
}

void
LteUeMeasurementsPiecewiseTestCase1::DoTeardown()
{
    NS_LOG_FUNCTION(this);
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          true,
                          "Reporting should have occurred at " << m_itExpectedTime->As(Time::S));
    hasEnded = m_itExpectedRsrp == m_expectedRsrp.end();
    NS_ASSERT(hasEnded);
}

void
LteUeMeasurementsPiecewiseTestCase1::RecvMeasurementReportCallback(
    std::string context,
    uint64_t imsi,
    uint16_t cellId,
    uint16_t rnti,
    LteRrcSap::MeasurementReport report)
{
    NS_LOG_FUNCTION(this << context);
    NS_ASSERT(rnti == 1);
    NS_ASSERT(cellId == 1);

    if (report.measResults.measId == m_expectedMeasId)
    {
        // verifying the report completeness
        LteRrcSap::MeasResults measResults = report.measResults;
        NS_LOG_DEBUG(
            this << " rsrp=" << (uint16_t)measResults.measResultPCell.rsrpResult << " ("
                 << EutranMeasurementMapping::RsrpRange2Dbm(measResults.measResultPCell.rsrpResult)
                 << " dBm)"
                 << " rsrq=" << (uint16_t)measResults.measResultPCell.rsrqResult << " ("
                 << EutranMeasurementMapping::RsrqRange2Db(measResults.measResultPCell.rsrqResult)
                 << " dB)");
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              false,
                              "Report should not have neighboring cells information");
        NS_TEST_ASSERT_MSG_EQ(measResults.measResultListEutra.size(), 0, "Unexpected report size");

        bool hasEnded = m_itExpectedTime == m_expectedTime.end();
        NS_TEST_ASSERT_MSG_EQ(hasEnded,
                              false,
                              "Reporting should not have occurred at "
                                  << Simulator::Now().As(Time::S));
        if (!hasEnded)
        {
            hasEnded = m_itExpectedRsrp == m_expectedRsrp.end();
            NS_ASSERT(!hasEnded);

            // using milliseconds to avoid floating-point comparison
            uint64_t timeNowMs = Simulator::Now().GetMilliSeconds();
            uint64_t timeExpectedMs = m_itExpectedTime->GetMilliSeconds();
            m_itExpectedTime++;

            uint16_t observedRsrp = measResults.measResultPCell.rsrpResult;
            uint16_t referenceRsrp = *m_itExpectedRsrp;
            m_itExpectedRsrp++;

            NS_TEST_ASSERT_MSG_EQ(timeNowMs,
                                  timeExpectedMs,
                                  "Reporting should not have occurred at this time");
            NS_TEST_ASSERT_MSG_EQ(observedRsrp,
                                  referenceRsrp,
                                  "The RSRP observed differs with the reference RSRP");
        }
    }
}

void
LteUeMeasurementsPiecewiseTestCase1::TeleportVeryNear()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(100.0, 0.0, 0.0));
}

void
LteUeMeasurementsPiecewiseTestCase1::TeleportNear()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(300.0, 0.0, 0.0));
}

void
LteUeMeasurementsPiecewiseTestCase1::TeleportFar()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(600.0, 0.0, 0.0));
}

void
LteUeMeasurementsPiecewiseTestCase1::TeleportVeryFar()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(1000.0, 0.0, 0.0));
}

// ===== LTE-UE-MEASUREMENTS-PIECEWISE-2 TEST SUITE ======================== //

/*
 * Test Suite
 */

LteUeMeasurementsPiecewiseTestSuite2::LteUeMeasurementsPiecewiseTestSuite2()
    : TestSuite("lte-ue-measurements-piecewise-2", Type::SYSTEM)
{
    std::vector<Time> expectedTime;
    std::vector<uint8_t> expectedRsrp;

    /*
     * Higher level of fullness/duration are given to Event A1 and A2 because they
     * are supposed to be more intensively tested in Piecewise configuration #1.
     */

    // === Event A1 (serving cell becomes better than threshold) ===

    // With very low threshold
    LteRrcSap::ReportConfigEutra config;
    config.triggerType = LteRrcSap::ReportConfigEutra::EVENT;
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    config.threshold1.choice = LteRrcSap::ThresholdEutra::THRESHOLD_RSRP;
    config.threshold1.range = 0;
    config.triggerQuantity = LteRrcSap::ReportConfigEutra::RSRP;
    config.reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920 << 1160 << 1400 << 1640 << 1880 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 73 << 63 << 72 << 52 << 72 << 56 << 52 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A1 with very low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With normal threshold
    config.threshold1.range = 58;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 1000 << 1240 << 2000;
    expectedRsrp.clear();
    expectedRsrp << 73 << 63 << 72 << 72 << 59 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A1 with normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With hysteresis
    config.hysteresis = 6;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 1000 << 1240 << 1480 << 2200;
    expectedRsrp.clear();
    expectedRsrp << 73 << 63 << 72 << 72 << 59 << 56 << 72;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase2("Piecewise test case 2 - Event A1 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::EXTENSIVE);

    // With very high threshold
    config.threshold1.range = 97;
    config.hysteresis = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A1 with very high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // === Event A2 (serving cell becomes worse than threshold) ===

    // With very low threshold
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    config.threshold1.range = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A2 with very low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With normal threshold
    config.threshold1.range = 58;
    expectedTime.clear();
    expectedTime << 800 << 1400 << 1640 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 52 << 56;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A2 with normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With hysteresis
    config.hysteresis = 6;
    expectedTime.clear();
    expectedTime << 800 << 1600 << 1840 << 2080;
    expectedRsrp.clear();
    expectedRsrp << 52 << 52 << 56 << 59;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase2("Piecewise test case 2 - Event A2 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::EXTENSIVE);

    // With very high threshold
    config.threshold1.range = 97;
    config.hysteresis = 0;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920 << 1160 << 1400 << 1640 << 1880 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 73 << 63 << 72 << 52 << 72 << 56 << 52 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A2 with very high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // === Event A3 (neighbour becomes offset better than PCell) ===

    // With positive offset
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    config.threshold1.range = 0;
    config.a3Offset = 7;
    expectedTime.clear();
    expectedTime << 800 << 1600;
    expectedRsrp.clear();
    expectedRsrp << 52 << 52;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A3 with positive offset",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With zero offset
    config.a3Offset = 0;
    expectedTime.clear();
    expectedTime << 800 << 1400 << 1640 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 52 << 56;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase2("Piecewise test case 2 - Event A3 with zero offset",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::EXTENSIVE);

    // With short time-to-trigger
    config.timeToTrigger = 160;
    expectedTime.clear();
    expectedTime << 960 << 1560 << 1800 << 2040;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A3 with short time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With super time-to-trigger
    config.timeToTrigger = 320;
    expectedTime.clear();
    expectedTime << 1720 << 1960 << 2200;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 72;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A3 with super time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With hysteresis and reportOnLeave
    config.hysteresis = 6;
    config.reportOnLeave = true;
    config.timeToTrigger = 0;
    expectedTime.clear();
    expectedTime << 800 << 1000 << 1600 << 1840 << 2080 << 2200;
    expectedRsrp.clear();
    expectedRsrp << 52 << 72 << 52 << 56 << 59 << 72;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase2("Piecewise test case 2 - Event A3 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::QUICK);

    // With negative offset
    config.a3Offset = -7;
    config.hysteresis = 0;
    config.reportOnLeave = false;
    expectedTime.clear();
    expectedTime << 400 << 800 << 1200 << 1440 << 1680 << 1920 << 2160;
    expectedRsrp.clear();
    expectedRsrp << 63 << 52 << 59 << 56 << 52 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A3 with negative offset",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // === Event A4 (neighbour becomes better than threshold) ===

    // With very low threshold
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    config.threshold1.range = 0;
    config.a3Offset = 0;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920 << 1160 << 1400 << 1640 << 1880 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 73 << 63 << 72 << 52 << 72 << 56 << 52 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A4 with very low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With normal threshold
    config.threshold1.range = 58;
    expectedTime.clear();
    expectedTime << 400 << 800 << 1400 << 1640 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 63 << 52 << 56 << 52 << 56;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A4 with normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With short time-to-trigger
    config.timeToTrigger = 160;
    expectedTime.clear();
    expectedTime << 560 << 960 << 1560 << 1800 << 2040;
    expectedRsrp.clear();
    expectedRsrp << 63 << 52 << 56 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A4 with short time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With super time-to-trigger
    config.timeToTrigger = 320;
    expectedTime.clear();
    expectedTime << 1720 << 1960 << 2200;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 72;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A4 with super time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With hysteresis
    config.hysteresis = 6;
    config.timeToTrigger = 0;
    expectedTime.clear();
    expectedTime << 400 << 800 << 1600 << 1840 << 2080;
    expectedRsrp.clear();
    expectedRsrp << 63 << 52 << 52 << 56 << 59;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase2("Piecewise test case 2 - Event A4 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::QUICK);

    // With very high threshold
    config.threshold1.range = 97;
    config.hysteresis = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A4 with very high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // === Event A5 (PCell becomes worse than absolute threshold1 AND neighbour becomes better than
    // another absolute threshold2) ===

    // With low-low threshold
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A5;
    config.threshold1.range = 0;
    config.threshold2.range = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with low-low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With low-normal threshold
    config.threshold2.range = 58;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with low-normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With low-high threshold
    config.threshold2.range = 97;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with low-high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With normal-low threshold
    config.threshold1.range = 58;
    config.threshold2.range = 0;
    expectedTime.clear();
    expectedTime << 800 << 1400 << 1640 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 52 << 56;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with normal-low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With normal-normal threshold
    config.threshold2.range = 58;
    expectedTime.clear();
    expectedTime << 800 << 1400 << 1640 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 52 << 56;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with normal-normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With short time-to-trigger
    config.timeToTrigger = 160;
    expectedTime.clear();
    expectedTime << 960 << 1560 << 1800 << 2040;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with short time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With super time-to-trigger
    config.timeToTrigger = 320;
    expectedTime.clear();
    expectedTime << 1720 << 1960 << 2200;
    expectedRsrp.clear();
    expectedRsrp << 52 << 56 << 72;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with super time-to-trigger",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::QUICK);

    // With hysteresis
    config.hysteresis = 6;
    config.timeToTrigger = 0;
    expectedTime.clear();
    expectedTime << 800 << 1600 << 1840 << 2080;
    expectedRsrp.clear();
    expectedRsrp << 52 << 52 << 56 << 59;
    AddTestCase(
        new LteUeMeasurementsPiecewiseTestCase2("Piecewise test case 2 - Event A5 with hysteresis",
                                                config,
                                                expectedTime,
                                                expectedRsrp),
        TestCase::Duration::QUICK);

    // With normal-high threshold
    config.threshold2.range = 97;
    config.hysteresis = 0;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with normal-high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With high-low threshold
    config.threshold1.range = 97;
    config.threshold2.range = 0;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920 << 1160 << 1400 << 1640 << 1880 << 2120;
    expectedRsrp.clear();
    expectedRsrp << 73 << 63 << 72 << 52 << 72 << 56 << 52 << 56 << 59;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with high-low threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

    // With high-normal threshold
    config.threshold2.range = 58;
    expectedTime.clear();
    expectedTime << 400 << 800 << 1400 << 1640 << 1880;
    expectedRsrp.clear();
    expectedRsrp << 63 << 52 << 56 << 52 << 56;
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with high-normal threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::TAKES_FOREVER);

    // With high-high threshold
    config.threshold2.range = 97;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsPiecewiseTestCase2(
                    "Piecewise test case 2 - Event A5 with high-high threshold",
                    config,
                    expectedTime,
                    expectedRsrp),
                TestCase::Duration::EXTENSIVE);

} // end of LteUeMeasurementsPiecewiseTestSuite2::LteUeMeasurementsPiecewiseTestSuite2

/**
 * @ingroup lte-test
 * Static variable for test initialization
 */
static LteUeMeasurementsPiecewiseTestSuite2 lteUeMeasurementsPiecewiseTestSuite2;

/*
 * Test Case
 */

LteUeMeasurementsPiecewiseTestCase2::LteUeMeasurementsPiecewiseTestCase2(
    std::string name,
    LteRrcSap::ReportConfigEutra config,
    std::vector<Time> expectedTime,
    std::vector<uint8_t> expectedRsrp)
    : TestCase(name),
      m_config(config),
      m_expectedTime(expectedTime),
      m_expectedRsrp(expectedRsrp)
{
    // input sanity check
    uint16_t size = m_expectedTime.size();

    if (size != m_expectedRsrp.size())
    {
        NS_FATAL_ERROR("Vectors of expected results are not of the same size");
    }

    m_itExpectedTime = m_expectedTime.begin();
    m_itExpectedRsrp = m_expectedRsrp.begin();

    NS_LOG_INFO(this << " name=" << name);
}

LteUeMeasurementsPiecewiseTestCase2::~LteUeMeasurementsPiecewiseTestCase2()
{
    NS_LOG_FUNCTION(this);
}

void
LteUeMeasurementsPiecewiseTestCase2::DoRun()
{
    NS_LOG_INFO(this << " " << GetName());

    Ptr<LteHelper> lteHelper = CreateObject<LteHelper>();
    lteHelper->SetAttribute("PathlossModel", StringValue("ns3::FriisSpectrumPropagationLossModel"));
    lteHelper->SetAttribute("UseIdealRrc", BooleanValue(true));

    // Disable Uplink Power Control
    Config::SetDefault("ns3::LteUePhy::EnableUplinkPowerControl", BooleanValue(false));

    // Create Nodes: eNodeB and UE
    NodeContainer enbNodes;
    NodeContainer ueNodes;
    enbNodes.Create(2);
    ueNodes.Create(1);

    /*
     * The topology is the following:
     *
     * eNodeB    UE                                                eNodeB
     *    |      |                                                    |
     *    x ---- x --------------- x ------- x --------------- x ---- x
     *      50 m |      200 m      |  100 m  |      200 m      | 50 m
     *           |                 |         |                 |
     *        VeryNear            Near      Far             VeryFar
     */

    Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator>();
    positionAlloc->Add(Vector(0.0, 0.0, 0.0));   // Serving eNodeB
    positionAlloc->Add(Vector(600.0, 0.0, 0.0)); // Neighbour eNodeB
    positionAlloc->Add(Vector(50.0, 0.0, 0.0));  // UE
    MobilityHelper mobility;
    mobility.SetMobilityModel("ns3::ConstantPositionMobilityModel");
    mobility.SetPositionAllocator(positionAlloc);
    mobility.Install(enbNodes);
    mobility.Install(ueNodes);
    m_ueMobility = ueNodes.Get(0)->GetObject<MobilityModel>();

    // Disable layer-3 filtering
    Config::SetDefault("ns3::LteEnbRrc::RsrpFilterCoefficient", UintegerValue(0));

    // Create Devices and install them in the Nodes (eNB and UE)
    NetDeviceContainer enbDevs;
    NetDeviceContainer ueDevs;
    lteHelper->SetSchedulerType("ns3::RrFfMacScheduler");
    lteHelper->SetSchedulerAttribute("UlCqiFilter", EnumValue(FfMacScheduler::PUSCH_UL_CQI));
    enbDevs = lteHelper->InstallEnbDevice(enbNodes);
    ueDevs = lteHelper->InstallUeDevice(ueNodes);

    // Setup UE measurement configuration in serving cell
    Ptr<LteEnbRrc> enbRrc1 = enbDevs.Get(0)->GetObject<LteEnbNetDevice>()->GetRrc();
    m_expectedMeasId = enbRrc1->AddUeMeasReportConfig(m_config).at(0);

    // Disable handover in neighbour cell
    Ptr<LteEnbRrc> enbRrc2 = enbDevs.Get(1)->GetObject<LteEnbNetDevice>()->GetRrc();
    enbRrc2->SetAttribute("AdmitHandoverRequest", BooleanValue(false));

    // Attach UE to serving eNodeB
    lteHelper->Attach(ueDevs.Get(0), enbDevs.Get(0));

    // Activate an EPS bearer
    EpsBearer::Qci q = EpsBearer::GBR_CONV_VOICE;
    EpsBearer bearer(q);
    lteHelper->ActivateDataRadioBearer(ueDevs, bearer);

    // Connect to trace sources in serving eNodeB
    Config::Connect(
        "/NodeList/0/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
        MakeCallback(&LteUeMeasurementsPiecewiseTestCase2::RecvMeasurementReportCallback, this));

    /*
     * Schedule "teleports"
     *          0                   1                   2
     *          +-------------------+-------------------+---------> time
     * VeryNear |------  ----    ----                    --------
     *     Near |                    ----            ----
     *      Far |                        ----    ----
     *  VeryFar |      --    ----            ----
     */
    Simulator::Schedule(MilliSeconds(301),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportVeryFar,
                        this);
    Simulator::Schedule(MilliSeconds(401),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportVeryNear,
                        this);
    Simulator::Schedule(MilliSeconds(601),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportVeryFar,
                        this);
    Simulator::Schedule(MilliSeconds(801),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportVeryNear,
                        this);
    Simulator::Schedule(MilliSeconds(1001),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportNear,
                        this);
    Simulator::Schedule(MilliSeconds(1201),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportFar,
                        this);
    Simulator::Schedule(MilliSeconds(1401),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportVeryFar,
                        this);
    Simulator::Schedule(MilliSeconds(1601),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportFar,
                        this);
    Simulator::Schedule(MilliSeconds(1801),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportNear,
                        this);
    Simulator::Schedule(MilliSeconds(2001),
                        &LteUeMeasurementsPiecewiseTestCase2::TeleportVeryNear,
                        this);

    // Run simulation
    Simulator::Stop(Seconds(2.201));
    Simulator::Run();
    Simulator::Destroy();
}

void
LteUeMeasurementsPiecewiseTestCase2::DoTeardown()
{
    NS_LOG_FUNCTION(this);
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          true,
                          "Reporting should have occurred at " << m_itExpectedTime->As(Time::S));
    hasEnded = m_itExpectedRsrp == m_expectedRsrp.end();
    NS_ASSERT(hasEnded);
}

void
LteUeMeasurementsPiecewiseTestCase2::RecvMeasurementReportCallback(
    std::string context,
    uint64_t imsi,
    uint16_t cellId,
    uint16_t rnti,
    LteRrcSap::MeasurementReport report)
{
    NS_LOG_FUNCTION(this << context);
    NS_ASSERT(rnti == 1);
    NS_ASSERT(cellId == 1);

    if (report.measResults.measId != m_expectedMeasId)
    {
        return;
    }

    // verifying the report completeness
    LteRrcSap::MeasResults measResults = report.measResults;
    NS_LOG_DEBUG(
        this << " Serving cellId=" << cellId
             << " rsrp=" << (uint16_t)measResults.measResultPCell.rsrpResult << " ("
             << EutranMeasurementMapping::RsrpRange2Dbm(measResults.measResultPCell.rsrpResult)
             << " dBm)"
             << " rsrq=" << (uint16_t)measResults.measResultPCell.rsrqResult << " ("
             << EutranMeasurementMapping::RsrqRange2Db(measResults.measResultPCell.rsrqResult)
             << " dB)");

    // verifying reported best cells
    if (measResults.measResultListEutra.empty())
    {
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              false,
                              "Unexpected report content");
    }
    else
    {
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              true,
                              "Unexpected report content");
        auto it = measResults.measResultListEutra.begin();
        NS_ASSERT(it != measResults.measResultListEutra.end());
        NS_ASSERT(it->physCellId == 2);
        NS_TEST_ASSERT_MSG_EQ(it->haveCgiInfo,
                              false,
                              "Report contains cgi-info, which is not supported");
        NS_TEST_ASSERT_MSG_EQ(it->haveRsrpResult,
                              true,
                              "Report does not contain measured RSRP result");
        NS_TEST_ASSERT_MSG_EQ(it->haveRsrqResult,
                              true,
                              "Report does not contain measured RSRQ result");
        NS_LOG_DEBUG(this << " Neighbour cellId=" << it->physCellId
                          << " rsrp=" << (uint16_t)it->rsrpResult << " ("
                          << EutranMeasurementMapping::RsrpRange2Dbm(it->rsrpResult) << " dBm)"
                          << " rsrq=" << (uint16_t)it->rsrqResult << " ("
                          << EutranMeasurementMapping::RsrqRange2Db(it->rsrqResult) << " dB)");
    }

    // verifying the report timing
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          false,
                          "Reporting should not have occurred at " << Simulator::Now().As(Time::S));
    if (hasEnded)
    {
        return;
    }

    hasEnded = m_itExpectedRsrp == m_expectedRsrp.end();
    NS_ASSERT(!hasEnded);

    // using milliseconds to avoid floating-point comparison
    uint64_t timeNowMs = Simulator::Now().GetMilliSeconds();
    uint64_t timeExpectedMs = m_itExpectedTime->GetMilliSeconds();
    m_itExpectedTime++;

    uint16_t observedRsrp = measResults.measResultPCell.rsrpResult;
    uint16_t referenceRsrp = *m_itExpectedRsrp;
    m_itExpectedRsrp++;

    NS_TEST_ASSERT_MSG_EQ(timeNowMs,
                          timeExpectedMs,
                          "Reporting should not have occurred at this time");
    NS_TEST_ASSERT_MSG_EQ(observedRsrp,
                          referenceRsrp,
                          "The RSRP observed differs with the reference RSRP");
}

void
LteUeMeasurementsPiecewiseTestCase2::TeleportVeryNear()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(50.0, 0.0, 0.0));
}

void
LteUeMeasurementsPiecewiseTestCase2::TeleportNear()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(250.0, 0.0, 0.0));
}

void
LteUeMeasurementsPiecewiseTestCase2::TeleportFar()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(350.0, 0.0, 0.0));
}

void
LteUeMeasurementsPiecewiseTestCase2::TeleportVeryFar()
{
    NS_LOG_FUNCTION(this);
    m_ueMobility->SetPosition(Vector(550.0, 0.0, 0.0));
}

// ===== LTE-UE-MEASUREMENTS-PIECEWISE-3 TEST SUITE ======================== //

/*
 * Test Suite
 */

LteUeMeasurementsPiecewiseTestSuite3::LteUeMeasurementsPiecewiseTestSuite3()
    : TestSuite("lte-ue-measurements-piecewise-3", Type::SYSTEM)
{
    std::vector<Time> expectedTime;

    // === Event A4 (neighbor becomes better than threshold) ===

    // The threshold value was chosen to achieve the following:
    // 1. Neighbor 1 (eNB2) RSRP would be above the chosen threshold, hence,
    // the UE will include it in its reports to its eNB (eNB1) from the beginning
    // of the simulation.
    // 2. When neighbor 2 (eNB3) is placed at a very far position, its RSRP would
    // be less than the chosen threshold, hence, UE will not include it in its
    // initial report(s) to its eNB.
    // 3. When neighbor 2 (eNB3) is placed at a near position, its RSRP would
    // always be above the chosen threshold, hence, the UE will include it in its
    // reports to its eNB (eNB1).
    LteRrcSap::ReportConfigEutra config;
    config.triggerType = LteRrcSap::ReportConfigEutra::EVENT;
    config.eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    config.threshold1.choice = LteRrcSap::ThresholdEutra::THRESHOLD_RSRP;
    config.threshold1.range = 6;
    config.triggerQuantity = LteRrcSap::ReportConfigEutra::RSRP;
    config.reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920 << 1160 << 1400 << 1640 << 1880 << 2120;

    AddTestCase(new LteUeMeasurementsPiecewiseTestCase3("Piecewise test case 3 - Event A4",
                                                        config,
                                                        expectedTime),
                TestCase::Duration::QUICK);
} // end of LteUeMeasurementsPiecewiseTestSuite3::LteUeMeasurementsPiecewiseTestSuite3

/**
 * @ingroup lte-test
 * Static variable for test initialization
 */
static LteUeMeasurementsPiecewiseTestSuite3 lteUeMeasurementsPiecewiseTestSuite3;

/*
 * Test Case
 */

LteUeMeasurementsPiecewiseTestCase3::LteUeMeasurementsPiecewiseTestCase3(
    std::string name,
    LteRrcSap::ReportConfigEutra config,
    std::vector<Time> expectedTime)
    : TestCase(name),
      m_config(config),
      m_expectedTime(expectedTime)
{
    m_expectedMeasId = std::numeric_limits<uint8_t>::max();

    m_itExpectedTime = m_expectedTime.begin();

    NS_LOG_INFO(this << " name=" << name);
}

LteUeMeasurementsPiecewiseTestCase3::~LteUeMeasurementsPiecewiseTestCase3()
{
    NS_LOG_FUNCTION(this);
}

void
LteUeMeasurementsPiecewiseTestCase3::DoRun()
{
    NS_LOG_INFO(this << " " << GetName());

    Ptr<LteHelper> lteHelper = CreateObject<LteHelper>();
    lteHelper->SetAttribute("PathlossModel", StringValue("ns3::FriisSpectrumPropagationLossModel"));
    lteHelper->SetAttribute("UseIdealRrc", BooleanValue(true));

    // Disable Uplink Power Control
    Config::SetDefault("ns3::LteUePhy::EnableUplinkPowerControl", BooleanValue(false));

    // Create Nodes: eNodeB and UE
    NodeContainer enbNodes;
    NodeContainer ueNodes;
    enbNodes.Create(3);
    ueNodes.Create(1);

    /*
     * The topology is the following:
     *
     * We place the 3rd eNB initially very far so it does not fulfills
     * the entry condition to be reported.
     *
     * eNodeB    UE              eNodeB                                  eNodeB
     *    |      |                 |                                       |
     *    x ---- x --------------- x -------------- x ---------------------x
     *      50 m         100 m             500      |         1000000
     *                                             Near
     */

    Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator>();
    positionAlloc->Add(Vector(0.0, 0.0, 0.0));       // Serving eNodeB
    positionAlloc->Add(Vector(200.0, 0.0, 0.0));     // Neighbour eNodeB1
    positionAlloc->Add(Vector(1000700.0, 0.0, 0.0)); // Neighbour eNodeB2
    positionAlloc->Add(Vector(50.0, 0.0, 0.0));      // UE
    MobilityHelper mobility;
    mobility.SetMobilityModel("ns3::ConstantPositionMobilityModel");
    mobility.SetPositionAllocator(positionAlloc);
    mobility.Install(enbNodes);
    mobility.Install(ueNodes);
    m_enbMobility = enbNodes.Get(2)->GetObject<MobilityModel>();

    // Disable layer-3 filtering
    Config::SetDefault("ns3::LteEnbRrc::RsrpFilterCoefficient", UintegerValue(0));

    // Create Devices and install them in the Nodes (eNB and UE)
    NetDeviceContainer enbDevs;
    NetDeviceContainer ueDevs;
    lteHelper->SetSchedulerType("ns3::RrFfMacScheduler");
    lteHelper->SetSchedulerAttribute("UlCqiFilter", EnumValue(FfMacScheduler::PUSCH_UL_CQI));
    enbDevs = lteHelper->InstallEnbDevice(enbNodes);
    ueDevs = lteHelper->InstallUeDevice(ueNodes);

    // Setup UE measurement configuration in serving cell
    Ptr<LteEnbRrc> enbRrc1 = enbDevs.Get(0)->GetObject<LteEnbNetDevice>()->GetRrc();
    m_expectedMeasId = enbRrc1->AddUeMeasReportConfig(m_config).at(0);

    // Disable handover in neighbour cells
    Ptr<LteEnbRrc> enbRrc2 = enbDevs.Get(1)->GetObject<LteEnbNetDevice>()->GetRrc();
    enbRrc2->SetAttribute("AdmitHandoverRequest", BooleanValue(false));
    Ptr<LteEnbRrc> enbRrc3 = enbDevs.Get(2)->GetObject<LteEnbNetDevice>()->GetRrc();
    enbRrc3->SetAttribute("AdmitHandoverRequest", BooleanValue(false));

    // Attach UE to serving eNodeB
    lteHelper->Attach(ueDevs.Get(0), enbDevs.Get(0));

    // Activate an EPS bearer
    EpsBearer::Qci q = EpsBearer::GBR_CONV_VOICE;
    EpsBearer bearer(q);
    lteHelper->ActivateDataRadioBearer(ueDevs, bearer);

    // Connect to trace sources in serving eNodeB
    Config::Connect(
        "/NodeList/0/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
        MakeCallback(&LteUeMeasurementsPiecewiseTestCase3::RecvMeasurementReportCallback, this));
    /*
     * Schedule "teleport" for the 2nd neighbour
     *
     * We bring the 2nd neighbour near once the UE has already scheduled the periodic
     * reporting after detecting the 1st neighbour, which ideally should be at
     * 200 ms.
     */
    Simulator::Schedule(MilliSeconds(301),
                        &LteUeMeasurementsPiecewiseTestCase3::TeleportEnbNear,
                        this);

    // Run simulation
    Simulator::Stop(Seconds(2.201));
    Simulator::Run();
    Simulator::Destroy();
}

void
LteUeMeasurementsPiecewiseTestCase3::DoTeardown()
{
    NS_LOG_FUNCTION(this);
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          true,
                          "Reporting should have occurred at " << m_itExpectedTime->GetSeconds()
                                                               << "s");
}

void
LteUeMeasurementsPiecewiseTestCase3::RecvMeasurementReportCallback(
    std::string context,
    uint64_t imsi,
    uint16_t cellId,
    uint16_t rnti,
    LteRrcSap::MeasurementReport report)
{
    NS_LOG_FUNCTION(this << context);
    NS_ASSERT(rnti == 1);
    NS_ASSERT(cellId == 1);

    if (report.measResults.measId != m_expectedMeasId)
    {
        return;
    }

    // verifying the report completeness
    LteRrcSap::MeasResults measResults = report.measResults;
    NS_LOG_DEBUG(
        this << " Serving cellId=" << cellId
             << " rsrp=" << (uint16_t)measResults.measResultPCell.rsrpResult << " ("
             << EutranMeasurementMapping::RsrpRange2Dbm(measResults.measResultPCell.rsrpResult)
             << " dBm)"
             << " rsrq=" << (uint16_t)measResults.measResultPCell.rsrqResult << " ("
             << EutranMeasurementMapping::RsrqRange2Db(measResults.measResultPCell.rsrqResult)
             << " dB)");

    // verifying reported best cells
    if (measResults.measResultListEutra.empty())
    {
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              false,
                              "Unexpected report content");
    }
    else
    {
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              true,
                              "Unexpected report content");
        auto it = measResults.measResultListEutra.begin();
        NS_ASSERT(it != measResults.measResultListEutra.end());
        for (const auto& it : measResults.measResultListEutra)
        {
            NS_ASSERT(it.physCellId == 2 || it.physCellId == 3);
            NS_TEST_ASSERT_MSG_EQ(it.haveCgiInfo,
                                  false,
                                  "Report contains cgi-info, which is not supported");
            NS_TEST_ASSERT_MSG_EQ(it.haveRsrpResult,
                                  true,
                                  "Report does not contain measured RSRP result");
            NS_TEST_ASSERT_MSG_EQ(it.haveRsrqResult,
                                  true,
                                  "Report does not contain measured RSRQ result");
            NS_LOG_DEBUG(this << " Neighbour cellId=" << it.physCellId
                              << " rsrp=" << (uint16_t)it.rsrpResult << " ("
                              << EutranMeasurementMapping::RsrpRange2Dbm(it.rsrpResult) << " dBm)"
                              << " rsrq=" << (uint16_t)it.rsrqResult << " ("
                              << EutranMeasurementMapping::RsrqRange2Db(it.rsrqResult) << " dB)");
        }
    }

    // verifying the report timing
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          false,
                          "Reporting should not have occurred at " << Simulator::Now().GetSeconds()
                                                                   << "s");
    if (hasEnded)
    {
        return;
    }

    // using milliseconds to avoid floating-point comparison
    uint64_t timeNowMs = Simulator::Now().GetMilliSeconds();
    uint64_t timeExpectedMs = m_itExpectedTime->GetMilliSeconds();
    m_itExpectedTime++;

    NS_TEST_ASSERT_MSG_EQ(timeNowMs,
                          timeExpectedMs,
                          "Reporting should not have occurred at this time");
}

void
LteUeMeasurementsPiecewiseTestCase3::TeleportEnbNear()
{
    NS_LOG_FUNCTION(this);
    m_enbMobility->SetPosition(Vector(700.0, 0.0, 0.0));
}

// ===== LTE-UE-MEASUREMENTS-HANDOVER TEST SUITE =========================== //

/*
 * Test Suite
 */

LteUeMeasurementsHandoverTestSuite::LteUeMeasurementsHandoverTestSuite()
    : TestSuite("lte-ue-measurements-handover", Type::SYSTEM)
{
    std::list<LteRrcSap::ReportConfigEutra> sourceConfigList;
    std::list<LteRrcSap::ReportConfigEutra> targetConfigList;
    std::vector<Time> expectedTime;
    std::vector<uint8_t> expectedRsrp;

    LteRrcSap::ReportConfigEutra sourceConfig;
    sourceConfig.triggerType = LteRrcSap::ReportConfigEutra::EVENT;
    sourceConfig.eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    sourceConfig.threshold1.choice = LteRrcSap::ThresholdEutra::THRESHOLD_RSRP;
    sourceConfig.threshold1.range = 0;
    sourceConfig.triggerQuantity = LteRrcSap::ReportConfigEutra::RSRP;
    sourceConfig.reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    sourceConfigList.push_back(sourceConfig);

    LteRrcSap::ReportConfigEutra targetConfig;
    targetConfig.triggerType = LteRrcSap::ReportConfigEutra::EVENT;
    targetConfig.eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    targetConfig.threshold1.choice = LteRrcSap::ThresholdEutra::THRESHOLD_RSRP;
    targetConfig.threshold1.range = 0;
    targetConfig.triggerQuantity = LteRrcSap::ReportConfigEutra::RSRP;
    targetConfig.reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    targetConfigList.push_back(targetConfig);

    // === Report interval difference ===

    // decreasing report interval
    sourceConfigList.front().reportInterval = LteRrcSap::ReportConfigEutra::MS480;
    targetConfigList.front().reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    expectedTime.clear();
    expectedTime << 200 << 680 << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 53 << 53 << 53 << 53;
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - decreasing report interval",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::TAKES_FOREVER);

    // increasing report interval
    sourceConfigList.front().reportInterval = LteRrcSap::ReportConfigEutra::MS120;
    targetConfigList.front().reportInterval = LteRrcSap::ReportConfigEutra::MS640;
    expectedTime.clear();
    expectedTime << 200 << 320 << 440 << 560 << 680 << 800 << 920 << 1200 << 1840;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 55 << 55 << 55 << 55 << 55 << 53 << 53;
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - increasing report interval",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::QUICK);

    // === Event difference ===

    sourceConfigList.front().reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    targetConfigList.front().reportInterval = LteRrcSap::ReportConfigEutra::MS240;
    sourceConfigList.front().threshold1.range = 54;
    sourceConfigList.front().threshold2.range = 54;
    sourceConfigList.front().a3Offset = 1;
    targetConfigList.front().threshold1.range = 54;
    targetConfigList.front().threshold2.range = 54;
    targetConfigList.front().a3Offset = 1;

    // Event A1 to Event A2
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920 << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 55 << 55 << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A1 to Event A2",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::EXTENSIVE);

    // Event A2 to Event A1
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A2 to Event A1",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::TAKES_FOREVER);

    // Event A3 to Event A4
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A3 to Event A4",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::TAKES_FOREVER);

    // Event A4 to Event A3
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A4 to Event A3",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::QUICK);

    // Event A2 to Event A3
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A2 to Event A3",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::EXTENSIVE);

    // Event A3 to Event A2
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A3 to Event A2",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::TAKES_FOREVER);

    // Event A4 to Event A5
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A5;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A4 to Event A5",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::TAKES_FOREVER);

    // Event A5 to Event A4
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A5;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - Event A5 to Event A4",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::EXTENSIVE);

    // === Threshold/offset difference ===

    sourceConfigList.front().threshold1.range = 52;
    targetConfigList.front().threshold1.range = 56;

    // Event A1
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 55 << 55;
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - Event A1 threshold difference",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::EXTENSIVE);

    // Event A2
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A2;
    expectedTime.clear();
    expectedTime << 1200 << 1440 << 1680 << 1920;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53 << 53;
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - Event A2 threshold difference",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::QUICK);

    // Event A3
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    sourceConfigList.front().a3Offset = -30;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A3;
    targetConfigList.front().a3Offset = 30;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 55 << 55;
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - Event A3 offset difference",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::QUICK);

    // Event A4
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A4;
    expectedTime.clear();
    expectedTime << 200 << 440 << 680 << 920;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 55 << 55;
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - Event A4 threshold difference",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::EXTENSIVE);

    // Event A5
    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A5;
    sourceConfigList.front().threshold2.range = 52;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A5;
    targetConfigList.front().threshold2.range = 56;
    expectedTime.clear();
    expectedRsrp.clear();
    AddTestCase(
        new LteUeMeasurementsHandoverTestCase("Handover test case - Event A5 threshold difference",
                                              sourceConfigList,
                                              targetConfigList,
                                              expectedTime,
                                              expectedRsrp,
                                              Seconds(2)),
        TestCase::Duration::EXTENSIVE);

    // === Time-to-trigger (TTT) difference ===

    sourceConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    sourceConfigList.front().a3Offset = 1;
    sourceConfigList.front().threshold1.range = 0;
    sourceConfigList.front().threshold2.range = 0;
    targetConfigList.front().eventId = LteRrcSap::ReportConfigEutra::EVENT_A1;
    targetConfigList.front().a3Offset = 1;
    targetConfigList.front().threshold1.range = 0;
    targetConfigList.front().threshold2.range = 0;

    // decreasing time-to-trigger (short duration)
    sourceConfigList.front().timeToTrigger = 1024;
    targetConfigList.front().timeToTrigger = 100;
    expectedTime.clear();
    expectedTime << 1300 << 1540 << 1780;
    expectedRsrp.clear();
    expectedRsrp << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - decreasing TTT (short)",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(2)),
                TestCase::Duration::QUICK);

    // decreasing time-to-trigger (longer duration)
    sourceConfigList.front().timeToTrigger = 1024;
    targetConfigList.front().timeToTrigger = 640;
    expectedTime.clear();
    expectedTime << 1224 << 1464 << 1704 << 1944 << 2840 << 3080 << 3320 << 3560 << 3800 << 4040;
    expectedRsrp.clear();
    expectedRsrp << 55 << 55 << 55 << 55 << 53 << 53 << 53 << 53 << 53 << 53;
    AddTestCase(new LteUeMeasurementsHandoverTestCase("Handover test case - decreasing TTT (long)",
                                                      sourceConfigList,
                                                      targetConfigList,
                                                      expectedTime,
                                                      expectedRsrp,
                                                      Seconds(4.2)),
                TestCase::Duration::EXTENSIVE);

} // end of LteUeMeasurementsHandoverTestSuite::LteUeMeasurementsHandoverTestSuite

/**
 * @ingroup lte-test
 * Static variable for test initialization
 */
static LteUeMeasurementsHandoverTestSuite lteUeMeasurementsHandoverTestSuite;

/*
 * Test Case
 */

LteUeMeasurementsHandoverTestCase::LteUeMeasurementsHandoverTestCase(
    std::string name,
    std::list<LteRrcSap::ReportConfigEutra> sourceConfigList,
    std::list<LteRrcSap::ReportConfigEutra> targetConfigList,
    std::vector<Time> expectedTime,
    std::vector<uint8_t> expectedRsrp,
    Time duration)
    : TestCase(name),
      m_sourceConfigList(sourceConfigList),
      m_targetConfigList(targetConfigList),
      m_expectedTime(expectedTime),
      m_expectedRsrp(expectedRsrp),
      m_duration(duration)
{
    // input sanity check
    uint16_t size = m_expectedTime.size();

    if (size != m_expectedRsrp.size())
    {
        NS_FATAL_ERROR("Vectors of expected results are not of the same size");
    }

    m_itExpectedTime = m_expectedTime.begin();
    m_itExpectedRsrp = m_expectedRsrp.begin();

    NS_LOG_INFO(this << " name=" << name);
}

LteUeMeasurementsHandoverTestCase::~LteUeMeasurementsHandoverTestCase()
{
    NS_LOG_FUNCTION(this);
}

void
LteUeMeasurementsHandoverTestCase::DoRun()
{
    NS_LOG_INFO(this << " " << GetName());

    Ptr<LteHelper> lteHelper = CreateObject<LteHelper>();
    Ptr<PointToPointEpcHelper> epcHelper = CreateObject<PointToPointEpcHelper>();
    lteHelper->SetEpcHelper(epcHelper);
    lteHelper->SetAttribute("PathlossModel", StringValue("ns3::FriisSpectrumPropagationLossModel"));
    lteHelper->SetAttribute("UseIdealRrc", BooleanValue(true));

    // Disable Uplink Power Control
    Config::SetDefault("ns3::LteUePhy::EnableUplinkPowerControl", BooleanValue(false));

    // Create Nodes: eNodeB and UE
    NodeContainer enbNodes;
    NodeContainer ueNodes;
    enbNodes.Create(2);
    ueNodes.Create(1);

    /*
     * The topology is the following:
     *
     * eNodeB                   UE                     eNodeB
     *    |                     |                         |
     *    x ------------------- x ----------------------- x
     *             400 m                   500 m
     */

    Ptr<ListPositionAllocator> positionAlloc = CreateObject<ListPositionAllocator>();
    positionAlloc->Add(Vector(0.0, 0.0, 0.0));   // Source eNodeB
    positionAlloc->Add(Vector(900.0, 0.0, 0.0)); // Target eNodeB
    positionAlloc->Add(Vector(400.0, 0.0, 0.0)); // UE
    MobilityHelper mobility;
    mobility.SetMobilityModel("ns3::ConstantPositionMobilityModel");
    mobility.SetPositionAllocator(positionAlloc);
    mobility.Install(enbNodes);
    mobility.Install(ueNodes);

    // Create P-GW node
    Ptr<Node> pgw = epcHelper->GetPgwNode();

    // Create a single RemoteHost
    NodeContainer remoteHostContainer;
    remoteHostContainer.Create(1);
    Ptr<Node> remoteHost = remoteHostContainer.Get(0);
    InternetStackHelper internet;
    internet.Install(remoteHostContainer);

    // Create the Internet
    PointToPointHelper p2ph;
    p2ph.SetDeviceAttribute("DataRate", DataRateValue(DataRate("100Gb/s")));
    p2ph.SetDeviceAttribute("Mtu", UintegerValue(1500));
    p2ph.SetChannelAttribute("Delay", TimeValue(Seconds(0.010)));
    NetDeviceContainer internetDevices = p2ph.Install(pgw, remoteHost);
    Ipv4AddressHelper ipv4h;
    ipv4h.SetBase("1.0.0.0", "255.0.0.0");
    Ipv4InterfaceContainer internetIpIfaces = ipv4h.Assign(internetDevices);

    // Routing of the Internet Host (towards the LTE network)
    Ipv4StaticRoutingHelper ipv4RoutingHelper;
    Ptr<Ipv4StaticRouting> remoteHostStaticRouting =
        ipv4RoutingHelper.GetStaticRouting(remoteHost->GetObject<Ipv4>());
    remoteHostStaticRouting->AddNetworkRouteTo(Ipv4Address("7.0.0.0"), Ipv4Mask("255.0.0.0"), 1);

    // Enable layer-3 filtering
    Config::SetDefault("ns3::LteEnbRrc::RsrpFilterCoefficient", UintegerValue(4));

    // Disable control channel error model
    Config::SetDefault("ns3::LteSpectrumPhy::CtrlErrorModelEnabled", BooleanValue(false));

    // Create Devices and install them in the Nodes (eNB and UE)
    NetDeviceContainer enbDevs;
    NetDeviceContainer ueDevs;
    enbDevs = lteHelper->InstallEnbDevice(enbNodes);
    ueDevs = lteHelper->InstallUeDevice(ueNodes);

    // Setup UE measurement configuration in eNodeBs
    uint8_t measId;
    Ptr<LteEnbRrc> enbRrc1 = enbDevs.Get(0)->GetObject<LteEnbNetDevice>()->GetRrc();
    Ptr<LteEnbRrc> enbRrc2 = enbDevs.Get(1)->GetObject<LteEnbNetDevice>()->GetRrc();

    for (auto itReportConfig = m_sourceConfigList.begin();
         itReportConfig != m_sourceConfigList.end();
         itReportConfig++)
    {
        measId = enbRrc1->AddUeMeasReportConfig(*itReportConfig).at(0);
        m_expectedSourceCellMeasId.insert(measId);
    }

    for (auto itReportConfig = m_targetConfigList.begin();
         itReportConfig != m_targetConfigList.end();
         itReportConfig++)
    {
        measId = enbRrc2->AddUeMeasReportConfig(*itReportConfig).at(0);
        m_expectedTargetCellMeasId.insert(measId);
    }

    // Install the IP stack on the UEs
    internet.Install(ueNodes);
    Ipv4InterfaceContainer ueIpIfaces;
    ueIpIfaces = epcHelper->AssignUeIpv4Address(NetDeviceContainer(ueDevs));

    // Assign IP address to UEs
    for (uint32_t u = 0; u < ueNodes.GetN(); ++u)
    {
        Ptr<Node> ueNode = ueNodes.Get(u);
        // Set the default gateway for the UE
        Ptr<Ipv4StaticRouting> ueStaticRouting =
            ipv4RoutingHelper.GetStaticRouting(ueNode->GetObject<Ipv4>());
        ueStaticRouting->SetDefaultRoute(epcHelper->GetUeDefaultGatewayAddress(), 1);
    }

    // Attach UE to serving eNodeB
    lteHelper->Attach(ueDevs.Get(0), enbDevs.Get(0));

    // Add X2 interface
    lteHelper->AddX2Interface(enbNodes);

    // Connect to trace sources in source eNodeB
    Config::Connect(
        "/NodeList/3/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
        MakeCallback(&LteUeMeasurementsHandoverTestCase::RecvMeasurementReportCallback, this));

    // Connect to trace sources in target eNodeB
    Config::Connect(
        "/NodeList/4/DeviceList/0/LteEnbRrc/RecvMeasurementReport",
        MakeCallback(&LteUeMeasurementsHandoverTestCase::RecvMeasurementReportCallback, this));

    // Schedule handover
    lteHelper->HandoverRequest(MilliSeconds(m_duration.GetMilliSeconds() / 2),
                               ueDevs.Get(0),
                               enbDevs.Get(0),
                               enbDevs.Get(1));

    // Run simulation
    Simulator::Stop(m_duration);
    Simulator::Run();
    Simulator::Destroy();
}

void
LteUeMeasurementsHandoverTestCase::DoTeardown()
{
    NS_LOG_FUNCTION(this);
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          true,
                          "Reporting should have occurred at " << m_itExpectedTime->As(Time::S));
    hasEnded = m_itExpectedRsrp == m_expectedRsrp.end();
    NS_ASSERT(hasEnded);
}

void
LteUeMeasurementsHandoverTestCase::RecvMeasurementReportCallback(
    std::string context,
    uint64_t imsi,
    uint16_t cellId,
    uint16_t rnti,
    LteRrcSap::MeasurementReport report)
{
    uint8_t measId = report.measResults.measId;
    NS_LOG_FUNCTION(this << context << (uint16_t)measId);

    bool isCorrectMeasId;
    if (cellId == 1)
    {
        auto itMeasId = m_expectedSourceCellMeasId.find(measId);
        isCorrectMeasId = (itMeasId != m_expectedSourceCellMeasId.end());
    }
    else if (cellId == 2)
    {
        auto itMeasId = m_expectedTargetCellMeasId.find(measId);
        isCorrectMeasId = (itMeasId != m_expectedTargetCellMeasId.end());
    }
    else
    {
        NS_FATAL_ERROR("Invalid cell ID " << cellId);
    }

    if (!isCorrectMeasId)
    {
        return;
    }

    // verifying the report completeness
    LteRrcSap::MeasResults measResults = report.measResults;
    NS_LOG_DEBUG(
        this << " Serving cellId=" << cellId
             << " rsrp=" << (uint16_t)measResults.measResultPCell.rsrpResult << " ("
             << EutranMeasurementMapping::RsrpRange2Dbm(measResults.measResultPCell.rsrpResult)
             << " dBm)"
             << " rsrq=" << (uint16_t)measResults.measResultPCell.rsrqResult << " ("
             << EutranMeasurementMapping::RsrqRange2Db(measResults.measResultPCell.rsrqResult)
             << " dB)");

    // verifying reported best cells
    if (measResults.measResultListEutra.empty())
    {
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              false,
                              "Unexpected report content");
    }
    else
    {
        NS_TEST_ASSERT_MSG_EQ(measResults.haveMeasResultNeighCells,
                              true,
                              "Unexpected report content");
        auto it = measResults.measResultListEutra.begin();
        NS_ASSERT(it != measResults.measResultListEutra.end());
        NS_ASSERT(it->physCellId != cellId);
        NS_ASSERT(it->physCellId <= 2);
        NS_TEST_ASSERT_MSG_EQ(it->haveCgiInfo,
                              false,
                              "Report contains cgi-info, which is not supported");
        NS_TEST_ASSERT_MSG_EQ(it->haveRsrpResult,
                              true,
                              "Report does not contain measured RSRP result");
        NS_TEST_ASSERT_MSG_EQ(it->haveRsrqResult,
                              true,
                              "Report does not contain measured RSRQ result");
        NS_LOG_DEBUG(this << " Neighbour cellId=" << it->physCellId
                          << " rsrp=" << (uint16_t)it->rsrpResult << " ("
                          << EutranMeasurementMapping::RsrpRange2Dbm(it->rsrpResult) << " dBm)"
                          << " rsrq=" << (uint16_t)it->rsrqResult << " ("
                          << EutranMeasurementMapping::RsrqRange2Db(it->rsrqResult) << " dB)");
    }

    // verifying the report timing
    bool hasEnded = m_itExpectedTime == m_expectedTime.end();
    NS_TEST_ASSERT_MSG_EQ(hasEnded,
                          false,
                          "Reporting should not have occurred at " << Simulator::Now().As(Time::S));
    if (hasEnded)
    {
        return;
    }

    hasEnded = m_itExpectedRsrp == m_expectedRsrp.end();
    NS_ASSERT(!hasEnded);

    // using milliseconds to avoid floating-point comparison
    uint64_t timeNowMs = Simulator::Now().GetMilliSeconds();
    uint64_t timeExpectedMs = m_itExpectedTime->GetMilliSeconds();
    m_itExpectedTime++;

    uint16_t observedRsrp = measResults.measResultPCell.rsrpResult;
    uint16_t referenceRsrp = *m_itExpectedRsrp;
    m_itExpectedRsrp++;

    NS_TEST_ASSERT_MSG_EQ(timeNowMs,
                          timeExpectedMs,
                          "Reporting should not have occurred at this time");
    NS_TEST_ASSERT_MSG_EQ(observedRsrp,
                          referenceRsrp,
                          "The RSRP observed differs with the reference RSRP");
}
