Added tests for the protocol