Method TestInsertOrAssign

util/generic/hash_ut.cpp:1087–1136

}

void THashTest::TestInsertOrAssign() {
    static int constructorCounter = 0;
    static int assignmentCounter = 0;

    struct TCountConstruct {
        explicit TCountConstruct(int v)
            : Value(v)
        {
            ++constructorCounter;
        }

        TCountConstruct& operator=(int v) {
            Value = v;
            ++assignmentCounter;
            return *this;
        }

        TCountConstruct(const TCountConstruct&) = delete;
        int Value;
    };

    THashMap<int, TCountConstruct> hash;
    {
        auto r = hash.insert_or_assign(TNonCopyableInt<4>(4), 1);
        UNIT_ASSERT(r.second);
        UNIT_ASSERT_VALUES_EQUAL(1, hash.size());
        UNIT_ASSERT_VALUES_EQUAL(1, constructorCounter);
        UNIT_ASSERT_VALUES_EQUAL(0, assignmentCounter);
        UNIT_ASSERT_VALUES_EQUAL(1, r.first->second.Value);
    }
    {
        auto r = hash.insert_or_assign(TNonCopyableInt<4>(4), 5);
        UNIT_ASSERT(!r.second);
        UNIT_ASSERT_VALUES_EQUAL(1, hash.size());
        UNIT_ASSERT_VALUES_EQUAL(1, constructorCounter);
        UNIT_ASSERT_VALUES_EQUAL(1, assignmentCounter);
        UNIT_ASSERT_VALUES_EQUAL(5, r.first->second.Value);
    }
    {
        constexpr int iterations = 200;
        for (int iteration = 0; iteration < iterations; ++iteration) {
            hash.insert_or_assign(iteration, iteration);
        }
        UNIT_ASSERT_VALUES_EQUAL(iterations, hash.size());
        UNIT_ASSERT_VALUES_EQUAL(iterations, constructorCounter);
        UNIT_ASSERT_VALUES_EQUAL(2, assignmentCounter);
        UNIT_ASSERT_VALUES_EQUAL(4, hash.at(4).Value);
        UNIT_ASSERT_VALUES_EQUAL(44, hash.at(44).Value);
    }
}

void THashTest::TestHMMapEmplace() {
    using hash_t = THashMultiMap<int, TNonCopyableInt<0>>;