/*
 * Copyright (C) by Argonne National Laboratory
 *     See COPYRIGHT in top-level directory
 */

#include "mpi.h"
#include "mpitestconf.h"
#ifdef HAVE_IOSTREAM
// Not all C++ compilers have iostream instead of iostream.h
#include <iostream>
#ifdef HAVE_NAMESPACE_STD
// Those that do often need the std namespace; otherwise, a bare "cout"
// is likely to fail to compile
using namespace std;
#endif
#else
#include <iostream.h>
#endif
#include "mpitestcxx.h"
#ifdef HAVE_STRING_H
#include <string.h>
#endif

#define NKEYS 3
int main(int argc, char *argv[])
{
    int errs = 0;
    MPI::Info info;
    const char *keys[NKEYS] = { "file", "soft", "host" };
    const char *values[NKEYS] = { "runfile.txt", "2:1000:4,3:1000:7",
        "myhost.myorg.org"
    };
    char *value;
    int i, flag, nkeys;

    MTest_Init();

    value = new char[MPI::MAX_INFO_VAL];

    info = MPI::Info::Create();
    /* Use only named keys incase the info implementation only supports
     * the predefined keys (e.g., IBM) */
    for (i = 0; i < NKEYS; i++) {
        info.Set(keys[i], values[i]);
    }

    /* Check that all values are present */
    for (i = 0; i < NKEYS; i++) {
        flag = info.Get(keys[i], MPI::MAX_INFO_VAL, value);
        if (!flag) {
            errs++;
            cout << "No value for key " << keys[i] << "\n";
        }
        if (strcmp(value, values[i])) {
            errs++;
            cout << "Incorrect value for key " << keys[i] <<
                ", got " << value << " expected " << values[i] << "\n";
        }
    }

    /* Now, change one value and remove another, then check again */
    info.Delete(keys[NKEYS - 1]);
    nkeys = info.Get_nkeys();
    if (nkeys != NKEYS - 1) {
        errs++;
        cout << "Deleting a key did not change the number of keys\n";
    }

    values[0] = (char *) "backfile.txt";
    info.Set(keys[0], values[0]);
    for (i = 0; i < NKEYS - 1; i++) {
        flag = info.Get(keys[i], MPI::MAX_INFO_VAL, value);
        if (!flag) {
            errs++;
            cout << "(after reset) No value for key " << keys[i] << "\n";
        }
        if (strcmp(value, values[i])) {
            errs++;
            cout << "(after reset) Incorrect value for key " <<
                keys[i] << ", got " << value << " expected " << values[i] << "\n";
        }
    }

    info.Free();
    delete[]value;

    MTest_Finalize(errs);

    return 0;
}
