Add tests for datasource which will check that these two methods is working or not