Feature/59 create mock for class