mirror of
https://gitee.com/zlgopen/awtk.git
synced 2024-12-05 05:27:58 +08:00
127 lines
3.7 KiB
C++
127 lines
3.7 KiB
C++
#include "streams/mem/istream_mem.h"
|
|
#include "gtest/gtest.h"
|
|
|
|
TEST(IStreamMem, basic) {
|
|
uint8_t buff[4];
|
|
const char* str = "111122223333";
|
|
tk_istream_t* is = tk_istream_mem_create((uint8_t*)str, strlen(str), 0, FALSE);
|
|
|
|
ASSERT_EQ(tk_object_get_prop_bool(TK_OBJECT(is), TK_STREAM_PROP_IS_EOS, TRUE), FALSE);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 4), 4);
|
|
ASSERT_EQ(memcmp(buff, "1111", 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 4), 4);
|
|
ASSERT_EQ(memcmp(buff, "2222", 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 4), 4);
|
|
ASSERT_EQ(memcmp(buff, "3333", 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 4), 0);
|
|
ASSERT_EQ(tk_object_get_prop_bool(TK_OBJECT(is), TK_STREAM_PROP_IS_EOS, FALSE), TRUE);
|
|
|
|
tk_object_unref(TK_OBJECT(is));
|
|
}
|
|
|
|
TEST(IStreamMem, packet_size) {
|
|
uint8_t buff[12];
|
|
const char* str = "111122223333";
|
|
tk_istream_t* is = tk_istream_mem_create((uint8_t*)str, strlen(str), 4, FALSE);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 12), 4);
|
|
ASSERT_EQ(memcmp(buff, "1111", 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 10), 4);
|
|
ASSERT_EQ(memcmp(buff, "2222", 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 8), 4);
|
|
ASSERT_EQ(memcmp(buff, "3333", 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read(is, buff, 4), 0);
|
|
|
|
ASSERT_EQ(tk_istream_seek(is, 0), RET_OK);
|
|
ASSERT_EQ(tk_istream_read(is, buff, 12), 4);
|
|
ASSERT_EQ(memcmp(buff, "1111", 4), 0);
|
|
|
|
tk_object_unref(TK_OBJECT(is));
|
|
}
|
|
|
|
TEST(IStreamMem, read_len) {
|
|
uint8_t buff[12];
|
|
const char* str = "111122223333";
|
|
tk_istream_t* is = tk_istream_mem_create((uint8_t*)str, strlen(str), 4, FALSE);
|
|
|
|
ASSERT_EQ(tk_istream_read_len(is, buff, 12, 3000), 12);
|
|
ASSERT_EQ(memcmp(buff, str, 12), 0);
|
|
|
|
tk_object_unref(TK_OBJECT(is));
|
|
}
|
|
|
|
TEST(IStreamMem, seek) {
|
|
uint8_t buff[12];
|
|
const char* str = "111122223333";
|
|
tk_istream_t* is = tk_istream_mem_create((uint8_t*)str, strlen(str), 4, FALSE);
|
|
|
|
ASSERT_EQ(tk_istream_eos(is), FALSE);
|
|
ASSERT_EQ(tk_istream_read_len(is, buff, 12, 3000), 12);
|
|
ASSERT_EQ(memcmp(buff, str, 12), 0);
|
|
ASSERT_EQ(tk_istream_eos(is), TRUE);
|
|
ASSERT_EQ(tk_istream_tell(is), 12);
|
|
|
|
ASSERT_EQ(tk_istream_seek(is, 0), RET_OK);
|
|
ASSERT_EQ(tk_istream_tell(is), 0);
|
|
|
|
ASSERT_EQ(tk_istream_read_len(is, buff, 12, 3000), 12);
|
|
ASSERT_EQ(memcmp(buff, str, 12), 0);
|
|
ASSERT_EQ(tk_istream_eos(is), TRUE);
|
|
ASSERT_EQ(tk_istream_tell(is), 12);
|
|
|
|
tk_object_unref(TK_OBJECT(is));
|
|
}
|
|
|
|
TEST(IStreamMem, read_line_str) {
|
|
str_t str;
|
|
const char* content = "1111\n2222\r\n\r\n3333\n";
|
|
tk_istream_t* is = tk_istream_mem_create((uint8_t*)content, strlen(content), 4, FALSE);
|
|
|
|
str_init(&str, 1024);
|
|
ASSERT_EQ(tk_istream_read_line_str(is, &str), RET_OK);
|
|
ASSERT_STREQ(str.str, "1111");
|
|
|
|
ASSERT_EQ(tk_istream_read_line_str(is, &str), RET_OK);
|
|
ASSERT_STREQ(str.str, "2222");
|
|
|
|
ASSERT_EQ(tk_istream_read_line_str(is, &str), RET_OK);
|
|
ASSERT_STREQ(str.str, "");
|
|
|
|
ASSERT_EQ(tk_istream_read_line_str(is, &str), RET_OK);
|
|
ASSERT_STREQ(str.str, "3333");
|
|
|
|
ASSERT_NE(tk_istream_read_line_str(is, &str), RET_OK);
|
|
|
|
str_reset(&str);
|
|
tk_object_unref(TK_OBJECT(is));
|
|
}
|
|
|
|
TEST(IStreamMem, read_line) {
|
|
char str[128];
|
|
const char* content = "1111\n2222\r\n\r\n3333\n";
|
|
tk_istream_t* is = tk_istream_mem_create((uint8_t*)content, strlen(content), 4, FALSE);
|
|
uint32_t timeout = 100000000;
|
|
ASSERT_EQ(tk_istream_read_line(is, str, sizeof(str) - 1, timeout), 5);
|
|
ASSERT_STREQ(str, "1111\n");
|
|
|
|
ASSERT_EQ(tk_istream_read_line(is, str, sizeof(str) - 1, timeout), 6);
|
|
ASSERT_STREQ(str, "2222\r\n");
|
|
|
|
ASSERT_EQ(tk_istream_read_line(is, str, sizeof(str) - 1, timeout), 2);
|
|
ASSERT_STREQ(str, "\r\n");
|
|
|
|
ASSERT_EQ(tk_istream_read_line(is, str, sizeof(str) - 1, timeout), 5);
|
|
ASSERT_STREQ(str, "3333\n");
|
|
|
|
ASSERT_EQ(tk_istream_read_line(is, str, sizeof(str) - 1, timeout), 0);
|
|
|
|
tk_object_unref(TK_OBJECT(is));
|
|
}
|