Feature/remove rspec mock warnings

Hi Martyn, I came across your gem which we are using on a commercial project. I'd like to contribute, can you please review my pull request, it's a starting point for me.