llvm/libcxx/test/std/containers/associative/map/map.modifiers/insert_or_assign.pass.cpp

//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
// UNSUPPORTED: c++03, c++11, c++14

// <map>

// class map

// template <class M>
//  pair<iterator, bool> insert_or_assign(const key_type& k, M&& obj);            // C++17
// template <class M>
//  pair<iterator, bool> insert_or_assign(key_type&& k, M&& obj);                 // C++17
// template <class M>
//  iterator insert_or_assign(const_iterator hint, const key_type& k, M&& obj);   // C++17
// template <class M>
//  iterator insert_or_assign(const_iterator hint, key_type&& k, M&& obj);        // C++17

#include <map>
#include <cassert>
#include <iterator>
#include <tuple>

#include "test_macros.h"

class Moveable
{
    Moveable(const Moveable&);
    Moveable& operator=(const Moveable&);

    int int_;
    double double_;
public:
    Moveable() : int_(0), double_(0) {}
    Moveable(int i, double d) : int_(i), double_(d) {}
    Moveable(Moveable&& x)
        : int_(x.int_), double_(x.double_)
            {x.int_ = -1; x.double_ = -1;}
    Moveable& operator=(Moveable&& x)
        {int_ = x.int_; x.int_ = -1;
         double_ = x.double_; x.double_ = -1;
         return *this;
        }

    bool operator==(const Moveable& x) const
        {return int_ == x.int_ && double_ == x.double_;}
    bool operator<(const Moveable& x) const
        {return int_ < x.int_ || (int_ == x.int_ && double_ < x.double_);}

    int get() const {return int_;}
    bool moved() const {return int_ == -1;}
};


