Add support for different type of mocks in easy mocks.