257 строки
8.9 KiB
C
257 строки
8.9 KiB
C
/*
|
|
* unit test for name server replica.
|
|
|
|
--------------------------------------------------------------------------
|
|
|
|
Authors: Ralph H. Castain <rhc@lanl.gov>
|
|
|
|
--------------------------------------------------------------------------
|
|
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#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);
|
|
}
|