论坛 › 物联网 › 主动化压测工具开发(MFC)

张国伟 发表于 2024-6-19 20:53:43

主动化压测工具开发(MFC)

1. 背景

为了减轻测试人员在进行MFC程序压力测试时的重复手动操作，本文档描述了开发一个主动化压力测试工具的过程。该工具可以或许根据程序界面某块地区的预定状态变化，主动执行鼠标点击或键盘输入操作。
2. 技能概览



[*]串口控制：用于控制外部设备，如继电器。
[*]MFC CRectTracker：实现截图功能，创建简单的截图对话框。
[*]GDI+：用于图像处理和屏幕捕获。
[*]mouse_event：模仿鼠标点击。
[*]keybd_event：模仿键盘输入。
[*]MD5：用于图像内容校验。
3. 串口通信控制

串口通光荣于控制继电器的上下电状态，进而控制USB设备的电源。
3.1 串口初始化

bool CComTest::InitialCom(int iComID, int iComPort, DWORD iBaudRate) {
    // 省略部分错误检查和资源清理代码...

    // 配置串口参数
    DCB dcbSerialParams = {0};
    GetCommState(m_hCom, &dcbSerialParams);
    dcbSerialParams.BaudRate = iBaudRate;
    dcbSerialParams.ByteSize = 8;
    dcbSerialParams.StopBits = STOPBITS_ONE;
    dcbSerialParams.Parity = NOPARITY;
    SetCommState(m_hCom, &dcbSerialParams);

    // 设置串口超时
    COMMTIMEOUTS timeouts = {0};
    timeouts.ReadIntervalTimeout = MAXDWORD;
    SetCommTimeouts(m_hCom, &timeouts);

    // 完成串口初始化
    return TRUE;
}
3.2 读取串口

DWORD CComTest::ReadCom(BYTE *pBuff, int nCount) {
    DWORD dwRead = 0;
    ReadFile(m_hCom, pBuff, nCount, &dwRead, NULL);
    return dwRead;
}
3.3 写串口

BOOL CComTest::WriteCom(BYTE *pBuff, int nCount) {
    DWORD dwWritten = 0;
    return WriteFile(m_hCom, pBuff, nCount, &dwWritten, NULL);
}
3.4 发送串口命令示例

