293 lines
		
	
	
		
			7.2 KiB
		
	
	
	
		
			C++
		
	
	
	
			
		
		
	
	
			293 lines
		
	
	
		
			7.2 KiB
		
	
	
	
		
			C++
		
	
	
	
/**
 | 
						|
 * @file monTest.cpp
 | 
						|
 * @author slguan (slguan@taosdata.com)
 | 
						|
 * @brief monitor module tests
 | 
						|
 * @version 1.0
 | 
						|
 * @date 2022-03-05
 | 
						|
 *
 | 
						|
 * @copyright Copyright (c) 2022
 | 
						|
 *
 | 
						|
 */
 | 
						|
 | 
						|
#include <gtest/gtest.h>
 | 
						|
#include "os.h"
 | 
						|
 | 
						|
#include "monitor.h"
 | 
						|
#include "tglobal.h"
 | 
						|
 | 
						|
class MonitorTest : public ::testing::Test {
 | 
						|
 protected:
 | 
						|
  static void SetUpTestSuite() {
 | 
						|
    SMonCfg cfg;
 | 
						|
    cfg.maxLogs = 2;
 | 
						|
    cfg.port = 80;
 | 
						|
    cfg.server = "localhost";
 | 
						|
    cfg.comp = 1;
 | 
						|
    monInit(&cfg);
 | 
						|
  }
 | 
						|
 | 
						|
  static void TearDownTestSuite() { monCleanup(); }
 | 
						|
 | 
						|
 public:
 | 
						|
  void SetUp() override {}
 | 
						|
  void TearDown() override {}
 | 
						|
 | 
						|
  void GetBasicInfo(SMonBasicInfo *pInfo);
 | 
						|
  void GetDnodeInfo(SMonDnodeInfo *pInfo);
 | 
						|
  void GetSysInfo(SMonSysInfo *pInfo);
 | 
						|
 | 
						|
  void GetClusterInfo(SMonClusterInfo *pInfo);
 | 
						|
  void GetVgroupInfo(SMonVgroupInfo *pInfo);
 | 
						|
  void GetGrantInfo(SMonGrantInfo *pInfo);
 | 
						|
 | 
						|
  void GetVnodeStat(SVnodesStat *pStat);
 | 
						|
  void GetDiskInfo(SMonDiskInfo *pInfo);
 | 
						|
 | 
						|
  void GetLogInfo(SMonLogs *logs);
 | 
						|
 | 
						|
  void AddLogInfo1();
 | 
						|
  void AddLogInfo2();
 | 
						|
};
 | 
						|
 | 
						|
