32 lines
1.0 KiB
C++
32 lines
1.0 KiB
C++
#ifndef SUN_SERVER_AGENTSERVERTEST_MOCKS_MOCKUSER_H_
|
|
#define SUN_SERVER_AGENTSERVERTEST_MOCKS_MOCKUSER_H_
|
|
|
|
#pragma once
|
|
|
|
#include <gmock/gmock.h>
|
|
|
|
// Note for workarounds: expectations should be set on *Mocker() method.
|
|
|
|
class MockUser : public User
|
|
{
|
|
public:
|
|
virtual ~MockUser() {}
|
|
|
|
// Workaround for default arguments
|
|
// Expectations should be set on SendPacketMocker(), instead of SendPacket()
|
|
MOCK_METHOD3(SendPacketMocker, BOOL(MSG_BASE* pMsg, WORD wSize, const BOOLEAN bUsePacking));
|
|
virtual BOOL SendPacket(MSG_BASE* pMsg, WORD wSize, const BOOLEAN bUsePacking = TRUE) {
|
|
return SendPacketMocker(pMsg, wSize, bUsePacking);
|
|
}
|
|
|
|
// Mocking superclass UserSession ------------------------------------------
|
|
|
|
// Workaround for default arguments
|
|
MOCK_METHOD1(DisconnectUserMocker, VOID(BOOL bGracefulDisconnect));
|
|
virtual VOID DisconnectUser(BOOL bGracefulDisconnect = TRUE) {
|
|
return DisconnectUserMocker(bGracefulDisconnect);
|
|
}
|
|
};
|
|
|
|
#endif // SUN_SERVER_AGENTSERVERTEST_MOCKS_MOCKUSER_H_
|