Test cases added for mock impl