updating test cases with mock objects