Add mechanism for mock test