Added lib mock_wrapper and tests