greenplumn cdbdispatchresult_test 源码
greenplumn cdbdispatchresult_test 代码
文件路径：/src/backend/cdb/dispatcher/test/cdbdispatchresult_test.c
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include "cmockery.h"
#include "../cdbdispatchresult.c"
#define UNITTEST_NUM_SEGS 2
static struct CdbDispatchResults *
_init_cdbdisp_makeResult()
{
struct CdbDispatchResults *results =
(struct CdbDispatchResults *) palloc0(sizeof(*results));
results->resultArray = palloc0(UNITTEST_NUM_SEGS * sizeof(results->resultArray[0]));
results->resultCapacity = UNITTEST_NUM_SEGS;
return results;
}
/*
* Test cdbdisp_makeResult would return NULL if OOM happens
*/
static void
test__cdbdisp_makeResult__oom(void **state)
{
CdbDispatchResult *result = NULL;
struct CdbDispatchResults *results = _init_cdbdisp_makeResult();
struct SegmentDatabaseDescriptor *segdbDesc =
(struct SegmentDatabaseDescriptor *) palloc0(sizeof(struct SegmentDatabaseDescriptor));
/*
* createPQExpBuffer is supposed to return NULL in OOM cases
*/
will_return_count(createPQExpBuffer, NULL, -1);
expect_any_count(destroyPQExpBuffer, str, -1);
will_be_called_count(destroyPQExpBuffer, -1);
result = cdbdisp_makeResult(results, segdbDesc, 0);
assert_true(result == NULL);
}
int
main(int argc, char *argv[])
{
cmockery_parse_arguments(argc, argv);
const UnitTest tests[] =
{
unit_test(test__cdbdisp_makeResult__oom)
};
Gp_role = GP_ROLE_DISPATCH;
MemoryContextInit();
return run_tests(tests);
}
相关信息
相关文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
6、 golang
-
7、 openharmony