mirror of
https://gitee.com/zlgopen/awtk.git
synced 2024-12-05 05:27:58 +08:00
54 lines
1.6 KiB
C++
54 lines
1.6 KiB
C++
#include "gtest/gtest.h"
|
|
#include "tkc/data_writer_factory.h"
|
|
#include "tkc/data_writer_wbuffer.h"
|
|
|
|
TEST(DataWriterWbuffer, wbuffer) {
|
|
wbuffer_t wb;
|
|
char url[MAX_PATH + 1];
|
|
data_writer_t* writer = NULL;
|
|
data_writer_factory_t* f = data_writer_factory_create();
|
|
|
|
wbuffer_init_extendable(&wb);
|
|
data_writer_wbuffer_build_url(&wb, url);
|
|
writer = data_writer_factory_create_writer(f, url);
|
|
ASSERT_EQ(writer == NULL, true);
|
|
|
|
ASSERT_EQ(data_writer_factory_register(f, "wbuffer", data_writer_wbuffer_create), RET_OK);
|
|
|
|
writer = data_writer_factory_create_writer(f, url);
|
|
ASSERT_EQ(writer != NULL, true);
|
|
data_writer_destroy(writer);
|
|
|
|
writer = data_writer_factory_create_writer(f, url);
|
|
ASSERT_EQ(writer != NULL, true);
|
|
data_writer_write(writer, 0, "hello ", 6);
|
|
ASSERT_EQ(memcmp(wb.data, "hello ", 6), 0);
|
|
ASSERT_EQ(wb.cursor, 6u);
|
|
|
|
data_writer_write(writer, 6, "world", 5);
|
|
ASSERT_EQ(wb.cursor, 11u);
|
|
ASSERT_EQ(memcmp(wb.data, "hello world", 11), 0);
|
|
|
|
data_writer_destroy(writer);
|
|
|
|
data_writer_factory_destroy(f);
|
|
wbuffer_deinit(&wb);
|
|
}
|
|
|
|
TEST(DataWriterWbuffer, write_all) {
|
|
wbuffer_t wb;
|
|
char url[MAX_PATH + 1];
|
|
data_writer_factory_t* f = data_writer_factory_create();
|
|
|
|
wbuffer_init_extendable(&wb);
|
|
data_writer_wbuffer_build_url(&wb, url);
|
|
|
|
ASSERT_EQ(data_writer_factory_register(f, "wbuffer", data_writer_wbuffer_create), RET_OK);
|
|
ASSERT_EQ(data_writer_write_all(url, "hello world", 11u), 11);
|
|
ASSERT_EQ(wb.cursor, 11u);
|
|
ASSERT_EQ(memcmp(wb.data, "hello world", 11), 0);
|
|
|
|
data_writer_factory_destroy(f);
|
|
wbuffer_deinit(&wb);
|
|
}
|