Adding tests using the mock.