Adding mocks and initial test cases.