Add support for connection close message
Changes in test files. Check them before run to set parameters