com.allanbank.mongodb.client.connection

Class MockMongoDBServer

java.lang.Object

java.lang.Thread

com.allanbank.mongodb.client.connection.MockMongoDBServer

All Implemented Interfaces:

Runnable

public class MockMongoDBServer
extends Thread

Provides a simple single threaded socket server to act as a MongoDB server in tests. The server collects all messages it receives and can be loaded with replies to the requests it receives.

Copyright ©:

2011, Allanbank Consulting, Inc., All Rights Reserved

Nested Class Summary

Nested classes/interfaces inherited from class java.lang.Thread

Thread.State, Thread.UncaughtExceptionHandler

Field Summary

Fields

Modifier and Type	Field and Description
static byte[]	EMPTY_BYTES An empty Array of bytes.
protected int	myClientConnected Set to true when a client is connected.

Fields inherited from class java.lang.Thread

MAX_PRIORITY, MIN_PRIORITY, NORM_PRIORITY

Constructor Summary

Constructors

Constructor and Description
MockMongoDBServer() Creates a new MockMongoDBServer.

Method Summary

Methods

Modifier and Type	Method and Description
void	clear() Clears the requests received and replies to send.
void	close() Closes the server socket.
protected void	close(Socket socket) Closes the Socket and logs any error.
boolean	disconnectClient() Disconnects any active client..
InetSocketAddress	getInetSocketAddress() Returns the address for the server.
List<Reply>	getReplies() Returns the replies that will be returned after each message is received.
List<Message>	getRequests() Returns the requests that have been received.
protected void	handleClient(Socket clientSocket) Handles a single client connection.
boolean	isRunning() Returns if the server is running.
protected Header	readHeader(BsonInputStream bin) Receives a single message from the connection.
protected Message	readMessage(Header header, BsonInputStream bin) Receives a single message from the connection.
void	run() Runs the server loop waiting for connections and servicing a single client until it exits.
void	setReplies(List<Reply> replies) Sets the replies to return after each message is received.
void	setReplies(Reply... replies) Sets the replies to return after each message is received.
void	setRunning(boolean running) Controls if the server is running.
protected void	sleep() Yawn - go to slepp.
boolean	waitForClient(long timeout) Waits for a client to connect.
boolean	waitForDisconnect(long timeout) Waits for a client to disconnect.
boolean	waitForRequest(int count, long timeout) Waits for a client request.

Methods inherited from class java.lang.Thread

activeCount, checkAccess, clone, countStackFrames, currentThread, destroy, dumpStack, enumerate, getAllStackTraces, getContextClassLoader, getDefaultUncaughtExceptionHandler, getId, getName, getPriority, getStackTrace, getState, getThreadGroup, getUncaughtExceptionHandler, holdsLock, interrupt, interrupted, isAlive, isDaemon, isInterrupted, join, join, join, resume, setContextClassLoader, setDaemon, setDefaultUncaughtExceptionHandler, setName, setPriority, setUncaughtExceptionHandler, sleep, sleep, start, stop, stop, suspend, toString, yield

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Field Detail

EMPTY_BYTES

public static final byte[] EMPTY_BYTES

An empty Array of bytes.

myClientConnected

protected int myClientConnected

Set to true when a client is connected.

Constructor Detail

MockMongoDBServer

public MockMongoDBServer()
                  throws IOException

Creates a new MockMongoDBServer.

Throws:

IOException - On a failure creating the server socket.

Method Detail

clear

public void clear()

Clears the requests received and replies to send.

close

public void close()
           throws IOException

Closes the server socket.

Throws:

IOException - On a failure closing the server socket.

disconnectClient

public boolean disconnectClient()

Disconnects any active client..

Returns:

True if a client is connected, false otherwise.

getInetSocketAddress

public InetSocketAddress getInetSocketAddress()

Returns the address for the server.

Returns:

The address for the server.

getReplies

public List<Reply> getReplies()

Returns the replies that will be returned after each message is received.

Returns:

the replies to return.

getRequests

public List<Message> getRequests()

Returns the requests that have been received.

Returns:

the requests received.

isRunning

public boolean isRunning()

Returns if the server is running.

Returns:

the running

run

public void run()

Runs the server loop waiting for connections and servicing a single client until it exits.

Specified by:

run in interface Runnable

Overrides:

run in class Thread

setReplies

public void setReplies(List<Reply> replies)

Sets the replies to return after each message is received.

Parameters:

replies - the replies to send

setReplies

public void setReplies(Reply... replies)

Sets the replies to return after each message is received.

Parameters:

replies - the replies to send

setRunning

public void setRunning(boolean running)

Controls if the server is running.

Parameters:

running - the running to set

waitForClient

public boolean waitForClient(long timeout)

Waits for a client to connect.

Parameters:

timeout - Time to wait (in milliseconds) for the disconnect.

Returns:

True if a client is connected, false on timeout.

waitForDisconnect

public boolean waitForDisconnect(long timeout)

Waits for a client to disconnect.

Parameters:

timeout - Time to wait (in milliseconds) for the disconnect.

Returns:

True if a client is disconnected, false on timeout.

waitForRequest

public boolean waitForRequest(int count,
                     long timeout)

Waits for a client request.

Parameters:

count - The number of request to wait for.

timeout - Time to wait (in milliseconds) for the disconnect.

Returns:

True if a client is connected, false on timeout.

close

protected void close(Socket socket)

Closes the Socket and logs any error. Sockets do not implement Closeable in Java 6

Parameters:

socket - The connection to close. Sockets do not implement Closeable in Java 6

handleClient

protected void handleClient(Socket clientSocket)
                     throws IOException

Handles a single client connection.

Parameters:

clientSocket - The socket to receive messages from.

Throws:

IOException - On a connection error.

readHeader

protected Header readHeader(BsonInputStream bin)
                     throws IOException

Receives a single message from the connection.

Parameters:

bin - The stream to read the message.

Returns:

The Message received.

Throws:

IOException - On an error receiving the message.

readMessage

protected Message readMessage(Header header,
                  BsonInputStream bin)
                       throws IOException

Receives a single message from the connection.

Parameters:

header - The read message header.

bin - The stream to read the message.

Returns:

The Message received.

Throws:

IOException - On an error receiving the message.

sleep

protected void sleep()

Yawn - go to slepp.