[FEATURE] Added mock interface within a trait