Add mocks

DO NOT MERGE
=============