Add protocol tests