fixed compile errors of test units by updating mock classes