gtest hook 示例

2019独角兽企业重金招聘Python工程师标准>>>

#include <gtest/gtest.h>
#include "hookmon.h"

int main(int argc, char **argv)
{
    testing::InitGoogleTest(&argc, argv);
    return RUN_ALL_TESTS();
}

int (*old_func1)(void* instance, char* url, uint32_t timeout);
int new_func1(void* instance, char* url, uint32_t timeout)
{
    if (strcmp(url, "91himarket") == 0) {
        return 0;
    } else if (strcmp(url, "91assistant") == 0) {  }   
        return 1;
    }
}

void __attribute__((constructor)) hook_init(void)
{
    attach_func("full_namespace::MyClass::talk_with_http", (void*)new_func1, (void**)&old_func1);
}

class test_MyClass : public ::testing::Test{
protected:
    test_MyClass(){};
    virtual ~test_MyClass(){};
    virtual void SetUp() {}
    virtual void TearDown() {  }
};

TEST_F(test_VersionFetcher, test_build_assistant_http_req)
{
    MyClass myclass;
    EXPECT_EQ(0, myclass.talk_with_http("url1", 0));
}

class MyClass {
    int talk_with_http(char* url, uint32_t timeout);
};

注意：MyClass中talk_with_http成员函数在hook时，需要额外增加一个参数void*, 其值实际上是MyClass对象的指针，否则url地址会向左偏移。

转载于:https://my.oschina.net/zmlblog/blog/363596