add data layer unit tests