From 1840efb60fd08f83cc629ff560068f4644f48a6b Mon Sep 17 00:00:00 2001
From: lotodore <pokerth@lotharmay.de>
Date: Thu, 17 Feb 2011 12:09:05 +0000
Subject: [PATCH] Add new test case for unsubscribing/resubscribing lobby
 messages.

---
 .../pokerth_test/LobbySubscriptionTest.java   | 99 +++++++++++++++++++
 1 file changed, 99 insertions(+)
 create mode 100644 tests/src/pokerth_test/LobbySubscriptionTest.java

diff --git a/tests/src/pokerth_test/LobbySubscriptionTest.java b/tests/src/pokerth_test/LobbySubscriptionTest.java
new file mode 100644
index 00000000..93d9b28b
--- /dev/null
+++ b/tests/src/pokerth_test/LobbySubscriptionTest.java
@@ -0,0 +1,99 @@
+package pokerth_test;
+
+import static org.junit.Assert.assertTrue;
+import static org.junit.Assert.fail;
+
+import java.net.Socket;
+import java.util.ArrayList;
+import java.util.Collection;
+
+import org.junit.Test;
+
+import pokerth_protocol.NetGameInfo;
+import pokerth_protocol.PokerTHMessage;
+import pokerth_protocol.SubscriptionRequestMessage;
+import pokerth_protocol.NetGameInfo.EndRaiseModeEnumType;
+import pokerth_protocol.NetGameInfo.NetGameTypeEnumType;
+import pokerth_protocol.SubscriptionRequestMessage.SubscriptionRequestMessageSequenceType;
+import pokerth_protocol.SubscriptionRequestMessage.SubscriptionRequestMessageSequenceType.SubscriptionActionEnumType;
+
+
+public class LobbySubscriptionTest extends TestBase {
+
+	@Test
+	public void testLobbySubscription() throws Exception {
+		guestInit();
+
+		PokerTHMessage msg;
+		msg = receiveMessage();
+		if (!msg.isPlayerListMessageSelected()) {
+			failOnErrorMessage(msg);
+			fail("Invalid message.");
+		}
+
+		SubscriptionRequestMessageSequenceType subscriptionType = new SubscriptionRequestMessageSequenceType();
+		SubscriptionActionEnumType action = new SubscriptionActionEnumType();
+		action.setValue(SubscriptionActionEnumType.EnumType.unsubscribeGameList);
+		subscriptionType.setSubscriptionAction(action);
+		SubscriptionRequestMessage subscriptionRequest = new SubscriptionRequestMessage();
+		subscriptionRequest.setValue(subscriptionType);
+		msg = new PokerTHMessage();
+		msg.selectSubscriptionRequestMessage(subscriptionRequest);
+		sendMessage(msg);
+
+		// Create a new game.
+		Collection<Integer> l = new ArrayList<Integer>();
+		NetGameInfo gameInfo = createGameInfo(5, EndRaiseModeEnumType.EnumType.doubleBlinds, 0, 100, GuestUser + " game list normal game", l, 10, 0, 2, 2000);
+		sendMessage(createGameRequestMsg(
+				gameInfo,
+				NetGameTypeEnumType.EnumType.normalGame,
+				10,
+				5,
+				"",
+				false));
+
+		// No game list message should be sent by the server.
+		// Next message is join game ack.
+		msg = receiveMessage();
+		assertTrue(msg.isJoinGameReplyMessageSelected());
+		assertTrue(msg.getJoinGameReplyMessage().getValue().getJoinGameResult().isJoinGameAckSelected());
+		long gameId = msg.getJoinGameReplyMessage().getValue().getGameId().getValue().longValue();
+
+		Socket s[] = new Socket[9];
+		for (int i = 0; i < 9; i++) {
+			s[i] = new Socket("localhost", 7234);
+			String username = "test" + (i+1);
+			String password = username;
+			userInit(s[i], username, password);
+			sendMessage(joinGameRequestMsg(gameId, "", false), s[i]);
+		}
+
+		// No game list message should be received.
+		do {
+			msg = receiveMessage();
+			if (msg.isGameListMessageSelected() || msg.isPlayerListMessageSelected()) {
+				fail("Game/player list messages are switched off!");
+			}
+		} while (!msg.isStartEventMessageSelected());
+
+		// Resubscribe game list
+		subscriptionType = new SubscriptionRequestMessageSequenceType();
+		action = new SubscriptionActionEnumType();
+		action.setValue(SubscriptionActionEnumType.EnumType.resubscribeGameList);
+		subscriptionType.setSubscriptionAction(action);
+		subscriptionRequest = new SubscriptionRequestMessage();
+		subscriptionRequest.setValue(subscriptionType);
+		msg = new PokerTHMessage();
+		msg.selectSubscriptionRequestMessage(subscriptionRequest);
+		sendMessage(msg);
+
+		// Next messages should player list messages for all 10 players.
+		for (int i = 0; i < 10; i++) {
+			msg = receiveMessage();
+			assertTrue(msg.isPlayerListMessageSelected());
+		}
+		// Now there should be one game list message.
+		msg = receiveMessage();
+		assertTrue(msg.isGameListMessageSelected());
+	}
+}