def test_plugin_init_bad(self, data):
"""Tests plugin_init using an invalid set of values"""
ocs._logger = MagicMock()
with pytest.raises(Exception):
ocs.plugin_init(data)
def test_plugin_init_good(self):
"""Tests plugin_init using a good set of values"""
ocs._logger = MagicMock()
# Used to check the conversions
data = {
"stream_id": {
"value": 1
},
"_CONFIG_CATEGORY_NAME":
module_sp.SendingProcess._CONFIG_CATEGORY_NAME,
"URL": {
"value": "test_URL"
},
"producerToken": {
"value": "test_producerToken"
},
"OMFMaxRetry": {
"value": "100"
},
"OMFRetrySleepTime": {
"value": "100"
},
"OMFHttpTimeout": {
"value": "100"
},
"StaticData": {
"value":
json.dumps({
"Location": "Palo Alto",
"Company": "Dianomic"
})
},
'sending_process_instance': MagicMock(spec=SendingProcess),
"formatNumber": {
"value": "float64"
},
"formatInteger": {
"value": "int32"
},
}
config_default_omf_types = ocs._CONFIG_DEFAULT_OMF_TYPES
config_default_omf_types["type-id"]["value"] = "0001"
with patch.object(data['sending_process_instance'],
'_fetch_configuration',
return_value=config_default_omf_types):
config = ocs.plugin_init(data)
assert config[
'_CONFIG_CATEGORY_NAME'] == module_sp.SendingProcess._CONFIG_CATEGORY_NAME
assert config['URL'] == "test_URL"
assert config['producerToken'] == "test_producerToken"
assert config['OMFMaxRetry'] == 100
assert config['OMFRetrySleepTime'] == 100
assert config['OMFHttpTimeout'] == 100
# Check conversion from String to Dict
assert isinstance(config['StaticData'], dict)
def test_plugin_init_good(self):
"""Tests plugin_init using a good set of values"""
ocs._logger = MagicMock()
# Used to check the conversions
data = {
"stream_id": {
"value": 1
},
"_CONFIG_CATEGORY_NAME":
module_sp.SendingProcess._CONFIG_CATEGORY_NAME,
"URL": {
"value": "test_URL"
},
"producerToken": {
"value": "test_producerToken"
},
"OMFMaxRetry": {
"value": "100"
},
"OMFRetrySleepTime": {
"value": "100"
},
"OMFHttpTimeout": {
"value": "100"
},
"StaticData": {
"value":
json.dumps({
"Location": "Palo Alto",
"Company": "Dianomic"
})
},
"destination_type": {
"value": "3"
},
'sending_process_instance': MagicMock(spec=SendingProcess),
"formatNumber": {
"value": "float64"
},
"formatInteger": {
"value": "int64"
},
"notBlockingErrors": {
"value": "{'id': 400, 'message': 'none'}"
},
"compression": {
"value": "true"
},
"namespace": {
"value": "ocs_namespace_0001"
},
"tenant_id": {
"value": "ocs_tenant_id"
},
"client_id": {
"value": "ocs_client_id"
},
"client_secret": {
"value": "ocs_client_secret"
},
}
config_default_omf_types = ocs._CONFIG_DEFAULT_OMF_TYPES
config_default_omf_types["type-id"]["value"] = "0001"
data["debug_level"] = None
data["log_performance"] = None
data["destination_id"] = 1
data["stream_id"] = 1
with patch.object(data['sending_process_instance'],
'_fetch_configuration',
return_value=config_default_omf_types):
config = ocs.plugin_init(data)
assert config[
'_CONFIG_CATEGORY_NAME'] == module_sp.SendingProcess._CONFIG_CATEGORY_NAME
assert config['URL'] == "test_URL"
assert config['producerToken'] == "test_producerToken"
assert config['OMFMaxRetry'] == 100
assert config['OMFRetrySleepTime'] == 100
assert config['OMFHttpTimeout'] == 100
# Check conversion from String to Dict
assert isinstance(config['StaticData'], dict)