[DF-566] update unit tests.