# HG changeset patch
# User Matthew Wild <mwild1@gmail.com>
# Date 1450362333 0
# Node ID e032cdb517ab8c7acd3e4731cae1cc5508f90a2c
# Parent  909305ae72ac174eee62b5c353af259026c98200
client.lua: Handle unexpected disconnects in the middle of a test

diff -r 909305ae72ac -r e032cdb517ab scansion/objects/client.lua
--- a/scansion/objects/client.lua	Thu Dec 17 14:25:09 2015 +0000
+++ b/scansion/objects/client.lua	Thu Dec 17 14:25:33 2015 +0000
@@ -49,6 +49,11 @@
 		client.stream:hook("outgoing-raw", function (s) client.log("Data out: %s", s); end, 1000);
 		-- And incoming, parsed, stanzas
 		client.stream:hook("stanza", function (s) client.log("Stanza: %s", s) end);
+		-- Handle unexpected disconnects
+		client.stream:hook("disconnected", function (s)
+			client.log("Unexpected disconnect!");
+			error("Unexpected disconnect");
+		end);
 	end;
 
 	connects = function (client)