def test_failing_test_module_with_fetch(mocker, demisto_params, cluster_tree, expected_failure):
"""
Given:
- initialized KafkaCommunicator
When:
- testing the module with fetch without offset
Then:
- Assert the relevant error is raised when the fetch parameters are bad.
"""
mocker.patch.object(KConsumer, '__init__', return_value=None)
mocker.patch.object(KProducer, '__init__', return_value=None)
cluster_metadata = create_cluster_metadata(cluster_tree)
mocker.patch.object(KConsumer, 'list_topics', return_value=cluster_metadata)
mocker.patch.object(KProducer, 'list_topics', return_value=cluster_metadata)
with pytest.raises(DemistoException) as exception_info:
command_test_module(KAFKA, demisto_params)
assert expected_failure in str(exception_info.value)
def test_failing_simple_test_module(mocker):
"""
Given:
- initialized KafkaCommunicator
When:
- testing the module without fetch
Then:
- Assert relevant error is raised if communication failed.
"""
mocker.patch.object(KConsumer, '__init__', return_value=None)
mocker.patch.object(KProducer, '__init__', return_value=None)
def raise_kafka_error():
raise Exception('Some connection error')
mocker.patch.object(KConsumer, 'list_topics', return_value=ClusterMetadata(), side_effect=raise_kafka_error)
mocker.patch.object(KProducer, 'list_topics', return_value=ClusterMetadata(), side_effect=raise_kafka_error)
with pytest.raises(DemistoException) as exception_info:
command_test_module(KAFKA, {'isFetch': False})
assert 'Error connecting to kafka' in str(exception_info.value)
def test_passing_simple_test_module(mocker):
"""
Given:
- initialized KafkaCommunicator
When:
- testing the module without fetch
Then:
- Assert 'ok' if ClusterMetadata object is returned from Kafka
"""
mocker.patch.object(KafkaV3, 'KConsumer')
mocker.patch.object(KafkaV3, 'KProducer')
mocker.patch.object(KConsumer, 'list_topics', return_value=ClusterMetadata())
mocker.patch.object(KProducer, 'list_topics', return_value=ClusterMetadata())
assert command_test_module(KAFKA, {'isFetch': False}) == 'ok'
def test_passing_test_module_with_fetch(mocker, demisto_params, cluster_tree):
"""
Given:
- initialized KafkaCommunicator
When:
- testing the module with fetch without offset
Then:
- Assert everything is 'ok'
"""
mocker.patch.object(KConsumer, '__init__', return_value=None)
mocker.patch.object(KProducer, '__init__', return_value=None)
cluster_metadata = create_cluster_metadata(cluster_tree)
mocker.patch.object(KConsumer, 'list_topics', return_value=cluster_metadata)
mocker.patch.object(KProducer, 'list_topics', return_value=cluster_metadata)
assert command_test_module(KAFKA, demisto_params) == 'ok'