xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

try fix window com error

This commit is contained in:
54liuyao 2024-12-20 13:33:32 +08:00
parent 9866b9437d
commit 5212710947
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
source/libs/stream/test/streamCheckPointTest.cpp
View File

@ -5,7 +5,7 @@
TEST(streamCheckpointTest, StreamTaskProcessCheckpointTriggerRsp) { TEST(streamCheckpointTest, StreamTaskProcessCheckpointTriggerRsp) {
SStreamTask* pTask = NULL; SStreamTask* pTask = NULL;
int64_t uid = 1111111111111111; int64_t uid = 1111111111111111;
SArray* array = taosArrayInit(4, sizeof(SStreamTask)); SArray* array = taosArrayInit(4, POINTER_BYTES);
int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array, int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array,
false, 1, &pTask); false, 1, &pTask);
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
@ -35,12 +35,13 @@ TEST(streamCheckpointTest, StreamTaskProcessCheckpointTriggerRsp) {
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
tFreeStreamTask(pTask); tFreeStreamTask(pTask);
taosArrayDestroy(array);
} }
TEST(streamCheckpointTest, StreamTaskSetFailedCheckpointId) { TEST(streamCheckpointTest, StreamTaskSetFailedCheckpointId) {
SStreamTask* pTask = NULL; SStreamTask* pTask = NULL;
int64_t uid = 1111111111111111; int64_t uid = 1111111111111111;
SArray* array = taosArrayInit(4, sizeof(SStreamTask)); SArray* array = taosArrayInit(4, POINTER_BYTES);
int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array, int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array,
false, 1, &pTask); false, 1, &pTask);
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
@ -62,12 +63,14 @@ TEST(streamCheckpointTest, StreamTaskSetFailedCheckpointId) {
streamTaskSetFailedCheckpointId(pTask, pInfo->failedId - 1); streamTaskSetFailedCheckpointId(pTask, pInfo->failedId - 1);
ASSERT_EQ(pInfo->failedId, failedCheckpointId); ASSERT_EQ(pInfo->failedId, failedCheckpointId);
tFreeStreamTask(pTask); tFreeStreamTask(pTask);
taosArrayDestroy(array);
} }
TEST(UploadCheckpointDataTest, UploadSuccess) { TEST(UploadCheckpointDataTest, UploadSuccess) {
streamMetaInit();
SStreamTask* pTask = NULL; SStreamTask* pTask = NULL;
int64_t uid = 1111111111111111; int64_t uid = 1111111111111111;
SArray* array = taosArrayInit(4, sizeof(SStreamTask)); SArray* array = taosArrayInit(4, POINTER_BYTES);
int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array, int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array,
false, 1, &pTask); false, 1, &pTask);
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
@ -96,12 +99,16 @@ TEST(UploadCheckpointDataTest, UploadSuccess) {
int32_t result = uploadCheckpointData(pTask, checkpointId, dbRefId, type); int32_t result = uploadCheckpointData(pTask, checkpointId, dbRefId, type);
EXPECT_EQ(result, TSDB_CODE_SUCCESS) << "uploadCheckpointData should return 0 on success"; EXPECT_EQ(result, TSDB_CODE_SUCCESS) << "uploadCheckpointData should return 0 on success";
tFreeStreamTask(pTask);
taosRemoveDir(path);
streamStateClose(pState, true);
taosArrayDestroy(array);
} }
TEST(UploadCheckpointDataTest, UploadDisabled) { TEST(UploadCheckpointDataTest, UploadDisabled) {
SStreamTask* pTask = NULL; SStreamTask* pTask = NULL;
int64_t uid = 2222222222222; int64_t uid = 2222222222222;
SArray* array = taosArrayInit(4, sizeof(SStreamTask)); SArray* array = taosArrayInit(4, POINTER_BYTES);
int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array, int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array,
false, 1, &pTask); false, 1, &pTask);
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
@ -131,12 +138,16 @@ TEST(UploadCheckpointDataTest, UploadDisabled) {
int32_t result = uploadCheckpointData(pTask, checkpointId, dbRefId, type); int32_t result = uploadCheckpointData(pTask, checkpointId, dbRefId, type);
EXPECT_NE(result, TSDB_CODE_SUCCESS) << "uploadCheckpointData should return 0 when backup type is disabled"; EXPECT_NE(result, TSDB_CODE_SUCCESS) << "uploadCheckpointData should return 0 when backup type is disabled";
streamStateClose(pState, true);
tFreeStreamTask(pTask);
taosArrayDestroy(array);
} }
TEST(StreamTaskAlreadySendTriggerTest, AlreadySendTrigger) { TEST(StreamTaskAlreadySendTriggerTest, AlreadySendTrigger) {
SStreamTask* pTask = NULL; SStreamTask* pTask = NULL;
int64_t uid = 2222222222222; int64_t uid = 2222222222222;
SArray* array = taosArrayInit(4, sizeof(SStreamTask)); SArray* array = taosArrayInit(4, POINTER_BYTES);
int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array, int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array,
false, 1, &pTask); false, 1, &pTask);
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
@ -167,13 +178,14 @@ TEST(StreamTaskAlreadySendTriggerTest, AlreadySendTrigger) {
EXPECT_TRUE(result) << "The trigger message should have been sent to the downstream node"; EXPECT_TRUE(result) << "The trigger message should have been sent to the downstream node";
taosArrayDestroy(pTask->chkInfo.pActiveInfo->pDispatchTriggerList); tFreeStreamTask(pTask);
taosArrayDestroy(array);
} }
TEST(ChkptTriggerRecvMonitorHelperTest, chkptTriggerRecvMonitorHelper) { TEST(ChkptTriggerRecvMonitorHelperTest, chkptTriggerRecvMonitorHelper) {
SStreamTask* pTask = NULL; SStreamTask* pTask = NULL;
int64_t uid = 2222222222222; int64_t uid = 2222222222222;
SArray* array = taosArrayInit(4, sizeof(SStreamTask)); SArray* array = taosArrayInit(4, POINTER_BYTES);
int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array, int32_t code = tNewStreamTask(uid, TASK_LEVEL__SINK, NULL, false, 0, 0, array,
false, 1, &pTask); false, 1, &pTask);
ASSERT_EQ(code, TSDB_CODE_SUCCESS); ASSERT_EQ(code, TSDB_CODE_SUCCESS);
@ -215,4 +227,7 @@ TEST(ChkptTriggerRecvMonitorHelperTest, chkptTriggerRecvMonitorHelper) {
pTask->pMeta->fatalInfo.code = TSDB_CODE_FAILED; pTask->pMeta->fatalInfo.code = TSDB_CODE_FAILED;
streamSetFatalError(pTask->pMeta, code, __func__, __LINE__); streamSetFatalError(pTask->pMeta, code, __func__, __LINE__);
tFreeStreamTask(pTask);
taosArrayDestroy(array);
taosArrayDestroy(array1);
} }