/* Boost.MultiIndex test for replace(), modify() and modify_key().
 *
 * Copyright 2003-2005 Joaquín M López Muñoz.
 * Distributed under the Boost Software License, Version 1.0.
 * (See accompanying file LICENSE_1_0.txt or copy at
 * http://www.boost.org/LICENSE_1_0.txt)
 *
 * See http://www.boost.org/libs/multi_index for library home page.
 */

#include "test_update.hpp"

#include <boost/config.hpp> /* keep it first to prevent nasty warns in MSVC */
#include <algorithm>
#include "pre_multi_index.hpp"
#include "employee.hpp"
#include "pair_of_ints.hpp"
#include <boost/test/test_tools.hpp>

using namespace boost::multi_index;

void test_update()
{
  employee_set              es;
  employee_set_as_inserted& i=get<as_inserted>(es);

  es.insert(employee(0,"Joe",31,1123));
  es.insert(employee(1,"Robert",27,5601));
  es.insert(employee(2,"John",40,7889));
  es.insert(employee(3,"Olbert",20,9012));
  es.insert(employee(4,"John",57,1002));

  employee_set::iterator             it=es.find(employee(0,"Joe",31,1123));
  employee_set_as_inserted::iterator it1=
    project<as_inserted>(es,get<name>(es).find("Olbert"));

  BOOST_CHECK(es.replace(it,*it));
  BOOST_CHECK(!es.replace(it,employee(3,"Joe",31,1123))&&it->id==0);
  BOOST_CHECK(es.replace(it,employee(0,"Joe",32,1123))&&it->age==32);
  BOOST_CHECK(i.replace(it1,employee(3,"Albert",20,9012))&&it1->name==
                "Albert");

  {
    typedef multi_index_container<
      pair_of_ints,
      indexed_by<
        ordered_unique<BOOST_MULTI_INDEX_MEMBER(pair_of_ints,int,first)>,
        hashed_unique<BOOST_MULTI_INDEX_MEMBER(pair_of_ints,int,second)>,
        sequenced<> > >
    int_int_set;

    int_int_set iis;
    nth_index<int_int_set,1>::type& ii1=get<1>(iis);
    nth_index<int_int_set,2>::type& ii2=get<2>(iis);
    iis.insert(pair_of_ints(0,0));
    iis.insert(pair_of_ints(5,5));
    iis.insert(pair_of_ints(10,10));

    BOOST_CHECK(!iis.replace(iis.begin(),pair_of_ints(5,0)));
    BOOST_CHECK(!ii2.replace(ii2.begin(),pair_of_ints(0,5)));
    BOOST_CHECK(!ii1.replace(project<1>(iis,iis.begin()),pair_of_ints(5,11)));
    BOOST_CHECK(!iis.replace(iis.begin(),pair_of_ints(11,5)));
    BOOST_CHECK(!iis.replace(++iis.begin(),pair_of_ints(10,5)));
    BOOST_CHECK(!ii1.replace(
      project<1>(iis,++iis.begin()),pair_of_ints(5,10)));
    BOOST_CHECK(!iis.replace(--iis.end(),pair_of_ints(5,10)));
    BOOST_CHECK(!ii2.replace(--ii2.end(),pair_of_ints(10,5)));

    BOOST_CHECK(iis.modify(iis.begin(),increment_first));
    BOOST_CHECK(ii2.modify(ii2.begin(),increment_first));
    BOOST_CHECK(ii1.modify(project<1>(iis,iis.begin()),increment_first));
    BOOST_CHECK(ii2.modify(ii2.begin(),increment_first));

    BOOST_CHECK(!iis.modify(iis.begin(),increment_first));
    BOOST_CHECK(iis.size()==2);

    iis.insert(pair_of_ints(0,0));
    BOOST_CHECK(ii2.modify(--ii2.end(),increment_second));
    BOOST_CHECK(iis.modify(iis.begin(),increment_second));
    BOOST_CHECK(ii2.modify(--ii2.end(),increment_second));
    BOOST_CHECK(iis.modify(iis.begin(),increment_second));

    BOOST_CHECK(!ii2.modify(--ii2.end(),increment_second));
    BOOST_CHECK(ii2.size()==2);

    iis.insert(pair_of_ints(0,0));
    BOOST_CHECK(iis.modify_key(iis.begin(),increment_int));
    BOOST_CHECK(iis.modify_key(iis.begin(),increment_int));
    BOOST_CHECK(iis.modify_key(iis.begin(),increment_int));
    BOOST_CHECK(iis.modify_key(iis.begin(),increment_int));

    BOOST_CHECK(!iis.modify_key(iis.begin(),increment_int));
    BOOST_CHECK(iis.size()==2);

    nth_index_iterator<int_int_set,1>::type it=ii1.find(5);
    BOOST_CHECK(ii1.modify_key(it,increment_int));
    BOOST_CHECK(ii1.modify_key(it,increment_int));
    BOOST_CHECK(ii1.modify_key(it,increment_int));
    BOOST_CHECK(ii1.modify_key(it,increment_int));

    BOOST_CHECK(!ii1.modify_key(it,increment_int));
    BOOST_CHECK(ii1.size()==1);
  }
  {
    typedef multi_index_container<
      pair_of_ints,
      indexed_by<
        hashed_unique<BOOST_MULTI_INDEX_MEMBER(pair_of_ints,int,first)>,
        sequenced<>,
        ordered_unique<BOOST_MULTI_INDEX_MEMBER(pair_of_ints,int,second)> > >
    int_int_set;

    int_int_set iis;
    nth_index<int_int_set,1>::type& ii1=get<1>(iis);
    int_int_set::iterator p1=iis.insert(pair_of_ints(0,0)).first;
    int_int_set::iterator p2=iis.insert(pair_of_ints(5,5)).first;
    int_int_set::iterator p3=iis.insert(pair_of_ints(10,10)).first;

    BOOST_CHECK(!iis.replace(p1,pair_of_ints(5,0)));
    BOOST_CHECK(!ii1.replace(ii1.begin(),pair_of_ints(0,5)));
    BOOST_CHECK(!iis.replace(p1,pair_of_ints(5,11)));
    BOOST_CHECK(!iis.replace(p1,pair_of_ints(11,5)));
    BOOST_CHECK(!iis.replace(p2,pair_of_ints(10,5)));
    BOOST_CHECK(!iis.replace(p2,pair_of_ints(5,10)));
    BOOST_CHECK(!iis.replace(p3,pair_of_ints(5,10)));
    BOOST_CHECK(!ii1.replace(--ii1.end(),pair_of_ints(10,5)));

    BOOST_CHECK(iis.modify(p1,increment_first));
    BOOST_CHECK(ii1.modify(ii1.begin(),increment_first));
    BOOST_CHECK(iis.modify(p1,increment_first));
    BOOST_CHECK(ii1.modify(ii1.begin(),increment_first));

    BOOST_CHECK(!iis.modify(p1,increment_first));
    BOOST_CHECK(iis.size()==2);

    p1=iis.insert(pair_of_ints(0,0)).first;
    BOOST_CHECK(ii1.modify(--ii1.end(),increment_second));
    BOOST_CHECK(iis.modify(p1,increment_second));
    BOOST_CHECK(ii1.modify(--ii1.end(),increment_second));
    BOOST_CHECK(iis.modify(p1,increment_second));

    BOOST_CHECK(!ii1.modify(--ii1.end(),increment_second));
    BOOST_CHECK(ii1.size()==2);
  }
}
