Creating unit tests with mocks