Add unit tests for DataService.