gecko/ipc/ipdl/test/cxx/TestRPCRaces.h
2010-03-11 01:35:32 -06:00

144 lines
2.7 KiB
C++

#ifndef mozilla__ipdltest_TestRPCRaces_h
#define mozilla__ipdltest_TestRPCRaces_h
#include "mozilla/_ipdltest/IPDLUnitTests.h"
#include "mozilla/_ipdltest/PTestRPCRacesParent.h"
#include "mozilla/_ipdltest/PTestRPCRacesChild.h"
namespace mozilla {
namespace _ipdltest {
mozilla::ipc::RPCChannel::RacyRPCPolicy
MediateRace(const mozilla::ipc::RPCChannel::Message& parent,
const mozilla::ipc::RPCChannel::Message& child);
class TestRPCRacesParent :
public PTestRPCRacesParent
{
public:
TestRPCRacesParent() : mHasReply(false),
mChildHasReply(false),
mAnsweredParent(false)
{ }
virtual ~TestRPCRacesParent() { }
void Main();
protected:
NS_OVERRIDE
virtual bool
RecvStartRace();
NS_OVERRIDE
virtual bool
AnswerRace(bool* hasRace);
NS_OVERRIDE
virtual bool
AnswerStackFrame();
NS_OVERRIDE
virtual bool
AnswerStackFrame3();
NS_OVERRIDE
virtual bool
AnswerParent();
NS_OVERRIDE
virtual bool
RecvGetAnsweredParent(bool* answeredParent);
NS_OVERRIDE
virtual mozilla::ipc::RPCChannel::RacyRPCPolicy
MediateRPCRace(const Message& parent, const Message& child)
{
return MediateRace(parent, child);
}
NS_OVERRIDE
virtual void ActorDestroy(ActorDestroyReason why)
{
if (NormalShutdown != why)
fail("unexpected destruction!");
if (!(mHasReply && mChildHasReply))
fail("both sides should have replies!");
passed("ok");
QuitParent();
}
private:
void OnRaceTime();
void Test2();
void Test3();
bool mHasReply;
bool mChildHasReply;
bool mAnsweredParent;
};
class TestRPCRacesChild :
public PTestRPCRacesChild
{
public:
TestRPCRacesChild() : mHasReply(false) { }
virtual ~TestRPCRacesChild() { }
protected:
NS_OVERRIDE
virtual bool
RecvStart();
NS_OVERRIDE
virtual bool
AnswerRace(bool* hasRace);
NS_OVERRIDE
virtual bool
AnswerStackFrame();
NS_OVERRIDE
virtual bool
AnswerStackFrame3();
NS_OVERRIDE
virtual bool
RecvWakeup();
NS_OVERRIDE
virtual bool
RecvWakeup3();
NS_OVERRIDE
virtual bool
AnswerChild();
NS_OVERRIDE
virtual mozilla::ipc::RPCChannel::RacyRPCPolicy
MediateRPCRace(const Message& parent, const Message& child)
{
return MediateRace(parent, child);
}
NS_OVERRIDE
virtual void ActorDestroy(ActorDestroyReason why)
{
if (NormalShutdown != why)
fail("unexpected destruction!");
QuitChild();
}
private:
bool mHasReply;
};
} // namespace _ipdltest
} // namespace mozilla
#endif // ifndef mozilla__ipdltest_TestRPCRaces_h