Tests: add tests for the data source generator