libosmocore/tests/strrb/strrb_test.c

226 lines
6.1 KiB
C

/* (C) 2012-2013 by Katerina Barone-Adesi <kat.obsc@gmail.com>
* All Rights Reserved
*
* This program is iree software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*
*/
#include <stdio.h>
#include <string.h>
#include <osmocom/core/strrb.h>
#include <osmocom/core/talloc.h>
#include <osmocom/core/logging.h>
#include <osmocom/core/utils.h>
struct osmo_strrb *rb0, *rb1, *rb2, *rb3, *rb4, *rb5;
#define STR0 "hello"
#define STR1 "a"
#define STR2 "world"
#define STR3 "sky"
#define STR4 "moon"
#define TESTSIZE 3
void init_rbs(void)
{
rb0 = osmo_strrb_create(NULL, TESTSIZE);
rb1 = osmo_strrb_create(NULL, TESTSIZE);
osmo_strrb_add(rb1, STR0);
rb2 = osmo_strrb_create(NULL, TESTSIZE);
osmo_strrb_add(rb2, STR0);
osmo_strrb_add(rb2, STR1);
rb3 = osmo_strrb_create(NULL, TESTSIZE);
osmo_strrb_add(rb3, STR0);
osmo_strrb_add(rb3, STR1);
osmo_strrb_add(rb3, STR2);
rb4 = osmo_strrb_create(NULL, TESTSIZE);
osmo_strrb_add(rb4, STR0);
osmo_strrb_add(rb4, STR1);
osmo_strrb_add(rb4, STR2);
osmo_strrb_add(rb4, STR3);
rb5 = osmo_strrb_create(NULL, TESTSIZE);
osmo_strrb_add(rb5, STR0);
osmo_strrb_add(rb5, STR1);
osmo_strrb_add(rb5, STR2);
osmo_strrb_add(rb5, STR3);
osmo_strrb_add(rb5, STR4);
}
void free_rbs(void)
{
talloc_free(rb0);
talloc_free(rb1);
talloc_free(rb2);
talloc_free(rb3);
talloc_free(rb4);
talloc_free(rb5);
}
void test_offset_valid(void)
{
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb1, 0));
OSMO_ASSERT(!_osmo_strrb_is_bufindex_valid(rb1, 1));
OSMO_ASSERT(!_osmo_strrb_is_bufindex_valid(rb1, 2));
OSMO_ASSERT(!_osmo_strrb_is_bufindex_valid(rb3, 0));
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb3, 1));
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb3, 2));
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb4, 0));
OSMO_ASSERT(!_osmo_strrb_is_bufindex_valid(rb4, 1));
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb4, 2));
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb5, 0));
OSMO_ASSERT(_osmo_strrb_is_bufindex_valid(rb5, 1));
OSMO_ASSERT(!_osmo_strrb_is_bufindex_valid(rb5, 2));
}
void test_elems(void)
{
OSMO_ASSERT(osmo_strrb_elements(rb0) == 0);
OSMO_ASSERT(osmo_strrb_elements(rb1) == 1);
OSMO_ASSERT(osmo_strrb_elements(rb2) == 2);
OSMO_ASSERT(osmo_strrb_elements(rb3) == 2);
}
void test_getn(void)
{
OSMO_ASSERT(!osmo_strrb_get_nth(rb0, 0));
OSMO_ASSERT(!strcmp(STR0, osmo_strrb_get_nth(rb2, 0)));
OSMO_ASSERT(!strcmp(STR1, osmo_strrb_get_nth(rb2, 1)));
OSMO_ASSERT(!strcmp(STR1, osmo_strrb_get_nth(rb3, 0)));
OSMO_ASSERT(!strcmp(STR2, osmo_strrb_get_nth(rb3, 1)));
OSMO_ASSERT(!osmo_strrb_get_nth(rb3, 2));
}
void test_getn_wrap(void)
{
OSMO_ASSERT(!strcmp(STR2, osmo_strrb_get_nth(rb4, 0)));
OSMO_ASSERT(!strcmp(STR3, osmo_strrb_get_nth(rb4, 1)));
OSMO_ASSERT(!strcmp(STR3, osmo_strrb_get_nth(rb5, 0)));
OSMO_ASSERT(!strcmp(STR4, osmo_strrb_get_nth(rb5, 1)));
}
void test_add(void)
{
struct osmo_strrb *rb = osmo_strrb_create(NULL, 4);
OSMO_ASSERT(rb->start == 0);
OSMO_ASSERT(rb->end == 0);
osmo_strrb_add(rb, "a");
osmo_strrb_add(rb, "b");
osmo_strrb_add(rb, "c");
OSMO_ASSERT(rb->start == 0);
OSMO_ASSERT(rb->end == 3);
OSMO_ASSERT(osmo_strrb_elements(rb) == 3);
osmo_strrb_add(rb, "d");
OSMO_ASSERT(rb->start == 1);
OSMO_ASSERT(rb->end == 0);
OSMO_ASSERT(osmo_strrb_elements(rb) == 3);
OSMO_ASSERT(!strcmp("b", osmo_strrb_get_nth(rb, 0)));
OSMO_ASSERT(!strcmp("c", osmo_strrb_get_nth(rb, 1)));
OSMO_ASSERT(!strcmp("d", osmo_strrb_get_nth(rb, 2)));
osmo_strrb_add(rb, "e");
OSMO_ASSERT(rb->start == 2);
OSMO_ASSERT(rb->end == 1);
OSMO_ASSERT(!strcmp("c", osmo_strrb_get_nth(rb, 0)));
OSMO_ASSERT(!strcmp("d", osmo_strrb_get_nth(rb, 1)));
OSMO_ASSERT(!strcmp("e", osmo_strrb_get_nth(rb, 2)));
osmo_strrb_add(rb, "f");
OSMO_ASSERT(rb->start == 3);
OSMO_ASSERT(rb->end == 2);
OSMO_ASSERT(!strcmp("d", osmo_strrb_get_nth(rb, 0)));
OSMO_ASSERT(!strcmp("e", osmo_strrb_get_nth(rb, 1)));
OSMO_ASSERT(!strcmp("f", osmo_strrb_get_nth(rb, 2)));
osmo_strrb_add(rb, "g");
OSMO_ASSERT(rb->start == 0);
OSMO_ASSERT(rb->end == 3);
OSMO_ASSERT(!strcmp("e", osmo_strrb_get_nth(rb, 0)));
OSMO_ASSERT(!strcmp("f", osmo_strrb_get_nth(rb, 1)));
OSMO_ASSERT(!strcmp("g", osmo_strrb_get_nth(rb, 2)));
osmo_strrb_add(rb, "h");
OSMO_ASSERT(rb->start == 1);
OSMO_ASSERT(rb->end == 0);
OSMO_ASSERT(!strcmp("f", osmo_strrb_get_nth(rb, 0)));
OSMO_ASSERT(!strcmp("g", osmo_strrb_get_nth(rb, 1)));
OSMO_ASSERT(!strcmp("h", osmo_strrb_get_nth(rb, 2)));
talloc_free(rb);
}
void test_long_msg(void)
{
struct osmo_strrb *rb = osmo_strrb_create(NULL, 2);
int test_size = RB_MAX_MESSAGE_SIZE + 7;
char *tests1, *tests2;
const char *rb_content;
int i;
tests1 = malloc(test_size);
tests2 = malloc(test_size);
/* Be certain allocating memory worked before continuing */
OSMO_ASSERT(tests1);
OSMO_ASSERT(tests2);
for (i = 0; i < RB_MAX_MESSAGE_SIZE; i += 2) {
tests1[i] = 'a';
tests1[i + 1] = 'b';
}
tests1[i] = '\0';
osmo_strrb_add(rb, tests1);
strcpy(tests2, tests1);
/* Verify that no stale data from test1 is lingering... */
bzero(tests1, test_size);
free(tests1);
rb_content = osmo_strrb_get_nth(rb, 0);
OSMO_ASSERT(!strncmp(tests2, rb_content, RB_MAX_MESSAGE_SIZE - 1));
OSMO_ASSERT(!rb_content[RB_MAX_MESSAGE_SIZE - 1]);
OSMO_ASSERT(strlen(rb_content) == RB_MAX_MESSAGE_SIZE - 1);
free(tests2);
talloc_free(rb);
}
int main(int argc, char **argv)
{
init_rbs();
test_offset_valid();
test_elems();
test_getn();
test_getn_wrap();
test_add();
test_long_msg();
printf("All tests passed\n");
free_rbs();
return 0;
}