def setUp(self):
tags = {'title': 'One_love', 'artist': 'bob marley', 'album': 'EXODUS'}
self.song = Mock()
self.song.__getitem__ = Mock(side_effect=lambda tag: tags[tag])
self.song.__setitem__ = Mock()
self.song.__contains__ = Mock()
self.humanize_operation = HumanizeOperation()
def from_args(self, args):
name = args.command
if name == 'get':
return GetCommand()
elif name == 'set':
return OperationCommand(SetOperation(args.__dict__))
elif name == 'rename':
template = FreiTemplate(args.format)
return OperationCommand(RenameOperation(template))
elif name == 'extract':
template = FreiTemplate(args.format)
return OperationCommand(ExtractOperation(template))
elif name == 'humanize':
return OperationCommand(HumanizeOperation())
