mirror of
https://github.com/datahub-project/datahub.git
synced 2025-07-13 04:02:34 +00:00
28 lines
1.2 KiB
Python
28 lines
1.2 KiB
Python
|
import unittest
|
||
|
|
from unittest.mock import patch, MagicMock
|
||
|
|
|
||
|
|
from gometa.ingestion.run.pipeline import Pipeline
|
||
|
|
|
||
|
|
|
||
|
|
class PipelineTest(unittest.TestCase):
|
||
|
|
|
||
|
|
@patch("gometa.ingestion.extractor.kafka.KafkaMetadataExtractor")
|
||
|
|
@patch("gometa.ingestion.source.kafka.KafkaSource")
|
||
|
|
@patch("gometa.ingestion.sink.kafka.KafkaSink")
|
||
|
|
def test_configure(self, mock_sink, mock_source, mock_extractor):
|
||
|
|
pipeline = Pipeline()
|
||
|
|
pipeline.configure({'source': {
|
||
|
|
'type': 'kafka',
|
||
|
|
'extractor': 'gometa.ingestion.source.kafka.KafkaMetadataExtractor',
|
||
|
|
'kafka': {
|
||
|
|
'bootstrap': "localhost:9092"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
'sink': {
|
||
|
|
'type': 'kafka'
|
||
|
|
}})
|
||
|
|
mock_source.assert_called_once()
|
||
|
|
mock_sink.assert_called_once()
|
||
|
|
|
||
|
|
|