Not know how to write unit test for this feature. Co-authored-by: Yan Chunwei <yanchunwei@outlook.com>