posts - 26, comments - 2, trackbacks - 0, articles - 0
  C++博客 :: 首页 :: 新随笔 :: 联系 :: 聚合  :: 管理

gtest的一个简单扩展

Posted on 2008-08-27 23:14 小夜 阅读(2286) 评论(0)  编辑 收藏 引用 所属分类: [07] 单元测试 、[51] C&C++

看了gtest的一个simple，其中有测试运行时间的方法，但使用起来稍微负责，因此做了一个简单的扩展。

扩展内容：
1. TEST_T(test_case_name, test_name)，用于定义运行时间测试用例。
2. TEST_T_SHOWTIME()，打开打印测试用例运行时间打印开关。
3. EXCEPT_TIME(second)和ASSERT_TIME(second)，断言，second为double类型，测试运行时间是否小于second。

使用说明：
向正常使用一样，只是在需要时间测试时include “gtest_e.h”即可，当然也得把相应的库链接到执行文件中。

具体实现：
源文件gtest_e.h----

/**
 * gtest_e.h
 */

#ifndef GTEST_E_H_
#define GTEST_E_H_

#include "gtest_time.h"

#define TEST_T(test_case_name, test_name)\
    GTEST_TEST(TIME_##test_case_name, test_name, ::TimeTest)

#define TEST_T_SHOWTIME() TimeTest_setShowFlag(1)

#define ASSERT_TIME(time) if(TimeTest_setTimePoint() - time > 0) \
    FAIL() << "Time: running " << TimeTest_getTime() << "(s) > " << time << "(s)"

#define EXCEPT_TIME(time) if(TimeTest_setTimePoint() - time > 0) \
    ADD_FAILURE() << "Time: running " << TimeTest_getTime() << "(s) > " << time << "(s)"

#endif /* GTEST_E_H_ */
源文件gtest_time.h----
/**
 * gtest_time.h
 */

#ifndef GTEST_TIME_H_
#define GTEST_TIME_H_

#include <gtest/gtest.h>

class TimeTest: public testing::Test
{
public:
    inline void TimeTest_setShowFlag(int flag)
    {
        show_time_ = flag;
    }

    inline double TimeTest_getTime()
    {
        return end_time_ - start_time_;
    }

    double TimeTest_setTimePoint();

protected:
    double start_time_;
    double end_time_;
    int show_time_;

    virtual void SetUp();
    virtual void TearDown();
};

#endif /* GTEST_TIME_H_ */
源文件gtest_time.cpp----
/**
 * gtest_time.cpp
 */

#include <iostream>
#include "gtest_time.h"

using namespace std;

#if defined(WIN32)
#include <sys/timeb.h>
double now()
{
    struct _timeb current;
    _ftime(&current);
    return (((double) current.time) + (1.0 * current.millitm) * 0.000001);
}
#else
double now()
{
    struct timeval current;
    gettimeofday(&current, NULL);
    return (((double) current.tv_sec) + 1.0e-6 * ((double) current.tv_usec));
}
#endif

void TimeTest::SetUp()
{
    start_time_ = now();
    end_time_ = 0;
    TimeTest_setShowFlag(0);
}

void TimeTest::TearDown()
{
    if (show_time_)
    {
        double used_time = TimeTest_setTimePoint();
        cout << "[   TIME   ] used time: " << used_time << "(s)"
                << endl;
    }
}

double TimeTest::TimeTest_setTimePoint()
{
    end_time_ = now();
    return TimeTest_getTime();
}

以上内容只是一个简单的实现，没有过多的测试，且时间精度不够，误差较大。

只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理