This looks good, but just wondering if the mocking/stubing is for speed or hiding abstraction.
Can we replace it with assertions