/*
 * unit test for name server replica.

 --------------------------------------------------------------------------

 Authors:	Ralph H. Castain <rhc@lanl.gov>

 --------------------------------------------------------------------------

*/

#include <stdio.h>
#include <string.h>

#include "ompi_config.h"
#include "support.h"

#include "include/constants.h"
#include "util/sys_info.h"
#include "util/proc_info.h"
#include "mca/mca.h"
#include "mca/base/base.h"
#include "mca/ns/base/base.h"

/* output files needed by the test */
static FILE *test_out=NULL;

static char *cmd_str="diff ./test_ns_replica_out ./test_ns_replica_out_std";

int main(int argc, char **argv)
{
    ompi_process_name_t *test_name;
    mca_ns_base_cellid_t cell;
    mca_ns_base_jobid_t job;
    mca_ns_base_vpid_t vpid;
    bool multi, hidden;
    int i, j;
    char *tmp;
    int result; /* result from system call */

    test_init("test_ns_replica");

    test_out = fopen( "test_ns_replica_out", "w+" );
    if( test_out == NULL ) {
      test_failure("test_ns_replica couldn't open test file failed");
      test_finalize();
      exit(1);
    } 

    ompi_process_info.seed = true;

    /* startup the MCA */
    if (OMPI_SUCCESS == mca_base_open()) {
	fprintf(test_out, "MCA started\n");
    } else {
	fprintf(test_out, "MCA could not start - please report error to bugs@open-mpi.org\n");
	exit (1);
    }

    /* open the name server */
    if (OMPI_SUCCESS == mca_ns_base_open()) {
	fprintf(test_out, "NS opened\n");
	test_success();
    } else {
	fprintf(test_out, "NS could not open\n");
        test_failure("test_ns_replica mca_ns_base_open failed");
        test_finalize();
	exit(1);
    }

    /* startup the name server */
    if (OMPI_SUCCESS != mca_ns_base_select(&multi, &hidden)) {
	fprintf(test_out, "NS could not start\n");
	test_failure("test_ns_replica mca_ns_base_select failed");
        test_finalize();
	exit(1);
    } else {
	fprintf(test_out, "NS started\n");
	test_success();
    }

    /* create a name */
    test_name = ompi_name_server.create_process_name(0, 1, 1);
    if (NULL == test_name) { /* got error */
	fprintf(test_out, "create process name failed\n");
	test_failure("test_ns_replica ompi_name_server create_process_name failed");
        test_finalize();
	exit(1);
    } else {
	fprintf(test_out, "got process name: %0X %0X %0X\n", 
		test_name->cellid, test_name->jobid, test_name->vpid);
	test_success();
    }

    /* convert a string to a name */
    tmp = strdup("1234.5678.9AEF");
    test_name = ompi_name_server.convert_string_to_process_name(tmp);
    if (NULL == test_name) {  /* got error */
	fprintf(test_out, "convert string to process name failed\n");
	test_failure("test_ns_replica ompi_name_server convert_string_to_process_name failed");
        test_finalize();
	exit(1);
    } else {
      fprintf(test_out, "got process name: %0X(%ld)  %0X(%ld)  %0X(%ld)\n", 
	      (int)test_name->cellid, (long int)test_name->cellid, (int)test_name->jobid, 
	      (long int)test_name->jobid, (unsigned int)test_name->vpid, 
	      (long int)test_name->vpid);
        test_success();
    }

    /* create a cellid */
    cell = ompi_name_server.create_cellid();
    if (0 == cell) { /* got error */
        test_failure("test_ns_replica ompi_name_server test create_cellid");
	fprintf(test_out, "create cellid: error\n");
	test_finalize();
	exit(1);
    } else {
      fprintf(test_out, "cellid created: %d\n", cell);
      test_success();
    }

    for (i=0; i<10; i++) { /* loop through */
	/* create jobid */
	job = ompi_name_server.create_jobid();
	if (0 == job) { /* got error */
	    fprintf(test_out, "create jobid: error\n");
	    test_failure("test_ns_replica ompi_name_server create_jobid failed");
	    test_finalize();
	    exit(1);
	} else {
	    fprintf(test_out, "jobid created: %d\n", job);
	    test_success();
	}

	for (j=0; j<5; j++) { /* loop through several vpid ranges */
	    /* get range of vpids */
	    vpid = ompi_name_server.reserve_range(job, 250);
	    if (0 == vpid) { /* got error */
		fprintf(test_out, "get range: error\n");
		test_failure("test_ns_replica ompi_name_server reserve_range failed");
		test_finalize();
		exit(1);
	    } else {
		fprintf(test_out, "range reserved: %d\n", vpid);
		test_success();
	    }

	    /* create a name */
	    test_name = ompi_name_server.create_process_name((mca_ns_base_cellid_t)i, 
							     job, vpid);

	    /* get and print its string values */
	    tmp = ompi_name_server.get_proc_name_string(test_name);
	    fprintf(test_out, "(%d) strings: name - %s\n", i, tmp);
	    tmp = ompi_name_server.get_vpid_string(test_name);
	    fprintf(test_out, "\tvpid: %s\n", tmp);
	    tmp = ompi_name_server.get_jobid_string(test_name);
	    fprintf(test_out, "\tjobid: %s\n", tmp);
	    tmp = ompi_name_server.get_cellid_string(test_name);
	    fprintf(test_out, "\tcellid: %s\n", tmp);

	    /* get and print its numeric values */
	    vpid = ompi_name_server.get_vpid(test_name);
	    job = ompi_name_server.get_jobid(test_name);
	    cell = ompi_name_server.get_cellid(test_name);
	    fprintf(test_out, "(%d) ints cell %0X(%ld) job %0X(%ld) vpid %0x(%ld)\n\n", 
		    i, cell, (long int)cell, job, (long int)job, vpid, (long int)vpid);
	}

    }

    fclose( test_out );
    result = system( cmd_str );
    if( result == 0 ) {
        test_success();
    }
    else {
      test_failure( "test_ns_replica ompi_name_server get_proc_name_string, etc failed");
    }

    test_finalize();

    return(0);
}