def test_media_type_mismatch(self):
"""Tests calling IngestTriggerCondition.is_condition_met() with a mismatched media type"""
condition = IngestTriggerCondition('application/json', None)
source_file = source_test_utils.create_source(media_type='text/plain')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_no_conditions(self):
"""Tests calling IngestTriggerCondition.is_condition_met() with no conditions"""
condition = IngestTriggerCondition(None, None)
source_file = source_test_utils.create_source(media_type='text/plain')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_media_type_match(self):
"""Tests calling IngestTriggerCondition.is_condition_met() with a matching media type"""
condition = IngestTriggerCondition('text/plain', None)
source_file = source_test_utils.create_source(media_type='text/plain')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_media_type_mismatch(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with a mismatched media type'''
condition = IngestTriggerCondition('application/json', None)
source_file = source_test_utils.create_source(media_type='text/plain')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_media_type_match(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with a matching media type'''
condition = IngestTriggerCondition('text/plain', None)
source_file = source_test_utils.create_source(media_type='text/plain')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_no_conditions(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with no conditions'''
condition = IngestTriggerCondition(None, None)
source_file = source_test_utils.create_source(media_type='text/plain')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_does_not_have_data_types(self):
"""Tests calling IngestTriggerCondition.is_condition_met() with a source file that does not have all required data types"""
condition = IngestTriggerCondition(None, set(['A', 'B', 'C']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_does_not_have_data_types(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with a source file that does not have all required data types'''
condition = IngestTriggerCondition(None, set(['A', 'B', 'C']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_data_types_incorrect(self):
"""Tests calling IngestTriggerCondition.is_condition_met() with a source file that only has the correct media type"""
condition = IngestTriggerCondition('text/plain', set(['A', 'B', 'C', 'D']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
source_file.add_data_type_tag('C')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_both_correct(self):
"""Tests calling IngestTriggerCondition.is_condition_met() with a source file that meets both criteria"""
condition = IngestTriggerCondition('text/plain', set(['A', 'B', 'C']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
source_file.add_data_type_tag('C')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_both_correct(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with a source file that meets both criteria'''
condition = IngestTriggerCondition('text/plain', set(['A', 'B', 'C']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
source_file.add_data_type_tag('C')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_data_types_incorrect(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with a source file that only has the correct media type'''
condition = IngestTriggerCondition('text/plain',
set(['A', 'B', 'C', 'D']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
source_file.add_data_type_tag('C')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_has_any_data_types(self):
"""
Tests calling IngestTriggerCondition.is_condition_met() with a source file that has all required data types
"""
condition = IngestTriggerCondition(None, set([]), set(['A', 'B', 'C']),
set([]))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('B')
self.assertEqual(condition.is_condition_met(source_file), True)
def test_no_any_data_types(self):
"""
Tests calling IngestTriggerCondition.is_condition_met() with a source file that has no match with
any_of_data_types
"""
condition = IngestTriggerCondition(None, set([]), set(['A', 'B', 'C']),
set([]))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('F')
self.assertEqual(condition.is_condition_met(source_file), False)
def test_has_data_types(self):
'''Tests calling IngestTriggerCondition.is_condition_met() with a source file that has all required data types'''
condition = IngestTriggerCondition(None, set(['A', 'B', 'C']))
source_file = source_test_utils.create_source(media_type='text/plain')
source_file.add_data_type_tag('A')
source_file.add_data_type_tag('B')
source_file.add_data_type_tag('C')
source_file.add_data_type_tag('D')
source_file.add_data_type_tag('E')
self.assertEqual(condition.is_condition_met(source_file), True)
def get_condition(self):
"""Returns the condition for this ingest trigger rule
:return: The trigger condition
:rtype: :class:`ingest.triggers.ingest_trigger_condition.IngestTriggerCondition`
"""
media_type = None
if self._dict['condition']['media_type']:
media_type = self._dict['condition']['media_type']
data_types = set(self._dict['condition']['data_types'])
return IngestTriggerCondition(media_type, data_types)
