Added a proper mock transport. Updated tests for models and protocol