/* * unit test for name server replica. -------------------------------------------------------------------------- Authors: Ralph H. Castain -------------------------------------------------------------------------- */ #include "orte_config.h" #include #include #include "support.h" #include "include/orte_constants.h" #include "include/orte_schema.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) { orte_process_name_t *test_name; orte_cellid_t cell; orte_jobid_t job; orte_vpid_t vpid; int i, j, rc; char *tmp; 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); } /* Open up the output streams */ if (!ompi_output_init()) { return OMPI_ERROR; } /* * If threads are supported - assume that we are using threads - and reset otherwise. */ ompi_set_using_threads(OMPI_HAVE_THREAD_SUPPORT); /* For malloc debugging */ ompi_malloc_init(); /* set seed to true to force replica selection */ orte_process_info.seed = true; /* ensure the replica is isolated */ setenv("OMPI_MCA_ns_replica_isolate", "1", 1); /* init the proc info structure */ orte_proc_info(); /* 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 == orte_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 != orte_ns_base_select()) { 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 */ if (ORTE_SUCCESS != (rc = orte_ns.create_process_name(&test_name, 0, 1, 1))) { /* got error */ fprintf(test_out, "create process name failed with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica orte_ns create_process_name failed"); test_finalize(); exit(1); } else { fprintf(test_out, "got process name: %0X %0X %0X\n", ORTE_NAME_ARGS(test_name)); test_success(); } free(test_name); /* convert a string to a name */ tmp = strdup("1234.5678.9AEF"); if (ORTE_SUCCESS != (rc = orte_ns.convert_string_to_process_name(&test_name, tmp))) { /* got error */ fprintf(test_out, "convert string to process name failed with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica orte_ns convert_string_to_process_name failed"); test_finalize(); exit(1); } else { fprintf(test_out, "got process name: %0X %0X %0X\n", ORTE_NAME_ARGS(test_name)); test_success(); } free(tmp); free(test_name); /* create a cellid */ if (ORTE_SUCCESS != (rc = orte_ns.create_cellid(&cell))) { /* got error */ test_failure("test_ns_replica orte_ns test create_cellid failed"); fprintf(test_out, "create cellid: error with error %s\n", ORTE_ERROR_NAME(rc)); 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 */ if (ORTE_SUCCESS != (rc = orte_ns.create_jobid(&job))) { /* got error */ fprintf(test_out, "create jobid: error with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica orte_ns 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 */ if (ORTE_SUCCESS != (rc = orte_ns.reserve_range(job, 250, &vpid))) { /* got error */ fprintf(test_out, "get range: error with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica orte_ns reserve_range failed"); test_finalize(); exit(1); } else { fprintf(test_out, "range reserved: %d\n", vpid); test_success(); } /* create a name */ if (ORTE_SUCCESS != (rc = orte_ns.create_process_name(&test_name, (orte_cellid_t)i, job, vpid))) { fprintf(test_out, "test_ns_replica: failed to create proc name after vpid range with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to create proc name after vpid range"); test_finalize(); exit(1); } /* get and print its string values */ if (ORTE_SUCCESS != (rc = orte_ns.get_proc_name_string(&tmp, test_name))) { fprintf(test_out, "test_ns_replica: failed to get proc_name_string with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get proc_name_string"); test_finalize(); exit(1); } else { fprintf(test_out, "(%d) strings: name - %s\n", i, tmp); } free(tmp); if (ORTE_SUCCESS != (rc = orte_ns.get_vpid_string(&tmp, test_name))) { fprintf(test_out, "test_ns_replica: failed to get vpid_string with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get vpid_string"); test_finalize(); exit(1); } else { fprintf(test_out, "(%d) strings: vpid - %s\n", i, tmp); } free(tmp); if (ORTE_SUCCESS != (rc = orte_ns.get_jobid_string(&tmp, test_name))) { fprintf(test_out, "test_ns_replica: failed to get jobid_string with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get jobid_string"); test_finalize(); exit(1); } else { fprintf(test_out, "(%d) strings: jobid - %s\n", i, tmp); } free(tmp); if (ORTE_SUCCESS != (rc = orte_ns.get_cellid_string(&tmp, test_name))) { fprintf(test_out, "test_ns_replica: failed to get cellid_string with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get cellid_string"); test_finalize(); exit(1); } else { fprintf(test_out, "(%d) strings: cellid - %s\n", i, tmp); } free(tmp); /* get and print its numeric values */ if (ORTE_SUCCESS != (rc = orte_ns.get_vpid(&vpid, test_name))) { fprintf(test_out, "test_ns_replica: failed to get vpid with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get vpid"); test_finalize(); exit(1); } if (ORTE_SUCCESS != (rc = orte_ns.get_jobid(&job, test_name))) { fprintf(test_out, "test_ns_replica: failed to get jobid with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get jobid"); test_finalize(); exit(1); } if (ORTE_SUCCESS != (rc = orte_ns.get_cellid(&cell, test_name))) { fprintf(test_out, "test_ns_replica: failed to get cellid with error %s\n", ORTE_ERROR_NAME(rc)); test_failure("test_ns_replica: failed to get cellid"); test_finalize(); exit(1); } 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); free(test_name); } } /* finalize and see if memory cleared */ orte_ns_base_close(); orte_proc_info_finalize(); mca_base_close(); ompi_malloc_finalize(); ompi_output_finalize(); fclose( test_out ); test_finalize(); return(0); }