107 строки
3.1 KiB
C
107 строки
3.1 KiB
C
/* -*- C -*-
|
|
*
|
|
* $HEADER$
|
|
*/
|
|
|
|
#include "ompi_config.h"
|
|
#include "support.h"
|
|
|
|
#include "class/ompi_object.h"
|
|
#include "class/ompi_list.h"
|
|
#include "mca/llm/base/base.h"
|
|
#include "mca/llm/base/base_internal.h"
|
|
|
|
static char *cmd1_str="diff ./test1_out ./test1_out_std";
|
|
static char *cmd2_str="diff ./test2_out ./test2_out_std";
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
ompi_list_t *hostlist;
|
|
ompi_rte_node_allocation_t *node;
|
|
ompi_rte_valuepair_t *valpair;
|
|
ompi_list_item_t *nodeitem, *valpairitem;
|
|
FILE *test1_out=NULL; /* output file for first test */
|
|
FILE *test2_out=NULL; /* output file for second test */
|
|
int result; /* result of system call */
|
|
|
|
test_init("parse_hostfile_t");
|
|
|
|
/* Open output files for the tests */
|
|
test1_out = fopen("./test1_out", "w+" );
|
|
if( test1_out == NULL ) {
|
|
printf("can't open test 1 output file\n");
|
|
exit(-1);
|
|
}
|
|
test2_out = fopen("./test2_out", "w+" );
|
|
if( test2_out == NULL ) {
|
|
printf("can't open test 2 output file\n");
|
|
exit(-1);
|
|
}
|
|
|
|
/* test 1 - test base_parse_hostfile */
|
|
hostlist = mca_llm_base_parse_hostfile("testfile");
|
|
assert(hostlist != NULL);
|
|
|
|
fprintf(test1_out,"Original hostfile\n");
|
|
|
|
for (nodeitem = ompi_list_get_first(hostlist);
|
|
nodeitem != ompi_list_get_end(hostlist);
|
|
nodeitem = ompi_list_get_next(nodeitem)) {
|
|
|
|
node = (ompi_rte_node_allocation_t*) nodeitem;
|
|
fprintf(test1_out, "\t%s %d\n", node->hostname, node->count);
|
|
|
|
for (valpairitem = ompi_list_get_first(&(node->info));
|
|
valpairitem != ompi_list_get_end(&(node->info));
|
|
valpairitem = ompi_list_get_next(valpairitem)) {
|
|
valpair = (ompi_rte_valuepair_t*) valpairitem;
|
|
fprintf(test1_out, "\t\t%s = %s\n", valpair->key, valpair->value);
|
|
}
|
|
}
|
|
fclose( test1_out );
|
|
|
|
/* See if the file matches the test standard */
|
|
result = system( cmd1_str );
|
|
if( result == 0 ) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure( "parse_hostfile test1 failed" );
|
|
}
|
|
|
|
/* test 2 */
|
|
mca_llm_base_collapse_resources(hostlist);
|
|
fprintf(test2_out, "Compressed hostfile\n");
|
|
|
|
for (nodeitem = ompi_list_get_first(hostlist);
|
|
nodeitem != ompi_list_get_end(hostlist);
|
|
nodeitem = ompi_list_get_next(nodeitem)) {
|
|
|
|
node = (ompi_rte_node_allocation_t*) nodeitem;
|
|
fprintf(test2_out, "\t%s %d\n", node->hostname, node->count);
|
|
|
|
for (valpairitem = ompi_list_get_first(&(node->info));
|
|
valpairitem != ompi_list_get_end(&(node->info));
|
|
valpairitem = ompi_list_get_next(valpairitem)) {
|
|
valpair = (ompi_rte_valuepair_t*) valpairitem;
|
|
fprintf(test2_out, "\t\t%s = %s\n", valpair->key, valpair->value);
|
|
}
|
|
}
|
|
fclose( test2_out );
|
|
|
|
/* See if the file matches the test standard */
|
|
result = system( cmd2_str );
|
|
if( result == 0 ) {
|
|
test_success();
|
|
}
|
|
else {
|
|
test_failure( "parse_hostfile test2 failed" );
|
|
}
|
|
|
|
OBJ_RELEASE(hostlist);
|
|
|
|
test_finalize();
|
|
return 0;
|
|
}
|