Skip to content

Commit 1b67944

Browse files
authored
Use #ifdef MOCK_METHOD (#1587)
1 parent 2cb36bc commit 1b67944

1 file changed

Lines changed: 14 additions & 11 deletions

File tree

unittest/lib/TestServerSai.cpp

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -48,17 +48,6 @@ static sai_service_method_table_t test_services = {
4848
profile_get_next_value
4949
};
5050

51-
class TestServerSaiMockChannel : public SelectableChannel
52-
{
53-
public:
54-
MOCK_METHOD(bool, empty, (), (override));
55-
MOCK_METHOD(void, pop, (swss::KeyOpFieldsValuesTuple & kco, bool initViewMode), (override));
56-
MOCK_METHOD(void, set, (const std::string &key, const std::vector<swss::FieldValueTuple> &values, const std::string &op), (override));
57-
MOCK_METHOD(int, getFd, (), (override));
58-
MOCK_METHOD(uint64_t, readData, (), (override));
59-
60-
};
61-
6251

6352
TEST(ServerSai, bulkGet)
6453
{
@@ -82,6 +71,19 @@ TEST(ServerSai, bulkGet)
8271
statuses));
8372
}
8473

74+
75+
#ifdef MOCK_METHOD
76+
class TestServerSaiMockChannel : public SelectableChannel
77+
{
78+
public:
79+
MOCK_METHOD(bool, empty, (), (override));
80+
MOCK_METHOD(void, pop, (swss::KeyOpFieldsValuesTuple & kco, bool initViewMode), (override));
81+
MOCK_METHOD(void, set, (const std::string &key, const std::vector<swss::FieldValueTuple> &values, const std::string &op), (override));
82+
MOCK_METHOD(int, getFd, (), (override));
83+
MOCK_METHOD(uint64_t, readData, (), (override));
84+
85+
};
86+
8587
TEST(ServerSai, stats_st_capability_query)
8688
{
8789
SWSS_LOG_ENTER();
@@ -100,3 +102,4 @@ TEST(ServerSai, stats_st_capability_query)
100102
EXPECT_EQ(SAI_STATUS_SUCCESS,
101103
sai.processStatsStCapabilityQuery(kco));
102104
}
105+
#endif

0 commit comments

Comments
 (0)