def test__add_tags_empty(self):
testpipeline = FFmpegPipeline(1, {"template": None, "type": "ffmpeg"}, "models")
testpipeline.request = {}
iemetadata_args = []
testpipeline._add_tags(iemetadata_args)
self.assertEqual(iemetadata_args, [])
def test__add_tags_error(self, mock_log_error):
testpipeline = FFmpegPipeline(1, {
"template": None,
"type": "ffmpeg"
}, "models")
testpipeline.request = {"tags": {"tag1": "value1", "tag2": "value2"}}
expected_iemetadata_args = [
'-custom_tag', 'tag1:value1,', '-custom_tag', 'tag2:value2'
]
testpipeline._add_tags(None)
mock_log_error.assert_called_with("Error adding tags")
def test__add_tags(self):
testpipeline = FFmpegPipeline(1, {
"template": None,
"type": "ffmpeg"
}, "models")
testpipeline.request = {"tags": {"tag1": "value1", "tag2": "value2"}}
iemetadata_args = []
expected_iemetadata_args = [
'-custom_tag', 'tag1:value1,', '-custom_tag', 'tag2:value2'
]
testpipeline._add_tags(iemetadata_args)
self.assertEqual(iemetadata_args, expected_iemetadata_args)
def test__add_tags(self):
testpipeline = FFmpegPipeline(1, {
"template": None,
"type": "ffmpeg"
}, "models")
testpipeline.request = {"tags": {"tag1": "value1", "tag2": "value2"}}
iemetadata_args = []
testpipeline._add_tags(iemetadata_args)
self.assertEqual(str(iemetadata_args).count('-custom_tag'), 2)
self.assertRegex(str(iemetadata_args), "tag.:value.,")
self.assertRegex(str(iemetadata_args), "tag.:value.\'")
self.assertIn('tag1:value1', str(iemetadata_args))
self.assertIn('tag2:value2', str(iemetadata_args))