void MonitorTest::GetBasicInfo(SMonBasicInfo *pInfo) {
 | 
						|
  pInfo->dnode_id = 1;
 | 
						|
  strcpy(pInfo->dnode_ep, "localhost");
 | 
						|
  pInfo->cluster_id = 6980428120398645172;
 | 
						|
  pInfo->protocol = 1;
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetDnodeInfo(SMonDnodeInfo *pInfo) {
 | 
						|
  pInfo->uptime = 1.2;
 | 
						|
  pInfo->has_mnode = 1;
 | 
						|
 | 
						|
  strcpy(pInfo->logdir.name, "/log/dir/d");
 | 
						|
  pInfo->logdir.size.avail = 41;
 | 
						|
  pInfo->logdir.size.total = 42;
 | 
						|
  pInfo->logdir.size.used = 43;
 | 
						|
 | 
						|
  strcpy(pInfo->tempdir.name, "/data/dir/d");
 | 
						|
  pInfo->tempdir.size.avail = 51;
 | 
						|
  pInfo->tempdir.size.total = 52;
 | 
						|
  pInfo->tempdir.size.used = 53;
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetSysInfo(SMonSysInfo *pInfo) {
 | 
						|
  pInfo->cpu_engine = 2.1;
 | 
						|
  pInfo->cpu_system = 2.1;
 | 
						|
  pInfo->cpu_cores = 2;
 | 
						|
  pInfo->mem_engine = 3.1;
 | 
						|
  pInfo->mem_system = 3.2;
 | 
						|
  pInfo->mem_total = 3.3;
 | 
						|
  pInfo->disk_engine = 4.1;
 | 
						|
  pInfo->disk_used = 4.2;
 | 
						|
  pInfo->disk_total = 4.3;
 | 
						|
  pInfo->net_in = 5.1;
 | 
						|
  pInfo->net_out = 5.2;
 | 
						|
  pInfo->io_read = 6.1;
 | 
						|
  pInfo->io_write = 6.2;
 | 
						|
  pInfo->io_read_disk = 7.1;
 | 
						|
  pInfo->io_write_disk = 7.2;
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetClusterInfo(SMonClusterInfo *pInfo) {
 | 
						|
  strcpy(pInfo->first_ep, "localhost:6030");
 | 
						|
  pInfo->first_ep_dnode_id = 1;
 | 
						|
  strcpy(pInfo->version, "3.0.0.0");
 | 
						|
  pInfo->master_uptime = 1;
 | 
						|
  pInfo->monitor_interval = 2;
 | 
						|
  pInfo->vgroups_total = 3;
 | 
						|
  pInfo->vgroups_alive = 43;
 | 
						|
  pInfo->vnodes_total = 5;
 | 
						|
  pInfo->vnodes_alive = 6;
 | 
						|
  pInfo->connections_total = 7;
 | 
						|
 | 
						|
  pInfo->dnodes = taosArrayInit(4, sizeof(SMonDnodeDesc));
 | 
						|
  SMonDnodeDesc d1 = {0};
 | 
						|
  d1.dnode_id = 1;
 | 
						|
  strcpy(d1.dnode_ep, "localhost:6030");
 | 
						|
  strcpy(d1.status, "ready");
 | 
						|
  taosArrayPush(pInfo->dnodes, &d1);
 | 
						|
  SMonDnodeDesc d2 = {0};
 | 
						|
  d2.dnode_id = 2;
 | 
						|
  strcpy(d2.dnode_ep, "localhost:7030");
 | 
						|
  strcpy(d2.status, "offline");
 | 
						|
  taosArrayPush(pInfo->dnodes, &d2);
 | 
						|
 | 
						|
  pInfo->mnodes = taosArrayInit(4, sizeof(SMonMnodeDesc));
 | 
						|
  SMonMnodeDesc m1 = {0};
 | 
						|
  m1.mnode_id = 1;
 | 
						|
  strcpy(m1.mnode_ep, "localhost:6030");
 | 
						|
  strcpy(m1.role, "master");
 | 
						|
  taosArrayPush(pInfo->mnodes, &m1);
 | 
						|
  SMonMnodeDesc m2 = {0};
 | 
						|
  m2.mnode_id = 2;
 | 
						|
  strcpy(m2.mnode_ep, "localhost:7030");
 | 
						|
  strcpy(m2.role, "unsynced");
 | 
						|
  taosArrayPush(pInfo->mnodes, &m2);
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetVgroupInfo(SMonVgroupInfo *pInfo) {
 | 
						|
  pInfo->vgroups = taosArrayInit(4, sizeof(SMonVgroupDesc));
 | 
						|
 | 
						|
  SMonVgroupDesc vg1 = {0};
 | 
						|
  vg1.vgroup_id = 1;
 | 
						|
  strcpy(vg1.database_name, "d1");
 | 
						|
  vg1.tables_num = 4;
 | 
						|
  strcpy(vg1.status, "ready");
 | 
						|
  vg1.vnodes[0].dnode_id = 1;
 | 
						|
  strcpy(vg1.vnodes[0].vnode_role, "master");
 | 
						|
  vg1.vnodes[1].dnode_id = 2;
 | 
						|
  strcpy(vg1.vnodes[1].vnode_role, "slave");
 | 
						|
  taosArrayPush(pInfo->vgroups, &vg1);
 | 
						|
 | 
						|
  SMonVgroupDesc vg2 = {0};
 | 
						|
  vg2.vgroup_id = 2;
 | 
						|
  strcpy(vg2.database_name, "d2");
 | 
						|
  vg2.tables_num = 5;
 | 
						|
  strcpy(vg2.status, "offline");
 | 
						|
  vg2.vnodes[0].dnode_id = 1;
 | 
						|
  strcpy(vg2.vnodes[0].vnode_role, "master");
 | 
						|
  vg2.vnodes[1].dnode_id = 2;
 | 
						|
  strcpy(vg2.vnodes[1].vnode_role, "unsynced");
 | 
						|
  taosArrayPush(pInfo->vgroups, &vg2);
 | 
						|
 | 
						|
  SMonVgroupDesc vg3 = {0};
 | 
						|
  vg3.vgroup_id = 3;
 | 
						|
  strcpy(vg3.database_name, "d3");
 | 
						|
  vg3.tables_num = 6;
 | 
						|
  strcpy(vg3.status, "ready");
 | 
						|
  vg3.vnodes[0].dnode_id = 1;
 | 
						|
  strcpy(vg3.vnodes[0].vnode_role, "master");
 | 
						|
  taosArrayPush(pInfo->vgroups, &vg3);
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetGrantInfo(SMonGrantInfo *pInfo) {
 | 
						|
  pInfo->expire_time = 1234567;
 | 
						|
  pInfo->timeseries_total = 234567;
 | 
						|
  pInfo->timeseries_used = 34567;
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetVnodeStat(SVnodesStat *pInfo) {
 | 
						|
  pInfo->numOfSelectReqs = 8;
 | 
						|
  pInfo->numOfInsertReqs = 9;
 | 
						|
  pInfo->numOfInsertSuccessReqs = 10;
 | 
						|
  pInfo->numOfBatchInsertReqs = 11;
 | 
						|
  pInfo->numOfBatchInsertSuccessReqs = 12;
 | 
						|
  pInfo->errors = 4;
 | 
						|
  pInfo->totalVnodes = 5;
 | 
						|
  pInfo->masterNum = 6;
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetDiskInfo(SMonDiskInfo *pInfo) {
 | 
						|
  pInfo->datadirs = taosArrayInit(2, sizeof(SMonDiskDesc));
 | 
						|
  SMonDiskDesc d1 = {0};
 | 
						|
  strcpy(d1.name, "/t1/d1/d");
 | 
						|
  d1.level = 0;
 | 
						|
  d1.size.avail = 11;
 | 
						|
  d1.size.total = 12;
 | 
						|
  d1.size.used = 13;
 | 
						|
  taosArrayPush(pInfo->datadirs, &d1);
 | 
						|
 | 
						|
  SMonDiskDesc d2 = {0};
 | 
						|
  strcpy(d2.name, "/t2d2/d");
 | 
						|
  d2.level = 2;
 | 
						|
  d2.size.avail = 21;
 | 
						|
  d2.size.total = 22;
 | 
						|
  d2.size.used = 23;
 | 
						|
  taosArrayPush(pInfo->datadirs, &d2);
 | 
						|
 | 
						|
  SMonDiskDesc d3 = {0};
 | 
						|
  strcpy(d3.name, "/t3/d3/d");
 | 
						|
  d3.level = 3;
 | 
						|
  d3.size.avail = 31;
 | 
						|
  d3.size.total = 32;
 | 
						|
  d3.size.used = 33;
 | 
						|
  taosArrayPush(pInfo->datadirs, &d3);
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::GetLogInfo(SMonLogs *logs) {
 | 
						|
  logs->logs = taosArrayInit(4, sizeof(SMonLogItem));
 | 
						|
 | 
						|
  SMonLogItem item1 = {0};
 | 
						|
  item1.level = DEBUG_INFO;
 | 
						|
 | 
						|
  item1.ts = taosGetTimestampMs();
 | 
						|
  strcpy(item1.content, "log test1");
 | 
						|
  taosArrayPush(logs->logs, &item1);
 | 
						|
 | 
						|
  SMonLogItem item2 = {0};
 | 
						|
  item2.level = DEBUG_ERROR;
 | 
						|
  item2.ts = taosGetTimestampMs();
 | 
						|
  strcpy(item2.content, "log test2");
 | 
						|
  taosArrayPush(logs->logs, &item2);
 | 
						|
 | 
						|
  logs->numOfErrorLogs = 1;
 | 
						|
  logs->numOfInfoLogs = 2;
 | 
						|
  logs->numOfDebugLogs = 3;
 | 
						|
  logs->numOfTraceLogs = 4;
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::AddLogInfo1() {
 | 
						|
  monRecordLog(taosGetTimestampMs(), DEBUG_INFO, "1 -------------------------- a");
 | 
						|
  monRecordLog(taosGetTimestampMs(), DEBUG_ERROR, "1 ------------------------ b");
 | 
						|
  monRecordLog(taosGetTimestampMs(), DEBUG_DEBUG, "1 ------- c");
 | 
						|
}
 | 
						|
 | 
						|
void MonitorTest::AddLogInfo2() {
 | 
						|
  monRecordLog(taosGetTimestampMs(), DEBUG_ERROR, "2 ------- a");
 | 
						|
  monRecordLog(taosGetTimestampMs(), DEBUG_ERROR, "2 ------- b");
 | 
						|
}
 | 
						|
 | 
						|
TEST_F(MonitorTest, 01_Full) {
 | 
						|
  AddLogInfo1();
 | 
						|
 | 
						|
  SMonDmInfo dmInfo = {0};
 | 
						|
  GetBasicInfo(&dmInfo.basic);
 | 
						|
  GetDnodeInfo(&dmInfo.dnode);
 | 
						|
  GetSysInfo(&dmInfo.sys);
 | 
						|
 | 
						|
  SMonMmInfo mmInfo = {0};
 | 
						|
  GetClusterInfo(&mmInfo.cluster);
 | 
						|
  GetVgroupInfo(&mmInfo.vgroup);
 | 
						|
  GetGrantInfo(&mmInfo.grant);
 | 
						|
  GetSysInfo(&mmInfo.sys);
 | 
						|
  GetLogInfo(&mmInfo.log);
 | 
						|
 | 
						|
  SMonVmInfo vmInfo = {0};
 | 
						|
  GetDiskInfo(&vmInfo.tfs);
 | 
						|
  GetVnodeStat(&vmInfo.vstat);
 | 
						|
  GetSysInfo(&vmInfo.sys);
 | 
						|
  GetLogInfo(&vmInfo.log);
 | 
						|
 | 
						|
  SMonQmInfo qmInfo = {0};
 | 
						|
  GetSysInfo(&qmInfo.sys);
 | 
						|
  GetLogInfo(&qmInfo.log);
 | 
						|
 | 
						|
  SMonSmInfo smInfo = {0};
 | 
						|
  GetSysInfo(&smInfo.sys);
 | 
						|
  GetLogInfo(&smInfo.log);
 | 
						|
 | 
						|
  SMonBmInfo bmInfo = {0};
 | 
						|
  GetSysInfo(&bmInfo.sys);
 | 
						|
  GetLogInfo(&bmInfo.log);
 | 
						|
 | 
						|
  monSetDmInfo(&dmInfo);
 | 
						|
  monSetMmInfo(&mmInfo);
 | 
						|
  monSetVmInfo(&vmInfo);
 | 
						|
  monSetQmInfo(&qmInfo);
 | 
						|
  monSetSmInfo(&smInfo);
 | 
						|
  monSetBmInfo(&bmInfo);
 | 
						|
 | 
						|
  tFreeSMonMmInfo(&mmInfo);
 | 
						|
  tFreeSMonVmInfo(&vmInfo);
 | 
						|
  tFreeSMonSmInfo(&smInfo);
 | 
						|
  tFreeSMonQmInfo(&qmInfo);
 | 
						|
  tFreeSMonBmInfo(&bmInfo);
 | 
						|
  monSendReport();
 | 
						|
}
 | 
						|
 | 
						|
TEST_F(MonitorTest, 02_Log) {
 | 
						|
  AddLogInfo2();
 | 
						|
  monSendReport();
 | 
						|
}
 |