897983ea30
* Update a bunch of the unit tests to either be disabled (someone who isn't me and knows that code needs to fix them) or to work properly This commit was SVN r4986.
376 строки
13 KiB
C
376 строки
13 KiB
C
/*
|
|
* Copyright (c) 2004-2005 The Trustees of Indiana University.
|
|
* All rights reserved.
|
|
* Copyright (c) 2004-2005 The Trustees of the University of Tennessee.
|
|
* All rights reserved.
|
|
* Copyright (c) 2004-2005 High Performance Computing Center Stuttgart,
|
|
* University of Stuttgart. All rights reserved.
|
|
* $COPYRIGHT$
|
|
*
|
|
* Additional copyrights may follow
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
/** @file:
|
|
*
|
|
* The Open MPI general purpose registry - unit test
|
|
*
|
|
*/
|
|
|
|
/*
|
|
* includes
|
|
*/
|
|
|
|
#include "orte_config.h"
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "include/orte_constants.h"
|
|
#include "include/orte_schema.h"
|
|
|
|
#include "support.h"
|
|
|
|
#include "class/orte_pointer_array.h"
|
|
#include "dps/dps.h"
|
|
#include "runtime/runtime.h"
|
|
#include "util/proc_info.h"
|
|
#include "util/sys_info.h"
|
|
|
|
#include "mca/gpr/base/base.h"
|
|
#include "mca/gpr/replica/api_layer/gpr_replica_api.h"
|
|
#include "mca/gpr/replica/functional_layer/gpr_replica_fn.h"
|
|
#include "mca/gpr/replica/communications/gpr_replica_comm.h"
|
|
#include "mca/gpr/replica/transition_layer/gpr_replica_tl.h"
|
|
|
|
/* output files needed by the test */
|
|
static FILE *test_out=NULL;
|
|
|
|
static char *cmd_str="diff ./test_gpr_replica_out ./test_gpr_replica_out_std";
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int rc, num_names, num_found;
|
|
int32_t i, j, cnt;
|
|
char *tmp=NULL, *tmp2=NULL, *names[15], *keys[5];
|
|
orte_gpr_replica_segment_t *seg=NULL;
|
|
orte_gpr_replica_itag_t itag[10], itag2, *itaglist;
|
|
orte_gpr_replica_container_t *cptr=NULL, **cptrs=NULL;
|
|
orte_gpr_keyval_t *kptr=NULL, **kvals;
|
|
orte_gpr_replica_itagval_t **ivals=NULL, *iptr;
|
|
orte_gpr_value_t **values, *val;
|
|
orte_process_name_t seed={0,0,0};
|
|
bool found;
|
|
|
|
test_init("test_gpr_replica");
|
|
|
|
if (getenv("TEST_WRITE_TO_FILE") != NULL) {
|
|
test_out = fopen( "test_gpr_replica_out", "w+" );
|
|
} else {
|
|
test_out = stderr;
|
|
}
|
|
if( test_out == NULL ) {
|
|
test_failure("gpr_test couldn't open test file failed");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
/* ENSURE THE REPLICA IS ISOLATED */
|
|
setenv("OMPI_MCA_gpr_replica_isolate", "1", 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();
|
|
|
|
/* Ensure the system_info structure is instantiated and initialized */
|
|
if (ORTE_SUCCESS != (rc = orte_sys_info())) {
|
|
return rc;
|
|
}
|
|
|
|
/* Ensure the process info structure is instantiated and initialized */
|
|
if (ORTE_SUCCESS != (rc = orte_proc_info())) {
|
|
return rc;
|
|
}
|
|
|
|
|
|
orte_process_info.seed = true;
|
|
orte_process_info.my_name = &seed;
|
|
|
|
/* startup the MCA */
|
|
if (OMPI_SUCCESS == mca_base_open()) {
|
|
fprintf(test_out, "MCA started\n");
|
|
} else {
|
|
fprintf(test_out, "MCA could not start\n");
|
|
exit (1);
|
|
}
|
|
|
|
if (ORTE_SUCCESS == orte_gpr_base_open()) {
|
|
fprintf(test_out, "GPR started\n");
|
|
} else {
|
|
fprintf(test_out, "GPR could not start\n");
|
|
exit (1);
|
|
}
|
|
|
|
if (ORTE_SUCCESS == orte_gpr_base_select()) {
|
|
fprintf(test_out, "GPR replica selected\n");
|
|
} else {
|
|
fprintf(test_out, "GPR replica could not be selected\n");
|
|
exit (1);
|
|
}
|
|
|
|
if (ORTE_SUCCESS == orte_dps_open()) {
|
|
fprintf(test_out, "DPS started\n");
|
|
} else {
|
|
fprintf(test_out, "DPS could not start\n");
|
|
exit (1);
|
|
}
|
|
|
|
fprintf(stderr, "put one value with single keyval\n");
|
|
val = OBJ_NEW(orte_gpr_value_t);
|
|
val->addr_mode = ORTE_GPR_NO_OVERWRITE | ORTE_GPR_TOKENS_XAND;
|
|
val->cnt = 1;
|
|
val->segment = strdup("test-put-segment");
|
|
val->num_tokens = 14;
|
|
val->tokens = (char**)malloc(val->num_tokens * sizeof(char*));
|
|
for (i=0; i < 14; i++) {
|
|
asprintf(&(val->tokens[i]), "dummy%d", i);
|
|
}
|
|
val->keyvals = (orte_gpr_keyval_t**)malloc(sizeof(orte_gpr_keyval_t*));
|
|
val->keyvals[0] = OBJ_NEW(orte_gpr_keyval_t);
|
|
(val->keyvals[0])->key = strdup("stupid-value-next-one");
|
|
(val->keyvals[0])->type = ORTE_INT32;
|
|
(val->keyvals[0])->value.i32 = 654321;
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_put(1, &val))) {
|
|
fprintf(test_out, "gpr_test: put of 1 value/1 keyval failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: put of 1 value/1 keyval failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: put of 1 value/1 keyval passed\n");
|
|
}
|
|
OBJ_RELEASE(val);
|
|
|
|
fprintf(stderr, "put one value with multiple keyvals\n");
|
|
val = OBJ_NEW(orte_gpr_value_t);
|
|
val->addr_mode = ORTE_GPR_NO_OVERWRITE | ORTE_GPR_TOKENS_XAND;
|
|
val->cnt = 20;
|
|
val->segment = strdup("test-put-segment");
|
|
val->num_tokens = 14;
|
|
val->tokens = (char**)malloc(val->num_tokens * sizeof(char*));
|
|
for (i=0; i < 14; i++) {
|
|
asprintf(&(val->tokens[i]), "dummy%d", i);
|
|
}
|
|
val->keyvals = (orte_gpr_keyval_t**)malloc(20*sizeof(orte_gpr_keyval_t*));
|
|
for (i=0; i<20; i++) {
|
|
val->keyvals[i] = OBJ_NEW(orte_gpr_keyval_t);
|
|
asprintf(&((val->keyvals[i])->key), "stupid-test-%d", i);
|
|
(val->keyvals[i])->type = ORTE_UINT32;
|
|
(val->keyvals[i])->value.ui32 = (uint32_t)i;
|
|
}
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_put(1, &val))) {
|
|
fprintf(test_out, "gpr_test: put 1 value/multiple keyval failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: put 1 value/multiple keyval failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: put 1 value/multiple keyval passed\n");
|
|
}
|
|
|
|
fprintf(stderr, "put 1 value/multiple keyvals - second container\n");
|
|
val->num_tokens = 10;
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_put(1, &val))) {
|
|
fprintf(test_out, "gpr_test: put 1 value/multiple keyval in second container failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: put 1 value/multiple keyval in second container failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: put 1 value/multiple keyval in second container passed\n");
|
|
}
|
|
OBJ_RELEASE(val);
|
|
|
|
fprintf(stderr, "dump\n");
|
|
if (ORTE_SUCCESS != (rc = orte_gpr.dump_all(0))) {
|
|
fprintf(test_out, "gpr_test: dump failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: dump failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: dump passed\n");
|
|
}
|
|
|
|
fprintf(stderr, "get\n");
|
|
names[0] = strdup("dummy0");
|
|
names[1] = NULL;
|
|
keys[0] = strdup("stupid-test-1");
|
|
keys[1] = NULL;
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_get(ORTE_GPR_KEYS_OR | ORTE_GPR_TOKENS_OR,
|
|
"test-put-segment",
|
|
names, keys,
|
|
&cnt, &values))) {
|
|
fprintf(test_out, "gpr_test: get failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: get failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: get passed\n");
|
|
}
|
|
|
|
fprintf(stderr, "get results:\n");
|
|
for (j=0; j < cnt; j++) {
|
|
fprintf(stderr, "value %d: cnt %d\t segment %s num_tokens %d\n", j,
|
|
values[j]->cnt,
|
|
values[j]->segment, values[j]->num_tokens);
|
|
for (i=0; i < values[j]->num_tokens; i++) {
|
|
fprintf(stderr, "token: %d %s\n", i, values[j]->tokens[i]);
|
|
}
|
|
kvals = values[j]->keyvals;
|
|
for (i=0; i < values[j]->cnt; i++) {
|
|
fprintf(stderr, "\tkey %s type %d\n", kvals[i]->key, kvals[i]->type);
|
|
}
|
|
OBJ_RELEASE(values[j]);
|
|
}
|
|
free(values);
|
|
|
|
fprintf(stderr, "get multiple in one segment, multiple containers\n");
|
|
keys[0] = strdup("stupid-test-1");
|
|
keys[1] = strdup("stupid-test-3");
|
|
keys[2] = strdup("stupid-test-5");
|
|
keys[3] = strdup("stupid-test-8");
|
|
keys[4] = NULL;
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_get(ORTE_GPR_KEYS_OR | ORTE_GPR_TOKENS_OR,
|
|
"test-put-segment",
|
|
names, keys,
|
|
&cnt, &values))) {
|
|
fprintf(test_out, "gpr_test: get failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: get failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: get passed\n");
|
|
}
|
|
|
|
fprintf(stderr, "get results:\n");
|
|
for (j=0; j < cnt; j++) {
|
|
fprintf(stderr, "value %d: cnt %d\t segment %s num_tokens %d\n", j,
|
|
values[j]->cnt,
|
|
values[j]->segment, values[j]->num_tokens);
|
|
for (i=0; i < values[j]->num_tokens; i++) {
|
|
fprintf(stderr, "token: %d %s\n", i, values[j]->tokens[i]);
|
|
}
|
|
kvals = values[j]->keyvals;
|
|
for (i=0; i < values[j]->cnt; i++) {
|
|
fprintf(stderr, "\tkey %s type %d\n", kvals[i]->key, kvals[i]->type);
|
|
}
|
|
OBJ_RELEASE(values[j]);
|
|
}
|
|
free(values);
|
|
|
|
fprintf(stderr, "put multiple copies of same entry in single container\n");
|
|
val = OBJ_NEW(orte_gpr_value_t);
|
|
val->addr_mode = ORTE_GPR_NO_OVERWRITE | ORTE_GPR_TOKENS_XAND;
|
|
val->cnt = 1;
|
|
val->segment = strdup("test-put-segment");
|
|
val->num_tokens = 5;
|
|
val->tokens = (char**)malloc(val->num_tokens * sizeof(char*));
|
|
for (i=0; i < 5; i++) {
|
|
asprintf(&(val->tokens[i]), "multi-dum-dum-%d", i);
|
|
}
|
|
val->keyvals = (orte_gpr_keyval_t**)malloc(sizeof(orte_gpr_keyval_t*));
|
|
val->keyvals[0] = OBJ_NEW(orte_gpr_keyval_t);
|
|
(val->keyvals[0])->key = strdup("stupid-value-next-one");
|
|
(val->keyvals[0])->type = ORTE_STRING;
|
|
(val->keyvals[0])->value.strptr = strdup("try-string-value");
|
|
for (i = 0; i < 10; i++) {
|
|
fprintf(stderr, "\tputting copy %d\n", i);
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_put(1, &val))) {
|
|
fprintf(test_out, "gpr_test: put multiple copies of one keyval in a container failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: put multiple copies of one keyval in a container failed");
|
|
test_finalize();
|
|
return rc;
|
|
}
|
|
}
|
|
OBJ_RELEASE(val);
|
|
|
|
orte_gpr.dump_all(0);
|
|
|
|
fprintf(stderr, "update multiple keyvals in a container\n");
|
|
if(ORTE_SUCCESS != orte_gpr_replica_find_seg(&seg, false, "test-put-segment")) {
|
|
return -1;
|
|
}
|
|
|
|
cptrs = (orte_gpr_replica_container_t**)((seg->containers)->addr);
|
|
for (i=0; i < (seg->containers)->size; i++) {
|
|
if (NULL != cptrs[i]) {
|
|
cptr = cptrs[i];
|
|
}
|
|
}
|
|
|
|
kptr = OBJ_NEW(orte_gpr_keyval_t);
|
|
kptr->key = strdup("really-stupid-value");
|
|
kptr->type = ORTE_INT32;
|
|
kptr->value.i32 = 123456;
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_update_keyval(seg, cptr, kptr))) {
|
|
fprintf(test_out, "gpr_test: update multiple keyvals failed with error code %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: update multiple keyvals failed");
|
|
test_finalize();
|
|
return rc;
|
|
} else {
|
|
fprintf(test_out, "gpr_test: update multiple keyvals passed\n");
|
|
}
|
|
|
|
orte_gpr.dump_all(0);
|
|
|
|
fprintf(stderr, "put with no tokens puts in every container\n");
|
|
val = OBJ_NEW(orte_gpr_value_t);
|
|
val->addr_mode = ORTE_GPR_NO_OVERWRITE;
|
|
val->cnt = 1;
|
|
val->segment = strdup("test-put-segment");
|
|
val->num_tokens = 0;
|
|
val->tokens = NULL;
|
|
val->keyvals = (orte_gpr_keyval_t**)malloc(sizeof(orte_gpr_keyval_t*));
|
|
val->keyvals[0] = OBJ_NEW(orte_gpr_keyval_t);
|
|
(val->keyvals[0])->key = strdup("stupid-value-next-one");
|
|
(val->keyvals[0])->type = ORTE_STRING;
|
|
(val->keyvals[0])->value.strptr = strdup("try-string-value");
|
|
if (ORTE_SUCCESS != (rc = orte_gpr_replica_put(1, &val))) {
|
|
fprintf(test_out, "gpr_test: put with no tokens failed - returned %s\n",
|
|
ORTE_ERROR_NAME(rc));
|
|
test_failure("gpr_test: put with no tokens failed");
|
|
test_finalize();
|
|
} else {
|
|
fprintf(test_out, "gpr_test: put with no tokens passed\n");
|
|
}
|
|
OBJ_RELEASE(val);
|
|
|
|
orte_gpr.dump_all(0);
|
|
|
|
fclose( test_out );
|
|
/* result = system( cmd_str );
|
|
if( result == 0 ) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure( "test_gpr_replica failed");
|
|
}
|
|
*/
|
|
test_finalize();
|
|
|
|
return(0);
|
|
}
|