1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495 |
- /* Boost.MultiIndex test for special set operations.
- *
- * Copyright 2003-2015 Joaquin M Lopez Munoz.
- * 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_special_set_ops.hpp"
- #include <boost/config.hpp> /* keep it first to prevent nasty warns in MSVC */
- #include <algorithm>
- #include <sstream>
- #include "pre_multi_index.hpp"
- #include "employee.hpp"
- #include <boost/detail/lightweight_test.hpp>
- using namespace boost::multi_index;
- static int string_to_int(const std::string& str)
- {
- std::istringstream iss(str);
- int res;
- iss>>res;
- return res;
- }
- struct comp_int_string
- {
- bool operator()(int x,const std::string& y)const
- {
- return x<string_to_int(y);
- }
- bool operator()(const std::string& x,int y)const
- {
- return string_to_int(x)<y;
- }
- };
- struct hash_string_as_int
- {
- int operator()(const std::string& x)const
- {
- return static_cast<int>(boost::hash<int>()(string_to_int(x)));
- }
- };
- struct eq_string_int
- {
- bool operator()(int x,const std::string& y)const
- {
- return x==string_to_int(y);
- }
- bool operator()(const std::string& x,int y)const
- {
- return operator()(y,x);
- }
- };
- void test_special_set_ops()
- {
- employee_set 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,"Albert",20,9012));
- es.insert(employee(4,"John",57,1002));
- std::pair<employee_set_by_ssn::iterator,employee_set_by_ssn::iterator> p=
- get<ssn>(es).equal_range(
- "7889",hash_string_as_int(),eq_string_int());
- BOOST_TEST(std::distance(p.first,p.second)==1&&(p.first)->id==2);
- BOOST_TEST(
- get<ssn>(es).count(
- "5601",hash_string_as_int(),eq_string_int())==1);
- BOOST_TEST(
- get<ssn>(es).find(
- "1123",hash_string_as_int(),eq_string_int())->name=="Joe");
- BOOST_TEST(
- std::distance(
- get<age>(es).lower_bound("27",comp_int_string()),
- get<age>(es).upper_bound("40",comp_int_string()))==3);
- BOOST_TEST(es.count(2,employee::comp_id())==1);
- BOOST_TEST(es.count(5,employee::comp_id())==0);
- }
|