Added integration test for DatasetSink.