Add test case that uses Heartbeat extension and callback.

tests/tlstest.py

@@ -857,6 +857,21 @@ def connect():
 
     test_no += 1
 
+    print("Test {0} - Heartbeat extension response callback".format(test_no))
+    synchro.recv(1)
+    connection = connect()
+    settings = HandshakeSettings()
+    heartbeatPayload = os.urandom(100)
+    def heartbeatResponseCheck(message):
+        assert heartbeatPayload == message.payload
+    settings.heartbeat_response_callback = heartbeatResponseCheck
+    connection.handshakeClientCert(serverName=address[0], settings=settings)
+    connection.sendHeartbeatRequest(heartbeatPayload, 16)
+    testConnClient(connection)
+    connection.close()
+
+    test_no += 1
+
     print('Test {0} - good standard XMLRPC https client'.format(test_no))
     address = address[0], address[1]+1
     synchro.recv(1)
@@ -1638,6 +1653,22 @@ def server_bind(self):
 
     test_no += 1
 
+    print("Test {0} - Heartbeat extension response callback".format(test_no))
+    synchro.send(b'R')
+    connection = connect()
+    settings = HandshakeSettings()
+    heartbeatPayload = os.urandom(100)
+    def heartbeatResponseCheck(message):
+        assert heartbeatPayload == message.payload
+    settings.heartbeat_response_callback = heartbeatResponseCheck
+    connection.handshakeServer(certChain=x509Chain, privateKey=x509Key,
+                               settings=settings)
+    connection.sendHeartbeatRequest(heartbeatPayload, 16)
+    testConnServer(connection)
+    connection.close()
+
+    test_no += 1
+
     print("Tests {0}-{1} - XMLRPXC server".format(test_no, test_no + 2))
     test_no += 2