xref: /haiku/src/tests/kits/app/bmessagerunner/MessageRunnerTestHelpers.h (revision dace24c6255b30ff7f266dd08e43441f517851d3)

// MessageRunnerTestHelpers.h

#ifndef MESSAGE_RUNNER_TEST_HELPERS_H
#define MESSAGE_RUNNER_TEST_HELPERS_H

#include <Application.h>
#include <List.h>

enum {
	MSG_RUNNER_MESSAGE	= 'rmsg',
	MSG_REPLY			= 'rply',
};

// MessageRunnerTestHandler
class MessageRunnerTestHandler : public BHandler {
public:
	MessageRunnerTestHandler();
	~MessageRunnerTestHandler();

	virtual void MessageReceived(BMessage *message);

	int32 CountReplies() const { return fReplyCount; }

private:
	int32					fReplyCount;
};

// MessageRunnerTestLooper
class MessageRunnerTestLooper : public BLooper {
public:
	MessageRunnerTestLooper();
	~MessageRunnerTestLooper();

	virtual void MessageReceived(BMessage *message);

	bool CheckMessages(bigtime_t startTime, bigtime_t interval, int32 count);
	bool CheckMessages(int32 skip, bigtime_t startTime, bigtime_t interval,
					   int32 count);

private:
	struct MessageInfo;

private:
	MessageInfo *MessageInfoAt(int32 index) const;

private:
	BList	fMessageInfos;
};

// MessageRunnerTestApp
class MessageRunnerTestApp : public BApplication {
public:
	MessageRunnerTestApp(const char *signature);
	~MessageRunnerTestApp();

	virtual void MessageReceived(BMessage *message);
	virtual bool QuitRequested();

	int32 CountReplies() const { return fReplyCount; }

	MessageRunnerTestLooper *TestLooper() const { return fLooper; }
	MessageRunnerTestHandler *TestHandler() const { return fHandler; }

private:
	static int32 _ThreadEntry(void *data);

private:
	thread_id					fThread;
	int32						fReplyCount;
	MessageRunnerTestLooper		*fLooper;
	MessageRunnerTestHandler	*fHandler;
};

#endif	// MESSAGE_RUNNER_TEST_HELPERS_H