test heartbeat logic

Author: rade

test/src/com/rabbitmq/client/test/functional/FunctionalTests.java

@@ -38,6 +38,7 @@
 public class FunctionalTests extends TestCase {
     public static TestSuite suite() {
         TestSuite suite = new TestSuite("functional");
+        suite.addTestSuite(Heartbeat.class);
         suite.addTestSuite(Tables.class);
         suite.addTestSuite(DoubleDeletion.class);
         suite.addTestSuite(Routing.class);

test/src/com/rabbitmq/client/test/functional/Heartbeat.java

@@ -0,0 +1,58 @@
+//   The contents of this file are subject to the Mozilla Public License
+//   Version 1.1 (the "License"); you may not use this file except in
+//   compliance with the License. You may obtain a copy of the License at
+//   http://www.mozilla.org/MPL/
+//
+//   Software distributed under the License is distributed on an "AS IS"
+//   basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the
+//   License for the specific language governing rights and limitations
+//   under the License.
+//
+//   The Original Code is RabbitMQ.
+//
+//   The Initial Developers of the Original Code are LShift Ltd,
+//   Cohesive Financial Technologies LLC, and Rabbit Technologies Ltd.
+//
+//   Portions created before 22-Nov-2008 00:00:00 GMT by LShift Ltd,
+//   Cohesive Financial Technologies LLC, or Rabbit Technologies Ltd
+//   are Copyright (C) 2007-2008 LShift Ltd, Cohesive Financial
+//   Technologies LLC, and Rabbit Technologies Ltd.
+//
+//   Portions created by LShift Ltd are Copyright (C) 2007-2010 LShift
+//   Ltd. Portions created by Cohesive Financial Technologies LLC are
+//   Copyright (C) 2007-2010 Cohesive Financial Technologies
+//   LLC. Portions created by Rabbit Technologies Ltd are Copyright
+//   (C) 2007-2010 Rabbit Technologies Ltd.
+//
+//   All Rights Reserved.
+//
+//   Contributor(s): ______________________________________.
+//
+
+package com.rabbitmq.client.test.functional;
+
+import java.io.IOException;
+
+import com.rabbitmq.client.impl.AMQConnection;
+import com.rabbitmq.client.test.BrokerTestCase;
+
+public class Heartbeat extends BrokerTestCase {
+
+    public Heartbeat()
+    {
+        super();
+        connectionFactory.setRequestedHeartbeat(1);
+    }
+
+    public void testHeartbeat()
+        throws IOException, InterruptedException
+    {
+        Thread.sleep(3100);
+        assertTrue(connection.isOpen());
+        ((AMQConnection)connection).setHeartbeat(0);
+        Thread.sleep(3100);
+        assertFalse(connection.isOpen());
+
+    }
+
+}