Add test snippets and mock gateway calls