2020-12-23 14:59:38 +08:00
|
|
|
|
#include "gtest/gtest.h"
|
|
|
|
|
|
#include "tkc/data_reader_factory.h"
|
|
|
|
|
|
#include "tkc/data_reader_mem.h"
|
|
|
|
|
|
|
|
|
|
|
|
TEST(DataReaderMem, basic) {
|
|
|
|
|
|
char buffer[32];
|
2020-12-23 15:55:57 +08:00
|
|
|
|
char url[MAX_PATH + 1];
|
2020-12-23 14:59:38 +08:00
|
|
|
|
const char* data = "hello world";
|
|
|
|
|
|
data_reader_t* reader = NULL;
|
|
|
|
|
|
data_reader_factory_t* f = data_reader_factory_create();
|
|
|
|
|
|
|
|
|
|
|
|
data_reader_mem_build_url(data, strlen(data), url);
|
|
|
|
|
|
reader = data_reader_factory_create_reader(f, url);
|
|
|
|
|
|
ASSERT_EQ(reader == NULL, true);
|
|
|
|
|
|
|
|
|
|
|
|
ASSERT_EQ(data_reader_factory_register(f, "mem", data_reader_mem_create), RET_OK);
|
|
|
|
|
|
|
|
|
|
|
|
reader = data_reader_factory_create_reader(f, url);
|
|
|
|
|
|
ASSERT_EQ(reader != NULL, true);
|
|
|
|
|
|
data_reader_destroy(reader);
|
|
|
|
|
|
|
|
|
|
|
|
reader = data_reader_factory_create_reader(f, url);
|
|
|
|
|
|
ASSERT_EQ(reader != NULL, true);
|
|
|
|
|
|
|
|
|
|
|
|
ASSERT_EQ(data_reader_read(reader, 0, buffer, 5), 5);
|
|
|
|
|
|
ASSERT_EQ(memcmp(buffer, "hello", 5), 0);
|
|
|
|
|
|
|
|
|
|
|
|
ASSERT_EQ(data_reader_read(reader, 1, buffer, 5), 5);
|
|
|
|
|
|
ASSERT_EQ(memcmp(buffer, "ello ", 5), 0);
|
|
|
|
|
|
data_reader_destroy(reader);
|
|
|
|
|
|
|
|
|
|
|
|
data_reader_factory_destroy(f);
|
|
|
|
|
|
}
|
