What's the reason not using this.sinon.stub? you won't need to restore in the same test function.