/* -*- Mode: C; c-basic-offset:4 ; indent-tabs-mode:nil ; -*- */

/*

 *  (C) 2011 by Argonne National Laboratory.

 *      See COPYRIGHT in top-level directory.

 */

/*

  Tests invocation of error handlers on a variety of communicators, including

  COMM_NULL.  

*/

#include "mpitestconf.h"

#include <mpi.h>

#include <iostream>

#include "mpitestcxx.h"

/* returns number of errors found */

int testCommCall( MPI::Comm &comm )

{

    int i;

    bool sawException = false;

    comm.Set_errhandler( MPI::ERRORS_THROW_EXCEPTIONS );

    try {

	// Invalid send

	comm.Send( &i, -1, MPI::DATATYPE_NULL, -1, -10 );

    } catch (MPI::Exception &ex ) {

	// This is good

	sawException = true;

    }

    catch (...) {

	// This is bad

    }

    if (!sawException) {

	int len;

	char commname[MPI_MAX_OBJECT_NAME];

	comm.Get_name( commname, len );

	std::cout << "Did not see MPI exception on invalid call on communicator " <<

	    commname << "\n";

    }

    return sawException ? 0 : 1;

}

int testNullCommCall( void )

{

    int i;

    bool sawException = false;

    const MPI::Comm &comm = MPI::COMM_NULL;

    MPI::COMM_WORLD.Set_errhandler( MPI::ERRORS_THROW_EXCEPTIONS );

    try {

	// Invalid send

	comm.Send( &i, -1, MPI::DATATYPE_NULL, -1, -10 );

    } catch (MPI::Exception &ex ) {

	// This is good

	sawException = true;

    }

    catch (...) {

	// This is bad

    }

    if (!sawException) {

	std::cout << "Did not see MPI exception on invalid call on communicator COMM_NULL\n";

    }

    return sawException ? 0 : 1;

}

int main( int argc, char *argv[] )

{

    int errs = 0;

    bool periods[2] = { false, false };

    int  dims[2] = { 0, 0 };

    MTest_Init( );

    MPI::Compute_dims( MPI::COMM_WORLD.Get_size(), 2, dims );

    MPI::Cartcomm cart = MPI::COMM_WORLD.Create_cart( 2, dims, periods, true );

    cart.Set_name( "Cart comm" );

    // Graphcomm graph = COMM_WORLD.Create_graph( );

    errs += testCommCall( MPI::COMM_WORLD );

    errs += testNullCommCall( );

    errs += testCommCall( MPI::COMM_SELF );

    errs += testCommCall( cart );

    //errs += testCommCall( graph );

    MTest_Finalize( errs );

    cart.Free();

    // graph.Free();

    MPI::Finalize();

    return 0;

}