Added tests to check custom message behavior.