From 19cbfdd0eefc182d8a6fd32227d3db8541052e13 Mon Sep 17 00:00:00 2001 From: Junhua Zhai Date: Fri, 25 Apr 2025 23:59:09 +0000 Subject: [PATCH] Use #ifdef MOCK_METHOD Signed-off-by: Junhua Zhai --- unittest/lib/TestServerSai.cpp | 25 ++++++++++++++----------- 1 file changed, 14 insertions(+), 11 deletions(-) diff --git a/unittest/lib/TestServerSai.cpp b/unittest/lib/TestServerSai.cpp index 224018fefc..da8e386197 100644 --- a/unittest/lib/TestServerSai.cpp +++ b/unittest/lib/TestServerSai.cpp @@ -48,17 +48,6 @@ static sai_service_method_table_t test_services = { profile_get_next_value }; -class TestServerSaiMockChannel : public SelectableChannel -{ -public: - MOCK_METHOD(bool, empty, (), (override)); - MOCK_METHOD(void, pop, (swss::KeyOpFieldsValuesTuple & kco, bool initViewMode), (override)); - MOCK_METHOD(void, set, (const std::string &key, const std::vector &values, const std::string &op), (override)); - MOCK_METHOD(int, getFd, (), (override)); - MOCK_METHOD(uint64_t, readData, (), (override)); - -}; - TEST(ServerSai, bulkGet) { @@ -82,6 +71,19 @@ TEST(ServerSai, bulkGet) statuses)); } + +#ifdef MOCK_METHOD +class TestServerSaiMockChannel : public SelectableChannel +{ +public: + MOCK_METHOD(bool, empty, (), (override)); + MOCK_METHOD(void, pop, (swss::KeyOpFieldsValuesTuple & kco, bool initViewMode), (override)); + MOCK_METHOD(void, set, (const std::string &key, const std::vector &values, const std::string &op), (override)); + MOCK_METHOD(int, getFd, (), (override)); + MOCK_METHOD(uint64_t, readData, (), (override)); + +}; + TEST(ServerSai, stats_st_capability_query) { SWSS_LOG_ENTER(); @@ -100,3 +102,4 @@ TEST(ServerSai, stats_st_capability_query) EXPECT_EQ(SAI_STATUS_SUCCESS, sai.processStatsStCapabilityQuery(kco)); } +#endif