public void testMessageWithDestroy() throws IOException, JSONException {
StringWriter stringWriter = new StringWriter();
PrintWriter printWriter = new PrintWriter(stringWriter);
ProtocolMessageWriter writer = new JsonMessageWriter(printWriter);
Display display = new Display();
Shell shell = new Shell(display);
Button button = new Button(shell, SWT.PUSH);
writer.addDestroyPayload(WidgetUtil.getId(button));
String widgetId = WidgetUtil.getId(button);
String actual = stringWriter.getBuffer().toString();
JSONObject message = new JSONObject(actual + "]}");
JSONArray widgetArray = message.getJSONArray(IProtocolConstants.MESSAGE_WIDGETS);
JSONObject widgetObject = widgetArray.getJSONObject(0);
String type = widgetObject.getString(IProtocolConstants.WIDGETS_TYPE);
assertEquals(IProtocolConstants.PAYLOAD_DESTROY, type);
String actualId = widgetObject.getString(IProtocolConstants.WIDGETS_ID);
assertEquals(widgetId, actualId);
Object payload = widgetObject.get(IProtocolConstants.WIDGETS_PAYLOAD);
assertEquals(JSONObject.NULL, payload);
writer.addDestroyPayload(WidgetUtil.getId(shell));
writer.finish();
String shellId = WidgetUtil.getId(shell);
actual = stringWriter.getBuffer().toString();
message = new JSONObject(actual);
widgetArray = message.getJSONArray(IProtocolConstants.MESSAGE_WIDGETS);
widgetObject = widgetArray.getJSONObject(1);
type = widgetObject.getString(IProtocolConstants.WIDGETS_TYPE);
assertEquals(IProtocolConstants.PAYLOAD_DESTROY, type);
actualId = widgetObject.getString(IProtocolConstants.WIDGETS_ID);
assertEquals(shellId, actualId);
payload = widgetObject.get(IProtocolConstants.WIDGETS_PAYLOAD);
assertEquals(JSONObject.NULL, payload);
}
