radare2/test/unit/test_stack.c

#include <r_util.h>
#include "minunit.h"

// Test that popping from an empty stack works.
bool test_r_stack_pop_empty(void) {
	RStack *stack = r_stack_new (10);
	void *elem = r_stack_pop (stack);
	mu_assert_eq ((int)(intptr_t)elem, 0, "pop empty stack");
	r_stack_free (stack);
	mu_end;
}

// Test that we can retrieve an item pushed onto a stack.
bool test_r_stack_push_pop(void) {
	RStack *stack = r_stack_new (10);
	r_stack_push (stack, (void*)(intptr_t)0x1337);
	void *elem = r_stack_pop (stack);
	mu_assert_eq ((int)(intptr_t)elem, 0x1337, "push pop stack");
	r_stack_free (stack);
	mu_end;
}

// Test that the FIFO behavior is done.
bool test_r_stack_push_pop_multi(void) {
	RStack *stack = r_stack_new (10);
	r_stack_push (stack, (void*)(intptr_t)0x1337);
	r_stack_push (stack, (void*)(intptr_t)0x8888);
	void *elem = r_stack_pop (stack);
	mu_assert_eq ((int)(intptr_t)elem, 0x8888, "push pop stack");
	elem = r_stack_pop (stack);
	mu_assert_eq ((int)(intptr_t)elem, 0x1337, "push pop stack");
	r_stack_free (stack);
	mu_end;
}

// Test that stack size grows when more than the allowable number are pushed on.
bool test_r_stack_grow(void) {
	RStack *stack = r_stack_new (2);
	mu_assert_eq (stack->n_elems, 2, "normal stack size");
	r_stack_push (stack, (void*)(intptr_t)0x1337);
	r_stack_push (stack, (void*)(intptr_t)0x8888);
	r_stack_push (stack, (void*)(intptr_t)0xB00B5);
	mu_assert_eq (stack->n_elems, 4, "stack grew!");
	r_stack_free (stack);
	mu_end;
}

int all_tests() {
	mu_run_test(test_r_stack_pop_empty);
	mu_run_test(test_r_stack_push_pop);
	mu_run_test(test_r_stack_push_pop_multi);
	mu_run_test(test_r_stack_grow);
	return tests_passed != tests_run;
}

int main(int argc, char **argv) {
	return all_tests();
}