Add unit test to DataManager