251 строка
7.1 KiB
C
251 строка
7.1 KiB
C
/*
|
|
* unit test for General Purpose Registry replica.
|
|
|
|
--------------------------------------------------------------------------
|
|
|
|
Authors: Ralph H. Castain <rhc@lanl.gov>
|
|
|
|
--------------------------------------------------------------------------
|
|
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
#include <stdio.h>
|
|
#include <string.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/gpr/base/base.h"
|
|
#include "mca/gpr/replica/gpr_replica.h"
|
|
#include "mca/gpr/replica/gpr_replica_internals.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)
|
|
{
|
|
mca_gpr_replica_key_t test_key, test_key2;
|
|
ompi_list_t *test_list, *internal_tests;
|
|
ompi_registry_index_value_t *ptr;
|
|
ompi_registry_internal_test_results_t *ptri;
|
|
ompi_registry_object_t test_buffer;
|
|
uint8_t *test_buf;
|
|
ompi_registry_object_size_t input_size;
|
|
ompi_registry_mode_t mode;
|
|
ompi_list_t *answer;
|
|
ompi_registry_value_t *ans;
|
|
bool multi, hidden;
|
|
int i, j, response;
|
|
bool success;
|
|
char *tmp;
|
|
char name[30], *name2[30], *name3[30];
|
|
int result, put_test; /* result from system call */
|
|
|
|
test_init("test_gpr_replica");
|
|
|
|
test_out = fopen( "test_gpr_replica_out", "w+" );
|
|
if( test_out == NULL ) {
|
|
test_failure("test_gpr_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);
|
|
}
|
|
|
|
fprintf(stderr, "opening gpr base\n");
|
|
/* open the GPR */
|
|
if (OMPI_SUCCESS == mca_gpr_base_open()) {
|
|
fprintf(test_out, "GPR opened\n");
|
|
test_success();
|
|
} else {
|
|
fprintf(test_out, "GPR could not open\n");
|
|
test_failure("test_gpr_replica mca_gpr_base_open failed");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
fprintf(stderr, "selecting gpr module\n");
|
|
/* startup the GPR replica */
|
|
if (OMPI_SUCCESS != mca_gpr_base_select(&multi, &hidden)) {
|
|
fprintf(test_out, "GPR replica could not start\n");
|
|
test_failure("test_gpr_replica mca_gpr_base_select failed");
|
|
test_finalize();
|
|
exit(1);
|
|
} else {
|
|
fprintf(test_out, "GPR replica started\n");
|
|
test_success();
|
|
}
|
|
|
|
fprintf(stderr, "testing get index\n");
|
|
/* check index */
|
|
test_list = ompi_registry.index(NULL);
|
|
if (0 == ompi_list_get_size(test_list)) { /* should have been something in dictionary */
|
|
fprintf(test_out, "GPR replica: index function failed\n");
|
|
test_failure("test_gpr_replica index_global_dictionary failed\n");
|
|
test_finalize();
|
|
exit(1);
|
|
} else {
|
|
fprintf(test_out, "GPR index returned list\n");
|
|
for (ptr = (ompi_registry_index_value_t*)ompi_list_get_first(test_list);
|
|
ptr != (ompi_registry_index_value_t*)ompi_list_get_end(test_list);
|
|
ptr = (ompi_registry_index_value_t*)ompi_list_get_next(ptr)) {
|
|
fprintf(test_out, "\t%s\n", ptr->token);
|
|
}
|
|
test_success();
|
|
}
|
|
|
|
fprintf(stderr, "testing internals\n");
|
|
/* check internals */
|
|
internal_tests = ompi_registry.test_internals(1);
|
|
if (0 == ompi_list_get_size(internal_tests)) { /* should have been something in list */
|
|
fprintf(test_out, "internal tests failed\n");
|
|
test_failure("test_gpr_replica internal_tests failed\n");
|
|
test_finalize();
|
|
exit(1);
|
|
} else {
|
|
fprintf(test_out, "internal test results list\n");
|
|
for (ptri = (ompi_registry_internal_test_results_t*)ompi_list_get_first(internal_tests);
|
|
ptri != (ompi_registry_internal_test_results_t*)ompi_list_get_end(internal_tests);
|
|
ptri = (ompi_registry_internal_test_results_t*)ompi_list_get_next(ptri)) {
|
|
fprintf(test_out, "\t%s\n", ptri->test);
|
|
fprintf(test_out, "\t%s\n", ptri->message);
|
|
}
|
|
test_success();
|
|
}
|
|
|
|
|
|
fprintf(stderr, "testing put\n");
|
|
/* test the put function */
|
|
success = true;
|
|
input_size = 10000;
|
|
test_buffer = (ompi_registry_object_t)malloc(input_size);
|
|
test_buf = (uint8_t*)test_buffer;
|
|
for (i=0; i<input_size; i++) {
|
|
*test_buf = i % 256;
|
|
test_buf++;
|
|
}
|
|
for (j=0; j<10; j++) {
|
|
asprintf(&name2[j], "test-key%d", j);
|
|
name3[j] = strdup(name2[j]);
|
|
}
|
|
name2[j] = NULL;
|
|
for (j=10; j<20; j++) {
|
|
asprintf(&name3[j], "dummy-key%d", j);
|
|
}
|
|
name3[j] = NULL;
|
|
|
|
for (i=0; i<5 && success; i++) {
|
|
sprintf(name, "test-def-seg%d", i);
|
|
fprintf(stderr, "testing seg %d\n", i);
|
|
if (OMPI_SUCCESS != ompi_registry.put(OMPI_REGISTRY_NONE, name,
|
|
name2, test_buffer, input_size)) {
|
|
fprintf(test_out, "put test failed for segment %s\n", name);
|
|
for (j=0; j<10; j++) {
|
|
fprintf(test_out, "\t%s\n", name2[j]);
|
|
}
|
|
success = false;
|
|
}
|
|
}
|
|
if (success) {
|
|
fprintf(test_out, "put test: success\n");
|
|
test_success();
|
|
} else {
|
|
fprintf(test_out, "put test failed\n");
|
|
test_failure("test_gpr_replica put_test failed\n");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
/* test the put overwrite function */
|
|
success = true;
|
|
for (i=0; i<5 && success; i++) {
|
|
sprintf(name, "test-def-seg%d", i);
|
|
if (10 % i) {
|
|
mode = OMPI_REGISTRY_OVERWRITE;
|
|
} else {
|
|
mode = OMPI_REGISTRY_NONE;
|
|
}
|
|
put_test = ompi_registry.put(mode, name, name2, test_buffer, input_size);
|
|
if ((OMPI_REGISTRY_OVERWRITE == mode && OMPI_SUCCESS != put_test) ||
|
|
(OMPI_REGISTRY_NONE == mode && OMPI_SUCCESS == put_test)) {
|
|
fprintf(test_out, "put test failed for segment %s\n", name);
|
|
for (j=0; j<10; j++) {
|
|
fprintf(test_out, "\t%s\n", name2[j]);
|
|
}
|
|
success = false;
|
|
}
|
|
}
|
|
if (success) {
|
|
fprintf(test_out, "put overwrite test: success\n");
|
|
test_success();
|
|
} else {
|
|
fprintf(test_out, "put overwrite test failed\n");
|
|
test_failure("test_gpr_replica put_test failed\n");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
fprintf(stderr, "testing get\n");
|
|
/* test the get function */
|
|
success = true;
|
|
for (i=0; i<5 && success; i++) {
|
|
sprintf(name, "test-def-seg%d", i);
|
|
if (10 % i) {
|
|
mode = OMPI_REGISTRY_AND;
|
|
answer = ompi_registry.get(mode, name, name2);
|
|
for (ans = (ompi_registry_value_t*)ompi_list_get_first(answer);
|
|
ans != (ompi_registry_value_t*)ompi_list_get_end(answer);
|
|
ans = (ompi_registry_value_t*)ompi_list_get_next(ans)) {
|
|
if (ans->object_size != input_size) {
|
|
success = false;
|
|
}
|
|
}
|
|
} else {
|
|
mode = OMPI_REGISTRY_XAND;
|
|
answer = ompi_registry.get(mode, name, name3);
|
|
if (0 < ompi_list_get_size(answer)) { /* should not have gotten a result */
|
|
success = false;
|
|
}
|
|
}
|
|
}
|
|
if (success) {
|
|
fprintf(test_out, "get test: success\n");
|
|
test_success();
|
|
} else {
|
|
fprintf(test_out, "get test failed\n");
|
|
test_failure("test_gpr_replica get_test failed\n");
|
|
test_finalize();
|
|
exit(1);
|
|
}
|
|
|
|
/* check the universe segment - should have a key value of "1" */
|
|
|
|
fclose( test_out );
|
|
/* result = system( cmd_str );
|
|
if( result == 0 ) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure( "test_gpr_replica ompi_registry init, etc failed");
|
|
}
|
|
*/
|
|
test_finalize();
|
|
|
|
return(0);
|
|
}
|