Formatted all code using indent
This commit is contained in:
Patrick J Cherry
@@ -6,366 +6,331 @@
|
||||
|
||||
#include <check.h>
|
||||
|
||||
struct server* mock_server(void)
|
||||
struct server *mock_server(void)
|
||||
{
|
||||
struct server* out = xmalloc( sizeof( struct server ) );
|
||||
out->l_start_mirror = flexthread_mutex_create();
|
||||
out->nbd_client = xmalloc( sizeof( struct client_tbl_entry ) * 4 );
|
||||
out->max_nbd_clients = 4;
|
||||
out->size = 65536;
|
||||
struct server *out = xmalloc(sizeof(struct server));
|
||||
out->l_start_mirror = flexthread_mutex_create();
|
||||
out->nbd_client = xmalloc(sizeof(struct client_tbl_entry) * 4);
|
||||
out->max_nbd_clients = 4;
|
||||
out->size = 65536;
|
||||
|
||||
out->allocation_map = bitset_alloc( 65536, 4096 );
|
||||
out->allocation_map = bitset_alloc(65536, 4096);
|
||||
|
||||
return out;
|
||||
return out;
|
||||
}
|
||||
|
||||
struct server* mock_mirroring_server(void)
|
||||
struct server *mock_mirroring_server(void)
|
||||
{
|
||||
struct server *out = mock_server();
|
||||
out->mirror = xmalloc( sizeof( struct mirror ) );
|
||||
out->mirror_super = xmalloc( sizeof( struct mirror_super ) );
|
||||
return out;
|
||||
struct server *out = mock_server();
|
||||
out->mirror = xmalloc(sizeof(struct mirror));
|
||||
out->mirror_super = xmalloc(sizeof(struct mirror_super));
|
||||
return out;
|
||||
}
|
||||
|
||||
void destroy_mock_server( struct server* serve )
|
||||
void destroy_mock_server(struct server *serve)
|
||||
{
|
||||
if ( NULL != serve->mirror ) {
|
||||
free( serve->mirror );
|
||||
}
|
||||
if (NULL != serve->mirror) {
|
||||
free(serve->mirror);
|
||||
}
|
||||
|
||||
if ( NULL != serve->mirror_super ) {
|
||||
free( serve->mirror_super );
|
||||
}
|
||||
if (NULL != serve->mirror_super) {
|
||||
free(serve->mirror_super);
|
||||
}
|
||||
|
||||
flexthread_mutex_destroy( serve->l_start_mirror );
|
||||
flexthread_mutex_destroy(serve->l_start_mirror);
|
||||
|
||||
bitset_free( serve->allocation_map );
|
||||
free( serve->nbd_client );
|
||||
free( serve );
|
||||
bitset_free(serve->allocation_map);
|
||||
free(serve->nbd_client);
|
||||
free(serve);
|
||||
}
|
||||
|
||||
START_TEST( test_status_create )
|
||||
START_TEST(test_status_create)
|
||||
{
|
||||
struct server * server = mock_server();
|
||||
struct status * status = status_create( server );
|
||||
struct server *server = mock_server();
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_if( NULL == status, "Status wasn't allocated" );
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
fail_if(NULL == status, "Status wasn't allocated");
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_gets_has_control )
|
||||
END_TEST START_TEST(test_gets_has_control)
|
||||
{
|
||||
struct server * server = mock_server();
|
||||
server->success = 1;
|
||||
struct server *server = mock_server();
|
||||
server->success = 1;
|
||||
|
||||
struct status * status = status_create( server );
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_unless( status->has_control == 1, "has_control wasn't copied" );
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
fail_unless(status->has_control == 1, "has_control wasn't copied");
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
}
|
||||
END_TEST
|
||||
|
||||
|
||||
START_TEST( test_gets_is_mirroring )
|
||||
END_TEST START_TEST(test_gets_is_mirroring)
|
||||
{
|
||||
struct server * server = mock_server();
|
||||
struct status * status = status_create( server );
|
||||
struct server *server = mock_server();
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_if( status->is_mirroring, "is_mirroring was set" );
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
fail_if(status->is_mirroring, "is_mirroring was set");
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
|
||||
server = mock_mirroring_server();
|
||||
status = status_create( server );
|
||||
server = mock_mirroring_server();
|
||||
status = status_create(server);
|
||||
|
||||
fail_unless( status->is_mirroring, "is_mirroring wasn't set" );
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
fail_unless(status->is_mirroring, "is_mirroring wasn't set");
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_gets_clients_allowed )
|
||||
END_TEST START_TEST(test_gets_clients_allowed)
|
||||
{
|
||||
struct server * server = mock_server();
|
||||
struct status * status = status_create( server );
|
||||
struct server *server = mock_server();
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_if( status->clients_allowed, "clients_allowed was set" );
|
||||
status_destroy( status );
|
||||
fail_if(status->clients_allowed, "clients_allowed was set");
|
||||
status_destroy(status);
|
||||
|
||||
server->allow_new_clients = 1;
|
||||
status = status_create( server );
|
||||
server->allow_new_clients = 1;
|
||||
status = status_create(server);
|
||||
|
||||
fail_unless( status->clients_allowed, "clients_allowed was not set" );
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
fail_unless(status->clients_allowed, "clients_allowed was not set");
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_gets_pid )
|
||||
END_TEST START_TEST(test_gets_pid)
|
||||
{
|
||||
struct server * server = mock_server();
|
||||
struct status * status = status_create( server );
|
||||
struct server *server = mock_server();
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_unless( getpid() == status->pid, "Pid wasn't gathered" );
|
||||
fail_unless(getpid() == status->pid, "Pid wasn't gathered");
|
||||
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_gets_size )
|
||||
END_TEST START_TEST(test_gets_size)
|
||||
{
|
||||
struct server * server = mock_server();
|
||||
server->size = 1024;
|
||||
struct server *server = mock_server();
|
||||
server->size = 1024;
|
||||
|
||||
struct status * status = status_create( server );
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_unless( 1024 == status->size, "Size wasn't gathered" );
|
||||
fail_unless(1024 == status->size, "Size wasn't gathered");
|
||||
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_gets_migration_statistics )
|
||||
END_TEST START_TEST(test_gets_migration_statistics)
|
||||
{
|
||||
struct server * server = mock_mirroring_server();
|
||||
server->mirror->all_dirty = 16384;
|
||||
server->mirror->max_bytes_per_second = 32768;
|
||||
server->mirror->offset = 0;
|
||||
struct server *server = mock_mirroring_server();
|
||||
server->mirror->all_dirty = 16384;
|
||||
server->mirror->max_bytes_per_second = 32768;
|
||||
server->mirror->offset = 0;
|
||||
|
||||
/* we have a bit of a time dependency here */
|
||||
server->mirror->migration_started = monotonic_time_ms();
|
||||
/* we have a bit of a time dependency here */
|
||||
server->mirror->migration_started = monotonic_time_ms();
|
||||
|
||||
struct status * status = status_create( server );
|
||||
struct status *status = status_create(server);
|
||||
|
||||
fail_unless (
|
||||
0 == status->migration_duration ||
|
||||
fail_unless(0 == status->migration_duration ||
|
||||
1 == status->migration_duration ||
|
||||
2 == status->migration_duration,
|
||||
"migration_duration is unreasonable!"
|
||||
);
|
||||
"migration_duration is unreasonable!");
|
||||
|
||||
fail_unless(
|
||||
16384 / ( status->migration_duration + 1 ) == status->migration_speed,
|
||||
"migration_speed not calculated correctly"
|
||||
);
|
||||
fail_unless(16384 / (status->migration_duration + 1) ==
|
||||
status->migration_speed,
|
||||
"migration_speed not calculated correctly");
|
||||
|
||||
fail_unless( 32768 == status->migration_speed_limit, "migration_speed_limit not read" );
|
||||
fail_unless(32768 == status->migration_speed_limit,
|
||||
"migration_speed_limit not read");
|
||||
|
||||
// ( size / current_bps ) + 1 happens to be 3 for this test
|
||||
fail_unless( 3 == status->migration_seconds_left, "migration_seconds_left not gathered" );
|
||||
// ( size / current_bps ) + 1 happens to be 3 for this test
|
||||
fail_unless(3 == status->migration_seconds_left,
|
||||
"migration_seconds_left not gathered");
|
||||
|
||||
status_destroy( status );
|
||||
destroy_mock_server( server );
|
||||
status_destroy(status);
|
||||
destroy_mock_server(server);
|
||||
}
|
||||
|
||||
END_TEST
|
||||
|
||||
|
||||
#define RENDER_TEST_SETUP \
|
||||
struct status status; \
|
||||
int fds[2]; \
|
||||
pipe( fds );
|
||||
|
||||
void fail_unless_rendered( int fd, char *fragment )
|
||||
void fail_unless_rendered(int fd, char *fragment)
|
||||
{
|
||||
char buf[1024] = {0};
|
||||
char emsg[1024] = {0};
|
||||
char *found = NULL;
|
||||
char buf[1024] = { 0 };
|
||||
char emsg[1024] = { 0 };
|
||||
char *found = NULL;
|
||||
|
||||
sprintf(emsg, "Fragment: %s not found", fragment );
|
||||
sprintf(emsg, "Fragment: %s not found", fragment);
|
||||
|
||||
fail_unless( read_until_newline( fd, buf, 1024 ) > 0, "Couldn't read" );
|
||||
found = strstr( buf, fragment );
|
||||
fail_if( NULL == found, emsg );
|
||||
fail_unless(read_until_newline(fd, buf, 1024) > 0, "Couldn't read");
|
||||
found = strstr(buf, fragment);
|
||||
fail_if(NULL == found, emsg);
|
||||
|
||||
return;
|
||||
return;
|
||||
}
|
||||
|
||||
void fail_if_rendered( int fd, char *fragment )
|
||||
void fail_if_rendered(int fd, char *fragment)
|
||||
{
|
||||
char buf[1024] = {0};
|
||||
char emsg[1024] = {0};
|
||||
char *found = NULL;
|
||||
char buf[1024] = { 0 };
|
||||
char emsg[1024] = { 0 };
|
||||
char *found = NULL;
|
||||
|
||||
sprintf(emsg, "Fragment: %s found", fragment );
|
||||
sprintf(emsg, "Fragment: %s found", fragment);
|
||||
|
||||
fail_unless( read_until_newline( fd, buf, 1024 ) > 0, "Couldn't read" );
|
||||
found = strstr( buf, fragment );
|
||||
fail_unless( NULL == found, emsg );
|
||||
fail_unless(read_until_newline(fd, buf, 1024) > 0, "Couldn't read");
|
||||
found = strstr(buf, fragment);
|
||||
fail_unless(NULL == found, emsg);
|
||||
|
||||
return;
|
||||
return;
|
||||
}
|
||||
|
||||
START_TEST( test_renders_has_control )
|
||||
START_TEST(test_renders_has_control)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
RENDER_TEST_SETUP status.has_control = 1;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "has_control=true");
|
||||
|
||||
status.has_control = 1;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "has_control=true" );
|
||||
|
||||
status.has_control = 0;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "has_control=false" );
|
||||
status.has_control = 0;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "has_control=false");
|
||||
}
|
||||
END_TEST
|
||||
|
||||
|
||||
START_TEST( test_renders_is_mirroring )
|
||||
END_TEST START_TEST(test_renders_is_mirroring)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
RENDER_TEST_SETUP status.is_mirroring = 1;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "is_mirroring=true");
|
||||
|
||||
status.is_mirroring = 1;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "is_mirroring=true" );
|
||||
|
||||
status.is_mirroring = 0;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "is_mirroring=false" );
|
||||
status.is_mirroring = 0;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "is_mirroring=false");
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_renders_clients_allowed )
|
||||
END_TEST START_TEST(test_renders_clients_allowed)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
RENDER_TEST_SETUP status.clients_allowed = 1;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "clients_allowed=true");
|
||||
|
||||
status.clients_allowed = 1;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "clients_allowed=true" );
|
||||
|
||||
status.clients_allowed = 0;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "clients_allowed=false" );
|
||||
status.clients_allowed = 0;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "clients_allowed=false");
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_renders_num_clients )
|
||||
END_TEST START_TEST(test_renders_num_clients)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
RENDER_TEST_SETUP status.num_clients = 0;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "num_clients=0");
|
||||
|
||||
status.num_clients = 0;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "num_clients=0" );
|
||||
|
||||
status.num_clients = 4000;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "num_clients=4000" );
|
||||
status.num_clients = 4000;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "num_clients=4000");
|
||||
|
||||
}
|
||||
END_TEST
|
||||
|
||||
|
||||
START_TEST( test_renders_pid )
|
||||
END_TEST START_TEST(test_renders_pid)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
|
||||
status.pid = 42;
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "pid=42" );
|
||||
RENDER_TEST_SETUP status.pid = 42;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "pid=42");
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_renders_size )
|
||||
END_TEST START_TEST(test_renders_size)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
|
||||
status.size = ( (uint64_t)1 << 33 );
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "size=8589934592" );
|
||||
RENDER_TEST_SETUP status.size = ((uint64_t) 1 << 33);
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "size=8589934592");
|
||||
}
|
||||
END_TEST
|
||||
|
||||
START_TEST( test_renders_migration_statistics )
|
||||
END_TEST START_TEST(test_renders_migration_statistics)
|
||||
{
|
||||
RENDER_TEST_SETUP
|
||||
RENDER_TEST_SETUP status.is_mirroring = 0;
|
||||
status.migration_duration = 8;
|
||||
status.migration_speed = 40000000;
|
||||
status.migration_speed_limit = 40000001;
|
||||
status.migration_seconds_left = 1;
|
||||
status.migration_bytes_left = 5000;
|
||||
|
||||
status.is_mirroring = 0;
|
||||
status.migration_duration = 8;
|
||||
status.migration_speed = 40000000;
|
||||
status.migration_speed_limit = 40000001;
|
||||
status.migration_seconds_left = 1;
|
||||
status.migration_bytes_left = 5000;
|
||||
status_write(&status, fds[1]);
|
||||
fail_if_rendered(fds[0], "migration_duration");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_if_rendered( fds[0], "migration_duration" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_if_rendered(fds[0], "migration_speed");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_if_rendered( fds[0], "migration_speed" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_if_rendered(fds[0], "migration_speed_limit");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_if_rendered( fds[0], "migration_speed_limit" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_if_rendered(fds[0], "migration_seconds_left");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_if_rendered( fds[0], "migration_seconds_left" );
|
||||
status.is_mirroring = 1;
|
||||
|
||||
status.is_mirroring = 1;
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "migration_duration=8");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "migration_duration=8" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "migration_speed=40000000");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "migration_speed=40000000" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "migration_speed_limit=40000001");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "migration_speed_limit=40000001" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "migration_seconds_left=1");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "migration_seconds_left=1" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_unless_rendered(fds[0], "migration_bytes_left=5000");
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_unless_rendered( fds[0], "migration_bytes_left=5000" );
|
||||
status.migration_speed_limit = UINT64_MAX;
|
||||
|
||||
status.migration_speed_limit = UINT64_MAX;
|
||||
|
||||
status_write( &status, fds[1] );
|
||||
fail_if_rendered( fds[0], "migration_speed_limit" );
|
||||
status_write(&status, fds[1]);
|
||||
fail_if_rendered(fds[0], "migration_speed_limit");
|
||||
}
|
||||
END_TEST
|
||||
|
||||
|
||||
Suite *status_suite(void)
|
||||
END_TEST Suite * status_suite(void)
|
||||
{
|
||||
Suite *s = suite_create("status");
|
||||
TCase *tc_create = tcase_create("create");
|
||||
TCase *tc_render = tcase_create("render");
|
||||
Suite *s = suite_create("status");
|
||||
TCase *tc_create = tcase_create("create");
|
||||
TCase *tc_render = tcase_create("render");
|
||||
|
||||
tcase_add_test(tc_create, test_status_create);
|
||||
tcase_add_test(tc_create, test_gets_has_control);
|
||||
tcase_add_test(tc_create, test_gets_is_mirroring);
|
||||
tcase_add_test(tc_create, test_gets_clients_allowed);
|
||||
tcase_add_test(tc_create, test_gets_pid);
|
||||
tcase_add_test(tc_create, test_gets_size);
|
||||
tcase_add_test(tc_create, test_gets_migration_statistics);
|
||||
tcase_add_test(tc_create, test_status_create);
|
||||
tcase_add_test(tc_create, test_gets_has_control);
|
||||
tcase_add_test(tc_create, test_gets_is_mirroring);
|
||||
tcase_add_test(tc_create, test_gets_clients_allowed);
|
||||
tcase_add_test(tc_create, test_gets_pid);
|
||||
tcase_add_test(tc_create, test_gets_size);
|
||||
tcase_add_test(tc_create, test_gets_migration_statistics);
|
||||
|
||||
|
||||
tcase_add_test(tc_render, test_renders_has_control);
|
||||
tcase_add_test(tc_render, test_renders_is_mirroring);
|
||||
tcase_add_test(tc_render, test_renders_clients_allowed);
|
||||
tcase_add_test(tc_render, test_renders_num_clients);
|
||||
tcase_add_test(tc_render, test_renders_pid);
|
||||
tcase_add_test(tc_render, test_renders_size);
|
||||
tcase_add_test(tc_render, test_renders_migration_statistics);
|
||||
tcase_add_test(tc_render, test_renders_has_control);
|
||||
tcase_add_test(tc_render, test_renders_is_mirroring);
|
||||
tcase_add_test(tc_render, test_renders_clients_allowed);
|
||||
tcase_add_test(tc_render, test_renders_num_clients);
|
||||
tcase_add_test(tc_render, test_renders_pid);
|
||||
tcase_add_test(tc_render, test_renders_size);
|
||||
tcase_add_test(tc_render, test_renders_migration_statistics);
|
||||
|
||||
suite_add_tcase(s, tc_create);
|
||||
suite_add_tcase(s, tc_render);
|
||||
suite_add_tcase(s, tc_create);
|
||||
suite_add_tcase(s, tc_render);
|
||||
|
||||
return s;
|
||||
return s;
|
||||
}
|
||||
|
||||
int main(void)
|
||||
{
|
||||
int number_failed;
|
||||
int number_failed;
|
||||
|
||||
Suite *s = status_suite();
|
||||
SRunner *sr = srunner_create(s);
|
||||
srunner_run_all(sr, CK_NORMAL);
|
||||
number_failed = srunner_ntests_failed(sr);
|
||||
srunner_free(sr);
|
||||
return (number_failed == 0) ? 0 : 1;
|
||||
Suite *s = status_suite();
|
||||
SRunner *sr = srunner_create(s);
|
||||
srunner_run_all(sr, CK_NORMAL);
|
||||
number_failed = srunner_ntests_failed(sr);
|
||||
srunner_free(sr);
|
||||
return (number_failed == 0) ? 0 : 1;
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user