cachemultimap_tests.cpp

// Copyright (c) 2014-2019 The Dash Core developers
// Copyright (c) 2020-2023 The Raptoreum developers

#include <cachemultimap.h>

#include <test/test_raptoreum.h>

#include <algorithm>
#include <iostream>

#include <boost/test/unit_test.hpp>

BOOST_FIXTURE_TEST_SUITE(cachemultimap_tests, BasicTestingSetup
)

bool Compare(const CacheMultiMap<int, int> &cmmap1, const CacheMultiMap<int, int> &cmmap2) {
    if (cmmap1.GetMaxSize() != cmmap2.GetMaxSize()) {
        std::cout << "Compare returning false: max size mismatch" << std::endl;
        return false;
    }

    if (cmmap1.GetSize() != cmmap2.GetSize()) {
        std::cout << "Compare returning false: size mismatch" << std::endl;
        return false;
    }

    const CacheMultiMap<int, int>::list_t &items1 = cmmap1.GetItemList();
    const CacheMultiMap<int, int>::list_t &items2 = cmmap2.GetItemList();
    CacheMultiMap<int, int>::list_cit it2 = items2.begin();
    for (CacheMultiMap<int, int>::list_cit it1 = items1.begin(); it1 != items1.end(); ++it1) {
        const CacheItem<int, int> &item1 = *it1;
        const CacheItem<int, int> &item2 = *it2;
        if (item1.key != item2.key) {
            return false;
        }
        if (item1.value != item2.value) {
            return false;
        }
        ++it2;
    }

    return true;
}

bool CheckExpected(const CacheMultiMap<int, int> &cmmap, int *expected, CacheMultiMap<int, int>::size_type nSize) {
    if (cmmap.GetSize() != nSize) {
        return false;
    }
    for (CacheMultiMap<int, int>::size_type i = 0; i < nSize; ++i) {
        int nVal = 0;
        int eVal = expected[i];
        if (!cmmap.Get(eVal, nVal)) {
            return false;
        }
        if (nVal != eVal) {
            return false;
        }
    }
    return true;
}

BOOST_AUTO_TEST_CASE(cachemultimap_test)
        {
                // create a CacheMultiMap limited to 10 items
                CacheMultiMap < int, int > cmmapTest1(10);

        // check that the max size is 10
        BOOST_CHECK(cmmapTest1.GetMaxSize() == 10);

        // check that the size is 0
        BOOST_CHECK(cmmapTest1.GetSize() == 0);

        // insert (-1, -1)
        cmmapTest1.Insert(-1, -1);

        // make sure that the size is updated
        BOOST_CHECK(cmmapTest1.GetSize() == 1);

        // make sure the map contains the key
        BOOST_CHECK(cmmapTest1.HasKey(-1) == true);

        // add 10 items
        for (int i = 0; i < 10; ++i) {
            cmmapTest1.Insert(i, i);
        }

        // check that the size is 10
        BOOST_CHECK(cmmapTest1.GetSize() == 10);

        // check that the map contains the expected items
        for (int i = 0; i < 10; ++i) {
            int nVal = 0;
            BOOST_CHECK(cmmapTest1.Get(i, nVal) == true);
            BOOST_CHECK(nVal == i);
        }

        // check that the map no longer contains the first item
        BOOST_CHECK(cmmapTest1.HasKey(-1) == false);

        // erase an item
        cmmapTest1.Erase(5);

        // check the size
        BOOST_CHECK(cmmapTest1.GetSize() == 9);

        // check that the map no longer contains the item
        BOOST_CHECK(cmmapTest1.HasKey(5) == false);

        // check that the map contains the expected items
        int expected[] = { 0, 1, 2, 3, 4, 6, 7, 8, 9 };
        BOOST_CHECK(CheckExpected(cmmapTest1, expected, 9 ) == true);

        // add multiple items for the same key
        cmmapTest1.Insert(5, 2);
        cmmapTest1.Insert(5, 1);
        cmmapTest1.Insert(5, 4);

        // check the size
        BOOST_CHECK(cmmapTest1.GetSize() == 10);

        // check that 2 keys have been removed
        BOOST_CHECK(cmmapTest1.HasKey(0) == false);
        BOOST_CHECK(cmmapTest1.HasKey(1) == false);
        BOOST_CHECK(cmmapTest1.HasKey(2) == true);

        // check multiple values
        std::vector<int> vecVals;
        BOOST_CHECK(cmmapTest1.GetAll(5, vecVals) == true);
        BOOST_CHECK(vecVals.size() == 3);
        BOOST_CHECK(vecVals[0] == 1);
        BOOST_CHECK(vecVals[1] == 2);
        BOOST_CHECK(vecVals[2] == 4);

//    std::cout << "cmmapTest1 dump:" << std::endl;
//    DumpMap(cmmapTest1);

        // test serialization
        CDataStream ss(SER_NETWORK, PROTOCOL_VERSION);
        ss << cmmapTest1;

        CacheMultiMap<int, int> cmmapTest2;
        ss >> cmmapTest2;

//    std::cout << "cmmapTest2 dump:" << std::endl;
//    DumpMap(cmmapTest2);

        // check multiple values
        std::vector<int> vecVals2;
        BOOST_CHECK(cmmapTest2.GetAll(5, vecVals2) == true);
        BOOST_CHECK(vecVals2.size() == 3);
        BOOST_CHECK(vecVals2[0] == 1);
        BOOST_CHECK(vecVals2[1] == 2);
        BOOST_CHECK(vecVals2[2] == 4);

        BOOST_CHECK(Compare(cmmapTest1, cmmapTest2));

        // test copy constructor
        CacheMultiMap<int, int> cmmapTest3(cmmapTest1);
        BOOST_CHECK(Compare(cmmapTest1, cmmapTest3));

        // test assignment operator
        CacheMultiMap<int, int> mapTest4;
        mapTest4 = cmmapTest1;
        BOOST_CHECK(Compare(cmmapTest1, mapTest4));
        }

BOOST_AUTO_TEST_SUITE_END()