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())