Add whole mock class test