#include <CuTest.h>
#include <blue/ds/string.h>

static void test_string_create(CuTest *tc)
{
	b_string *str = b_string_create();

	CuAssertPtrNotNull(tc, str);
	CuAssertIntEquals(tc, 0, b_string_get_size(str, B_STRLEN_NORMAL));
	CuAssertStrEquals(tc, "", b_string_ptr(str));

	b_string_release(str);

	str = b_string_create_from_c('A', 8);

	CuAssertPtrNotNull(tc, str);
	CuAssertIntEquals(tc, 8, b_string_get_size(str, B_STRLEN_NORMAL));
	CuAssertStrEquals(tc, "AAAAAAAA", b_string_ptr(str));

	b_string_release(str);

	str = b_string_create_from_cstr("Hello, world!");

	CuAssertPtrNotNull(tc, str);
	CuAssertIntEquals(tc, 13, b_string_get_size(str, B_STRLEN_NORMAL));
	CuAssertStrEquals(tc, "Hello, world!", b_string_ptr(str));

	b_string_release(str);
}

static void test_string_length(CuTest *tc)
{
	const char *cstr = "Hello, \033[91;1mworld!";
	b_string *s = b_string_create_from_cstr(cstr);

	CuAssertIntEquals(tc, 13, b_string_get_size(s, B_STRLEN_IGNORE_ESC));
	CuAssertIntEquals(tc, 20, b_string_get_size(s, B_STRLEN_NORMAL));

	b_string_release(s);
}

CuSuite *get_all_tests(void)
{
	CuSuite *suite = CuSuiteNew();

	SUITE_ADD_TEST(suite, test_string_create);
	SUITE_ADD_TEST(suite, test_string_length);

	return suite;
}