awtk/tests/wstr_test.cc
2019-02-03 17:13:56 +08:00

346 lines
8.9 KiB
C++

#include "tkc/wstr.h"
#include "tkc/mem.h"
#include "gtest/gtest.h"
static void testSetWStr(const char* utf8, const wchar_t* cstr) {
wstr_t str;
char sutf8[256];
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_set(&str, cstr), RET_OK);
ASSERT_EQ(str.size, wcslen(cstr));
ASSERT_EQ(wcs_cmp(str.str, cstr), 0);
ASSERT_EQ(wstr_get_utf8(&str, sutf8, sizeof(sutf8)), RET_OK);
ASSERT_EQ(strcmp(utf8, sutf8), 0);
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, demo) {
wstr_t s;
wstr_init(&s, 0);
wstr_append(&s, L"abc");
wstr_append(&s, L"123");
wstr_reset(&s);
}
TEST(WStr, basic) {
testSetWStr("Hello", L"Hello");
testSetWStr("中文", L"中文");
}
static void testSetUtf8(const char* utf8, const wchar_t* cstr) {
wstr_t str;
char sutf8[256];
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_set_utf8(&str, utf8), RET_OK);
ASSERT_EQ(str.size, wcslen(cstr));
ASSERT_EQ(wcs_cmp(str.str, cstr), 0);
ASSERT_EQ(wstr_get_utf8(&str, sutf8, sizeof(sutf8)), RET_OK);
ASSERT_EQ(strcmp(utf8, sutf8), 0);
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, utf8) {
testSetUtf8("Hello", L"Hello");
testSetUtf8("中文", L"中文");
}
static void testRemove(const wchar_t* cstr, uint16_t offset, uint16_t nr, ret_t ret,
const wchar_t* expected) {
wstr_t str;
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_set(&str, cstr), RET_OK);
ASSERT_EQ(str.size, wcslen(cstr));
ASSERT_EQ(wcs_cmp(str.str, cstr), 0);
ASSERT_EQ(wstr_remove(&str, offset, nr), ret);
if (ret == RET_OK) {
ASSERT_EQ(wcs_cmp(str.str, expected), 0);
}
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, remove) {
testRemove(L"abc", 1, 2, RET_OK, L"a");
testRemove(L"abc", 0, 3, RET_OK, L"");
testRemove(L"abc", 1, 3, RET_OK, L"a");
testRemove(L"abc", 1, 9, RET_OK, L"a");
testRemove(L"abc", 1, 9, RET_OK, L"a");
testRemove(L"abc", 2, 9, RET_OK, L"ab");
testRemove(L"汉字测试", 1, 2, RET_OK, L"汉试");
testRemove(L"汉字测试", 0, 3, RET_OK, L"");
testRemove(L"汉字测试", 1, 3, RET_OK, L"");
testRemove(L"汉字测试", 1, 9, RET_OK, L"");
testRemove(L"汉字测试", 1, 9, RET_OK, L"");
testRemove(L"汉字测试", 2, 9, RET_OK, L"汉字");
}
static void testInsert(const wchar_t* cstr, const wchar_t* insert, uint16_t offset, ret_t ret,
const wchar_t* expected) {
wstr_t str;
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_set(&str, cstr), RET_OK);
ASSERT_EQ(str.size, wcslen(cstr));
ASSERT_EQ(wcs_cmp(str.str, cstr), 0);
ASSERT_EQ(wstr_insert(&str, offset, insert, wcslen(insert)), ret);
if (ret == RET_OK) {
ASSERT_EQ(wcs_cmp(str.str, expected), 0);
}
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, insert) {
testInsert(L"汉字测试", L"中国", 0, RET_OK, L"中国汉字测试");
testInsert(L"汉字测试", L"中国", 1, RET_OK, L"汉中国字测试");
testInsert(L"汉字测试", L"中国", 3, RET_OK, L"汉字测中国试");
testInsert(L"汉字测试", L"中国", 4, RET_OK, L"汉字测试中国");
}
TEST(WStr, poppush) {
wstr_t str;
const wchar_t* cstr = L"汉字测试";
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_set(&str, cstr), RET_OK);
ASSERT_EQ(str.size, wcslen(cstr));
ASSERT_EQ(wcs_cmp(str.str, cstr), 0);
ASSERT_EQ(wstr_push(&str, 'a'), RET_OK);
ASSERT_EQ(str.size, wcslen(cstr) + 1);
ASSERT_EQ(str.str[str.size - 1], (wchar_t)'a');
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, int) {
wstr_t str;
int32_t v1 = 123;
int32_t v2 = 0;
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_from_int(&str, v1), RET_OK);
ASSERT_EQ(wstr_to_int(&str, &v2), RET_OK);
ASSERT_EQ(v1, v2);
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, double) {
wstr_t str;
double v1 = 123;
double v2 = 0;
ASSERT_EQ(wstr_init(&str, 100), &str);
ASSERT_EQ(str.capacity, 100);
ASSERT_EQ(wstr_from_float(&str, v1), RET_OK);
ASSERT_EQ(wstr_to_float(&str, &v2), RET_OK);
ASSERT_EQ(v1, v2);
ASSERT_EQ(wstr_set(&str, L"1.01"), RET_OK);
ASSERT_EQ(wstr_add_float(&str, 0.01), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1.02"), 0);
ASSERT_EQ(wstr_add_float(&str, 0.001), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1.021"), 0);
ASSERT_EQ(wstr_add_float(&str, 0.1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1.121"), 0);
ASSERT_EQ(wstr_add_float(&str, 1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"2.121"), 0);
ASSERT_EQ(wstr_set(&str, L"1.01"), RET_OK);
ASSERT_EQ(wstr_add_float(&str, -0.01), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1.00"), 0);
ASSERT_EQ(wstr_add_float(&str, -0.001), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"0.999"), 0);
ASSERT_EQ(wstr_add_float(&str, -0.1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"0.899"), 0);
ASSERT_EQ(wstr_add_float(&str, 1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1.899"), 0);
ASSERT_EQ(wstr_add_float(&str, -1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"0.899"), 0);
ASSERT_EQ(wstr_add_float(&str, 123), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"123.899"), 0);
ASSERT_EQ(wstr_set(&str, L"1.010"), RET_OK);
ASSERT_EQ(wstr_trim_float_zero(&str), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1.01"), 0);
ASSERT_EQ(wstr_set(&str, L"1.00"), RET_OK);
ASSERT_EQ(wstr_trim_float_zero(&str), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"1"), 0);
ASSERT_EQ(wstr_set(&str, L"100"), RET_OK);
ASSERT_EQ(wstr_trim_float_zero(&str), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"100"), 0);
ASSERT_EQ(wstr_set(&str, L"100.00000000"), RET_OK);
ASSERT_EQ(wstr_trim_float_zero(&str), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"100"), 0);
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, value) {
wstr_t str;
value_t v1;
int32_t vi;
double vf;
ASSERT_EQ(wstr_init(&str, 0), &str);
ASSERT_EQ(str.capacity, 0);
value_set_int(&v1, 123);
ASSERT_EQ(wstr_from_value(&str, &v1), RET_OK);
ASSERT_EQ(wstr_to_int(&str, &vi), RET_OK);
ASSERT_EQ(value_int(&v1), vi);
value_set_float(&v1, 1234);
ASSERT_EQ(wstr_from_value(&str, &v1), RET_OK);
ASSERT_EQ(wstr_to_float(&str, &vf), RET_OK);
ASSERT_EQ(value_float(&v1), vf);
value_set_str(&v1, "a12345");
ASSERT_EQ(wstr_from_value(&str, &v1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"a12345"), 0);
value_set_wstr(&v1, L"b12345");
ASSERT_EQ(wstr_from_value(&str, &v1), RET_OK);
ASSERT_EQ(wcs_cmp(str.str, L"b12345"), 0);
value_set_float32(&v1, 123);
ASSERT_EQ(wstr_from_value(&str, &v1), RET_OK);
ASSERT_EQ(wstr_to_int(&str, &vi), RET_OK);
ASSERT_EQ(value_int(&v1), vi);
value_set_double(&v1, 123);
ASSERT_EQ(wstr_from_value(&str, &v1), RET_OK);
ASSERT_EQ(wstr_to_int(&str, &vi), RET_OK);
ASSERT_EQ(value_int(&v1), vi);
ASSERT_EQ(wstr_reset(&str), RET_OK);
}
TEST(WStr, wcs_len) {
ASSERT_EQ(wcs_len(L""), 0);
ASSERT_EQ(wcs_len(L"a"), 1);
ASSERT_EQ(wcs_len(L"abc中文测试123"), 10);
}
TEST(WStr, wcs_cpy) {
wchar_t str[100];
ASSERT_EQ(wcs_len(wcs_cpy(str, L"")), 0);
ASSERT_EQ(wcs_len(wcs_cpy(str, L"a")), 1);
ASSERT_EQ(wcs_len(wcs_cpy(str, L"abc中文测试123")), 10);
}
TEST(WStr, wcs_chr) {
size_t i = 0;
const wchar_t* str = L"abc中文测试123";
size_t nr = wcslen(str);
ASSERT_EQ(nr, wcs_len(str));
for (i = 0; i < nr; i++) {
ASSERT_EQ(wcs_chr(str, str[i]), wcschr(str, str[i]));
}
}
TEST(WStr, wcs_dup) {
wchar_t* str = NULL;
str = wcs_dup(L"");
ASSERT_EQ(wcs_cmp(str, L""), 0);
TKMEM_FREE(str);
str = wcs_dup(L"123");
ASSERT_EQ(wcs_cmp(str, L"123"), 0);
TKMEM_FREE(str);
str = wcs_dup(L"abc中文测试123");
ASSERT_EQ(wcs_cmp(str, L"abc中文测试123"), 0);
TKMEM_FREE(str);
}
TEST(WStr, push) {
wstr_t str;
int32_t v = 0;
wstr_t* s = &str;
ASSERT_EQ(wstr_init(&str, 0), &str);
ASSERT_EQ(wstr_push(s, (wchar_t)'1'), RET_OK);
ASSERT_EQ(s->size, 1);
ASSERT_EQ(wstr_push(s, (wchar_t)'2'), RET_OK);
ASSERT_EQ(s->size, 2);
ASSERT_EQ(wstr_append_with_len(s, L"345", 3), RET_OK);
ASSERT_EQ(s->size, 5);
ASSERT_EQ(wstr_to_int(s, &v), RET_OK);
ASSERT_EQ(v, 12345);
wstr_reset(&str);
}
TEST(WStr, push_int) {
wstr_t str;
int32_t v = 0;
wstr_t* s = &str;
ASSERT_EQ(wstr_init(&str, 0), &str);
ASSERT_EQ(wstr_push_int(s, "%d", 12345), RET_OK);
ASSERT_EQ(s->size, 5);
ASSERT_EQ(wstr_to_int(s, &v), RET_OK);
ASSERT_EQ(v, 12345);
s->size = 0;
wstr_reset(&str);
}
TEST(WStr, append) {
wstr_t str;
wstr_init(&str, 0);
ASSERT_EQ(wstr_append(&str, L"123"), RET_OK);
ASSERT_EQ(str.size, 3);
ASSERT_EQ(wstr_append(&str, L"abc"), RET_OK);
ASSERT_EQ(str.size, 6);
ASSERT_EQ(wcscmp(str.str, L"123abc"), 0);
wstr_reset(&str);
}
TEST(WStr, append_len) {
wstr_t str;
wstr_init(&str, 0);
ASSERT_EQ(wstr_append_with_len(&str, L"123123", 3), RET_OK);
ASSERT_EQ(str.size, 3);
ASSERT_EQ(wstr_append_with_len(&str, L"abcabc", 3), RET_OK);
ASSERT_EQ(str.size, 6);
ASSERT_EQ(wcscmp(str.str, L"123abc"), 0);
wstr_reset(&str);
}