Add unit test for opal_db
This commit was SVN r30494.
Этот коммит содержится в:
Ralph Castain
родитель
4a8888d377
Коммит
ed3da20672
@ -1,4 +1,4 @@
|
||||
PROGS = no_op sigusr_trap spin orte_nodename orte_spawn orte_loop_spawn orte_loop_child orte_abort get_limits orte_ring spawn_child orte_tool orte_no_op binom oob_stress iof_stress iof_delay radix orte_barrier opal_interface orte_spin segfault orte_exit orte_db orte_sensor test-time event-threads psm_keygen regex orte_errors evpri-test opal-evpri-test evpri-test2 mapper reducer opal_hotel orte_dfs orte_allocate pmi_abort
|
||||
PROGS = no_op sigusr_trap spin orte_nodename orte_spawn orte_loop_spawn orte_loop_child orte_abort get_limits orte_ring spawn_child orte_tool orte_no_op binom oob_stress iof_stress iof_delay radix orte_barrier opal_interface orte_spin segfault orte_exit orte_db orte_sensor test-time event-threads psm_keygen regex orte_errors evpri-test opal-evpri-test evpri-test2 mapper reducer opal_hotel orte_dfs orte_allocate pmi_abort opal_db
|
||||
|
||||
all: $(PROGS)
|
||||
|
||||
|
||||
182
orte/test/system/opal_db.c
Обычный файл
182
orte/test/system/opal_db.c
Обычный файл
@ -0,0 +1,182 @@
|
||||
/* -*- C -*-
|
||||
*
|
||||
* $HEADER$
|
||||
*
|
||||
*/
|
||||
#include <stdio.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "opal/mca/base/base.h"
|
||||
#include "opal/util/cmd_line.h"
|
||||
#include "opal/util/opal_environ.h"
|
||||
|
||||
#include "orte/constants.h"
|
||||
|
||||
#include "opal/mca/db/base/base.h"
|
||||
#include "opal/runtime/opal.h"
|
||||
|
||||
static bool help;
|
||||
#define NKV 1024
|
||||
|
||||
static opal_cmd_line_init_t cmd_line_init[] = {
|
||||
{ NULL, 'h', NULL, "help", 0,
|
||||
&help, OPAL_CMD_LINE_TYPE_BOOL,
|
||||
"This help message" },
|
||||
|
||||
/* End of list */
|
||||
{ NULL, '\0', NULL, NULL, 0,
|
||||
NULL, OPAL_CMD_LINE_TYPE_NULL, NULL }
|
||||
};
|
||||
|
||||
int main(int argc, char* argv[])
|
||||
{
|
||||
int rc, i;
|
||||
opal_cmd_line_t cmd_line;
|
||||
opal_identifier_t uid;
|
||||
opal_value_t kvs;
|
||||
char *key;
|
||||
uint64_t data, *dptr;
|
||||
float *fval;
|
||||
opal_list_t kvlist;
|
||||
|
||||
opal_cmd_line_create(&cmd_line, cmd_line_init);
|
||||
mca_base_cmd_line_setup(&cmd_line);
|
||||
if (OPAL_SUCCESS != (rc = opal_cmd_line_parse(&cmd_line, true,
|
||||
argc, argv)) ) {
|
||||
if (OPAL_ERR_SILENT != rc) {
|
||||
fprintf(stderr, "%s: command line error (%d)\n", argv[0], rc);
|
||||
}
|
||||
return rc;
|
||||
}
|
||||
|
||||
/*
|
||||
* Since this process can now handle MCA/GMCA parameters, make sure to
|
||||
* process them.
|
||||
*/
|
||||
mca_base_cmd_line_process_args(&cmd_line, &environ, &environ);
|
||||
if (0 > (rc = opal_init(&argc, &argv))) {
|
||||
fprintf(stderr, "opal_db: couldn't init opal - error code %d\n", rc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
/* open and select the database framework */
|
||||
if (ORTE_SUCCESS != (rc = mca_base_framework_open(&opal_db_base_framework, 0))) {
|
||||
fprintf(stderr, "%s: db_framework_open failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
/* let the system pick any db plugin to test - will
|
||||
* be set by whatever param we were given on the cmd
|
||||
* line or environment
|
||||
*/
|
||||
if (ORTE_SUCCESS != (rc = opal_db_base_select(false))) {
|
||||
fprintf(stderr, "%s: db_framework_select failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
/* set our id */
|
||||
uid = 12345;
|
||||
opal_db.set_id((opal_identifier_t*)&uid);
|
||||
|
||||
/* create and store some arbitrary key-value pairs */
|
||||
for (i=0; i < NKV; i++) {
|
||||
asprintf(&key, "foo.%d", i);
|
||||
data = (uint64_t)i;
|
||||
if (ORTE_SUCCESS != (rc = opal_db.store((opal_identifier_t*)&uid, OPAL_SCOPE_GLOBAL,
|
||||
key, &data, OPAL_UINT64))) {
|
||||
fprintf(stderr, "%s: db_store failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
free(key);
|
||||
}
|
||||
|
||||
/* create another one */
|
||||
OBJ_CONSTRUCT(&kvs, opal_value_t);
|
||||
kvs.scope = OPAL_SCOPE_INTERNAL;
|
||||
kvs.key = strdup("pointer-value");
|
||||
kvs.type = OPAL_FLOAT;
|
||||
kvs.data.fval = 1.2334;
|
||||
|
||||
/* store a pointer to it */
|
||||
if (ORTE_SUCCESS != (rc = opal_db.store_pointer((opal_identifier_t*)&uid, &kvs))) {
|
||||
fprintf(stderr, "%s: db_store_pointer failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
/* commit the data - only has an effect if an external
|
||||
* database plugin was selected
|
||||
*/
|
||||
opal_db.commit((opal_identifier_t*)&uid);
|
||||
|
||||
/* fetch data and compare it to what was stored */
|
||||
dptr = &data;
|
||||
for (i=0; i < NKV; i++) {
|
||||
asprintf(&key, "foo.%d", i);
|
||||
if (ORTE_SUCCESS != (rc = opal_db.fetch((opal_identifier_t*)&uid, key, (void**)&dptr, OPAL_UINT64))) {
|
||||
fprintf(stderr, "%s: db_fetch failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
if (data != (uint64_t)i) {
|
||||
fprintf(stderr, "%s: db_fetch return incorrect data: %u vs %d\n", argv[0],
|
||||
data, i);
|
||||
return 1;
|
||||
}
|
||||
free(key);
|
||||
}
|
||||
|
||||
/* fetch a pointer to data and ensure the
|
||||
* pointer and its value is correct
|
||||
*/
|
||||
if (ORTE_SUCCESS != (rc = opal_db.fetch_pointer((opal_identifier_t*)&uid, "pointer-value", (void**)&fval, OPAL_FLOAT))) {
|
||||
fprintf(stderr, "%s: db_fetch_pointer failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
if (kvs.data.fval != (*fval)) {
|
||||
fprintf(stderr, "%s: db_fetch_pointer return incorrect data: %f vs %f\n", argv[0], *fval, kvs.data.fval);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* remove one key */
|
||||
if (ORTE_SUCCESS != (rc = opal_db.remove((opal_identifier_t*)&uid, "foo.0"))) {
|
||||
fprintf(stderr, "%s: db_remove failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
/* verify that it no longer is in the store */
|
||||
if (ORTE_SUCCESS == (rc = opal_db.fetch((opal_identifier_t*)&uid, "foo.0", (void**)&dptr, OPAL_UINT64))) {
|
||||
fprintf(stderr, "%s: db_fetch succeeded when it should have failed\n", argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* remove the pointer key */
|
||||
if (ORTE_SUCCESS != (rc = opal_db.remove((opal_identifier_t*)&uid, "pointer-value"))) {
|
||||
fprintf(stderr, "%s: db_remove failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
/* verify that it no longer is in the store */
|
||||
if (ORTE_SUCCESS == (rc = opal_db.fetch_pointer((opal_identifier_t*)&uid, "pointer-value", (void**)&fval, OPAL_FLOAT))) {
|
||||
fprintf(stderr, "%s: db_fetch_pointer succeeded when it should have failed\n", argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
/* remove all keys */
|
||||
if (ORTE_SUCCESS != (rc = opal_db.remove((opal_identifier_t*)&uid, NULL))) {
|
||||
fprintf(stderr, "%s: db_remove failed (%d)\n", argv[0], rc);
|
||||
return rc;
|
||||
}
|
||||
|
||||
/* verify that nothing is in the store */
|
||||
OBJ_CONSTRUCT(&kvlist, opal_list_t);
|
||||
if (ORTE_SUCCESS != (rc = opal_db.fetch_multiple((opal_identifier_t*)&uid, OPAL_SCOPE_ALL,
|
||||
NULL, &kvlist))) {
|
||||
fprintf(stderr, "%s: db_fetch_multiple failed\n", argv[0]);
|
||||
return 1;
|
||||
}
|
||||
if (0 < opal_list_get_size(&kvlist)) {
|
||||
fprintf(stderr, "%s: db_fetch_multiple returned values when the db should have been empty\n", argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
fprintf(stderr, "%s: SUCCESS\n", argv[0]);
|
||||
opal_finalize();
|
||||
return 0;
|
||||
}
|
||||
Загрузка…
Ссылка в новой задаче
Block a user