Get mocks working, and write first test with mock