[DF-4439] update unit test