Add testing for data privider