Exemplo n.º 1
0
    def test_rename_folder(self):
        '''
        Test renaming a Folder item.
        '''
        from frostmark import db_base, user_data
        from frostmark.db import get_session
        from frostmark.models import Folder
        from frostmark.editor import Editor

        data = dirname(abspath(user_data.__file__))
        self.assertNotIn(db_base.DB_NAME, listdir(data))

        item = {'id': 321, 'folder_name': 'old parent'}
        session = get_session()
        try:
            session.add(Folder(**item))
            session.commit()

            self.assertEqual(
                session.query(Folder).filter(
                    Folder.folder_name == item['folder_name']).first().id,
                item['id'])

            self.assertEqual(
                session.query(Folder).filter(
                    Folder.id == item['id']).first().folder_name,
                item['folder_name'])

            Editor.rename_folder(folder_id=item['id'],
                                 name=item['folder_name'][:3])

            self.assertEqual(
                session.query(Folder).filter(
                    Folder.id == item['id']).first().folder_name,
                item['folder_name'][:3])
        finally:
            session.close()

        self.assertIn(db_base.DB_NAME, listdir(data))
        remove(join(data, db_base.DB_NAME))
Exemplo n.º 2
0
            parent_id=int(kwargs['arg_values'][1]))))

PARSER.console_parser.add_argument(
    '-rf',
    '--rename-folder',
    help='rename folder',
    required=False,
    nargs=2,
    metavar=('FOLDER_ID', 'NAME'),

    # pylint: disable=unnecessary-lambda
    action=lambda *args, **kwargs: ExecuteAction(
        *args,
        **kwargs,
        func=lambda *args, **kwargs: Editor.rename_folder(
            folder_id=int(kwargs['arg_values'][0]),
            name=kwargs['arg_values'][1])))

PARSER.console_parser.add_argument(
    '-rb',
    '--rename-bookmark',
    help='rename bookmark',
    required=False,
    nargs=2,
    metavar=('BOOKMARK_ID', 'NAME'),

    # pylint: disable=unnecessary-lambda
    action=lambda *args, **kwargs: ExecuteAction(
        *args,
        **kwargs,
        func=lambda *args, **kwargs: Editor.rename_bookmark(