Adding some tests; since we're mocking, the tests weren't actually checking this code.