add friend class for unit test