int main(int, char**)
{
    { // pair<iterator, bool> insert_or_assign(const key_type& k, M&& obj);
        typedef std::map<int, Moveable> M;
        typedef std::pair<M::iterator, bool> R;
        M m;
        R r;
        for ( int i = 0; i < 20; i += 2 )
            m.emplace ( i, Moveable(i, (double) i));
        assert(m.size() == 10);

        for (int i=0; i < 20; i += 2)
        {
            Moveable mv(i+1, i+1);
            r = m.insert_or_assign(i, std::move(mv));
            assert(m.size() == 10);
            assert(!r.second);                    // was not inserted
            assert(mv.moved());                   // was moved from
            assert(r.first->first == i);          // key
            assert(r.first->second.get() == i+1); // value
        }

        Moveable mv1(5, 5.0);
        r = m.insert_or_assign(-1, std::move(mv1));
        assert(m.size() == 11);
        assert(r.second);                    // was inserted
        assert(mv1.moved());                 // was moved from
        assert(r.first->first        == -1); // key
        assert(r.first->second.get() == 5);  // value

        Moveable mv2(9, 9.0);
        r = m.insert_or_assign(3, std::move(mv2));
        assert(m.size() == 12);
        assert(r.second);                   // was inserted
        assert(mv2.moved());                // was moved from
        assert(r.first->first        == 3); // key
        assert(r.first->second.get() == 9); // value

        Moveable mv3(-1, 5.0);
        r = m.insert_or_assign(117, std::move(mv3));
        assert(m.size() == 13);
        assert(r.second);                     // was inserted
        assert(mv3.moved());                  // was moved from
        assert(r.first->first        == 117); // key
        assert(r.first->second.get() == -1);  // value
    }
    { // pair<iterator, bool> insert_or_assign(key_type&& k, M&& obj);
        typedef std::map<Moveable, Moveable> M;
        typedef std::pair<M::iterator, bool> R;
        M m;
        R r;
        for ( int i = 0; i < 20; i += 2 )
            m.emplace ( Moveable(i, (double) i), Moveable(i+1, (double) i+1));
        assert(m.size() == 10);

        Moveable mvkey1(2, 2.0);
        Moveable mv1(4, 4.0);
        r = m.insert_or_assign(std::move(mvkey1), std::move(mv1));
        assert(m.size() == 10);
        assert(!r.second);                  // was not inserted
        assert(!mvkey1.moved());            // was not moved from
        assert(mv1.moved());                // was moved from
        assert(r.first->first == mvkey1);   // key
        assert(r.first->second.get() == 4); // value

        Moveable mvkey2(3, 3.0);
        Moveable mv2(5, 5.0);
        r = m.try_emplace(std::move(mvkey2), std::move(mv2));
        assert(m.size() == 11);
        assert(r.second);                   // was inserted
        assert(mv2.moved());                // was moved from
        assert(mvkey2.moved());             // was moved from
        assert(r.first->first.get()  == 3); // key
        assert(r.first->second.get() == 5); // value
    }
    { // iterator insert_or_assign(const_iterator hint, const key_type& k, M&& obj);
        typedef std::map<int, Moveable> M;
        M m;
        M::iterator r;
        for ( int i = 0; i < 20; i += 2 )
            m.emplace ( i, Moveable(i, (double) i));
        assert(m.size() == 10);
        M::const_iterator it = m.find(2);

        Moveable mv1(3, 3.0);
        r = m.insert_or_assign(it, 2, std::move(mv1));
        assert(m.size() == 10);
        assert(mv1.moved());           // was moved from
        assert(r->first        == 2);  // key
        assert(r->second.get() == 3);  // value

        Moveable mv2(5, 5.0);
        r = m.insert_or_assign(it, 3, std::move(mv2));
        assert(m.size() == 11);
        assert(mv2.moved());           // was moved from
        assert(r->first        == 3);  // key
        assert(r->second.get() == 5);  // value

        // wrong hint: begin()
        Moveable mv3(7, 7.0);
        r = m.insert_or_assign(m.begin(), 4, std::move(mv3));
        assert(m.size() == 11);
        assert(mv3.moved());           // was moved from
        assert(r->first        == 4);  // key
        assert(r->second.get() == 7);  // value

        Moveable mv4(9, 9.0);
        r = m.insert_or_assign(m.begin(), 5, std::move(mv4));
        assert(m.size() == 12);
        assert(mv4.moved());           // was moved from
        assert(r->first        == 5);  // key
        assert(r->second.get() == 9);  // value

        // wrong hint: end()
        Moveable mv5(11, 11.0);
        r = m.insert_or_assign(m.end(), 6, std::move(mv5));
        assert(m.size() == 12);
        assert(mv5.moved());           // was moved from
        assert(r->first        == 6);  // key
        assert(r->second.get() == 11); // value

        Moveable mv6(13, 13.0);
        r = m.insert_or_assign(m.end(), 7, std::move(mv6));
        assert(m.size() == 13);
        assert(mv6.moved());           // was moved from
        assert(r->first        == 7);  // key
        assert(r->second.get() == 13); // value

        // wrong hint: third element
        Moveable mv7(15, 15.0);
        r = m.insert_or_assign(std::next(m.begin(), 2), 8, std::move(mv7));
        assert(m.size() == 13);
        assert(mv7.moved());           // was moved from
        assert(r->first        == 8);  // key
        assert(r->second.get() == 15); // value

        Moveable mv8(17, 17.0);
        r = m.insert_or_assign(std::next(m.begin(), 2), 9, std::move(mv8));
        assert(m.size() == 14);
        assert(mv8.moved());           // was moved from
        assert(r->first        == 9);  // key
        assert(r->second.get() == 17); // value
    }
    { // iterator insert_or_assign(const_iterator hint, key_type&& k, M&& obj);
        typedef std::map<Moveable, Moveable> M;
        M m;
        M::iterator r;
        for ( int i = 0; i < 20; i += 2 )
            m.emplace ( Moveable(i, (double) i), Moveable(i+1, (double) i+1));
        assert(m.size() == 10);
        M::const_iterator it = std::next(m.cbegin());

        Moveable mvkey1(2, 2.0);
        Moveable mv1(4, 4.0);
        r = m.insert_or_assign(it, std::move(mvkey1), std::move(mv1));
        assert(m.size() == 10);
        assert(mv1.moved());          // was moved from
        assert(!mvkey1.moved());      // was not moved from
        assert(r->first == mvkey1);   // key
        assert(r->second.get() == 4); // value

        Moveable mvkey2(3, 3.0);
        Moveable mv2(5, 5.0);
        r = m.insert_or_assign(it, std::move(mvkey2), std::move(mv2));
        assert(m.size() == 11);
        assert(mv2.moved());           // was moved from
        assert(mvkey2.moved());        // was moved from
        assert(r->first.get()  == 3);  // key
        assert(r->second.get() == 5);  // value

        // wrong hint: begin()
        Moveable mvkey3(6, 6.0);
        Moveable mv3(8, 8.0);
        r = m.insert_or_assign(m.begin(), std::move(mvkey3), std::move(mv3));
        assert(m.size() == 11);
        assert(mv3.moved());           // was moved from
        assert(!mvkey3.moved());       // was not moved from
        assert(r->first == mvkey3);    // key
        assert(r->second.get() == 8);  // value

        Moveable mvkey4(7, 7.0);
        Moveable mv4(9, 9.0);
        r = m.insert_or_assign(m.begin(), std::move(mvkey4), std::move(mv4));
        assert(m.size() == 12);
        assert(mv4.moved());           // was moved from
        assert(mvkey4.moved());        // was moved from
        assert(r->first.get()  == 7);  // key
        assert(r->second.get() == 9);  // value

        // wrong hint: end()
        Moveable mvkey5(8, 8.0);
        Moveable mv5(10, 10.0);
        r = m.insert_or_assign(m.end(), std::move(mvkey5), std::move(mv5));
        assert(m.size() == 12);
        assert(mv5.moved());           // was moved from
        assert(!mvkey5.moved());       // was not moved from
        assert(r->first == mvkey5);    // key
        assert(r->second.get() == 10); // value

        Moveable mvkey6(9, 9.0);
        Moveable mv6(11, 11.0);
        r = m.insert_or_assign(m.end(), std::move(mvkey6), std::move(mv6));
        assert(m.size() == 13);
        assert(mv6.moved());           // was moved from
        assert(mvkey6.moved());        // was moved from
        assert(r->first.get()  == 9);  // key
        assert(r->second.get() == 11); // value

        // wrong hint: third element
        Moveable mvkey7(10, 10.0);
        Moveable mv7(12, 12.0);
        r = m.insert_or_assign(std::next(m.begin(), 2), std::move(mvkey7), std::move(mv7));
        assert(m.size() == 13);
        assert(mv7.moved());           // was moved from
        assert(!mvkey7.moved());       // was not moved from
        assert(r->first == mvkey7);    // key
        assert(r->second.get() == 12); // value

        Moveable mvkey8(11, 11.0);
        Moveable mv8(13, 13.0);
        r = m.insert_or_assign(std::next(m.begin(), 2), std::move(mvkey8), std::move(mv8));
        assert(m.size() == 14);
        assert(mv8.moved());           // was moved from
        assert(mvkey8.moved());        // was moved from
        assert(r->first.get()  == 11); // key
        assert(r->second.get() == 13); // value
    }

  return 0;
}