BYTE pData[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
WriteCom(m_iComID, pData, sizeof(pData));
4. 程序界面状态监控

通过截图对话框实现地区状态监控，主动执行下一步动作。
4.1 截图对话框初始化

BOOL CCaptureDlg::OnInitDialog() {
    CDialog::OnInitDialog();
    // 屏幕截图并设置窗口大小
    copyScreenToBitmap(m_ScreenBmp);
    MoveWindow(-3, -3, GetSystemMetrics(SM_CXSCREEN) + 6, GetSystemMetrics(SM_CYSCREEN) + 6);

    // 初始化橡皮筋跟踪器
    m_rectTracker.m_nStyle = CRectTracker::resizeOutside | CRectTracker::dottedLine;
    m_rectTracker.m_rect.SetRect(0, 0, 0, 0);

    return TRUE;
}
4.2 截图对话框背景绘制

BOOL CCaptureDlg::OnEraseBkgnd(CDC *pDC) {
    CDC memDC;
    memDC.CreateCompatibleDC(pDC);
    memDC.SelectObject(&m_ScreenBmp);
    pDC->BitBlt(0, 0, GetClientRect().Width(), GetClientRect().Height(), &memDC, 0, 0, SRCCOPY);
    memDC.DeleteDC();
    return TRUE;
}
4.3 鼠标消息处理

void CCaptureDlg::OnLButtonDown(UINT nFlags, CPoint point) {
    if (m_rectTracker.HitTest(point) == CRectTracker::hitNothing) {
      m_rectTracker.TrackRubberBand(this, point, TRUE);
    } else {
      m_rectTracker.Track(this, point, TRUE);
      m_rectTracker.m_rect.NormalizeRect();
    }
    Invalidate(TRUE);
    CDialog::OnLButtonDown(nFlags, point);
}
4.4 双击变乱处理

void CCaptureDlg::OnLButtonDblClk(UINT nFlags, CPoint point) {
    if (m_rectTracker.HitTest(point) != CRectTracker::hitMiddle) {
      MessageBox(_T("截图失败,请重新截图！"));
      return;
    }
      // 检查双击是否在橡皮筋跟踪器的中间区域
    if (m_rectTracker.HitTest(point) != CRectTracker::hitMiddle)
    {
      MessageBox(_T("截图失败, 请重新截图！"));
      return;
    }

// 保存双击时坐标
m_startPointX = point.x;
m_startPointY = point.y;

    // 获取当前屏幕设备上下文
    CDC* pScreenDC = GetDC();
    if (!pScreenDC)
    {
      MessageBox(_T("无法获取屏幕设备上下文"));
      return;
    }

    // 创建内存设备上下文与屏幕DC兼容，并选择m_ScreenBmp位图进该内存DC
    CDC memDC;
    memDC.CreateCompatibleDC(pScreenDC);
    CBitmap* pOldBitmap = memDC.SelectObject(&m_ScreenBmp);

    // 获取橡皮筋跟踪器定义的区域
    CRect captureRect = m_rectTracker.m_rect;

    // 创建一个兼容的位图用于保存截取的区域
    CBitmap captureBmp;
    captureBmp.CreateCompatibleBitmap(pScreenDC, captureRect.Width(), captureRect.Height());

    // 创建另一个内存DC，选择刚才创建的兼容位图进这个DC
    CDC captureDC;
    captureDC.CreateCompatibleDC(pScreenDC);
    CBitmap* pOldCaptureBmp = captureDC.SelectObject(&captureBmp);

    // 从原屏幕位图中截取定义区域的图像到captureDC
    captureDC.BitBlt(0, 0, captureRect.Width(), captureRect.Height(), &memDC, captureRect.left, captureRect.top, SRCCOPY);

    // 使用CImage类来保存位图到文件
    CImage img;
    img.Attach(captureBmp);
    CString strCapturePath = _T("你想要保存的路径\\capture1.jpg");
    img.Save(strCapturePath);

    // 清理资源
    captureDC.SelectObject(pOldCaptureBmp);
    memDC.SelectObject(pOldBitmap);
    captureBmp.DeleteObject();
    memDC.DeleteDC();
    captureDC.DeleteDC();
    ReleaseDC(pScreenDC);

    // 截图操作完成后关闭对话框
    CDialog::OnCancel();
}
在压测前，将被压测程序运行到指定状态，用上述[截图窗口]截取指定地区的位图(目标状态)，在压测时，不停的检测此地区的变化，如达到目标状态，则执行模仿点击等操作。

4.5 截取屏幕上指定地区的图像

下面函数用于截取屏幕上指定地区的图像，在压测时，可用于实时检测指定地区的位图。
void CCaptureDlg::Screen(const CRect& cRect, int iNumber)
{
    // 使用局部函数来自动释放资源
    struct AutoRelease
    {
      HDC hDC;
      ~AutoRelease() { if (hDC) ::ReleaseDC(NULL, hDC); }
    } autoReleaseDC;

    // 获取屏幕设备上下文
    autoReleaseDC.hDC = ::GetDC(NULL);
    if (!autoReleaseDC.hDC)
    {
      MessageBox(_T("无法获取屏幕设备上下文。"));
      return;
    }

    CDC memDC;
    memDC.CreateCompatibleDC(NULL); // 创建与屏幕兼容的内存设备上下文

    // 创建与屏幕兼容的位图
    CBitmap memBitmap;
    memBitmap.CreateCompatibleBitmap(autoReleaseDC.hDC, cRect.Width(), cRect.Height());

    // 将新创建的位图选入内存设备上下文中
    CBitmap* pOldBitmap = memDC.SelectObject(&memBitmap);

    // 从屏幕设备上下文中复制图像到内存设备上下文中的位图
    memDC.BitBlt(0, 0, cRect.Width(), cRect.Height(), CDC::FromHandle(autoReleaseDC.hDC), cRect.left, cRect.top, SRCCOPY);

    // 使用CImage类来操作和保存位图
    CImage img;
    if (!img.Attach(memBitmap))
    {
      MessageBox(_T("无法将位图附加到CImage对象。"));
      return;
    }

    // 构造文件保存路径
    CString strCapturePath;
    strCapturePath.Format(_T("保存路径\\capture%d.jpg"), iNumber);

    // 保存位图到文件
    if (!img.Save(strCapturePath))
    {
      AfxMessageBox(_T("保存图像失败。"));
    }

    // 清理资源
    memDC.SelectObject(pOldBitmap); // 恢复原始位图
    img.Detach(); // 从CImage对象中分离位图
}
5. 模仿输入操作

5.1 模仿鼠标点击

void SimulateMouseClick(int x, int y) {
    SetCursorPos(x, y);
    mouse_event(MOUSEEVENTF_LEFTDOWN, x, y, 0, 0);
    mouse_event(MOUSEEVENTF_LEFTUP, x, y, 0, 0);
}
5.2 模仿键盘输入

void SimulateKeyboardInput(const char *szInput) {
    for (int i = 0; szInput != '\0'; ++i) {
      keybd_event(szInput, 0, 0, 0);
      keybd_event(szInput, 0, KEYEVENTF_KEYUP, 0);
    }
}
结束语

本文档提供了一个主动化压力测试工具的底子实现，包括串口通信、截图监控、状态校验和模仿输入操作。根据详细需求，可进一步扩展和优化工具的功能。

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。

查看完整版本: 主动化压测工具开发(MFC)