Update dfs_unit_test.cpp