// Copyright (C) 2017 Alain Miniussi & Vincent Chabannes

// Use, modification and distribution is subject to 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)

#include <string>
#include <iostream>
#include <sstream>
#include <vector>

#include <boost/mpi.hpp>
#include <boost/mpi/nonblocking.hpp>
#include <boost/serialization/string.hpp>

#define BOOST_TEST_MODULE mpi_wait_any
#include <boost/test/included/unit_test.hpp>

namespace mpi = boost::mpi;

BOOST_AUTO_TEST_CASE(wait_any)
{
  mpi::environment  env;
  mpi::communicator comm;
  
  int rank = comm.rank();
  int const sz = 10;
  std::vector<int> data;
  std::vector< mpi::request> reqs;
  if ( rank == 0 ) {
    for ( int i=0; i<sz; ++i ) {
      data.push_back( i );
    }
    reqs.push_back( comm.isend(1, 0, data) );
  } else if ( rank == 1 ) {
    reqs.push_back( comm.irecv(0, 0, data) );
  }
  mpi::wait_all( reqs.begin(), reqs.end() );
  
  if ( rank == 1 ) {
    BOOST_CHECK(data.size() == sz);
    for ( int i=0; i<sz; ++i ) {
      BOOST_CHECK(data[i] == i);
    }
  }  
}