anki-connect/tests/test_decks.py

#!/usr/bin/env python

import unittest
import util


class TestDecks(unittest.TestCase):
    def runTest(self):
        # deckNames (part 1)
        deckNames = util.invoke('deckNames')
        self.assertIn('Default', deckNames)

        # deckNamesAndIds
        result = util.invoke('deckNamesAndIds')
        self.assertIn('Default', result)
        self.assertEqual(result['Default'], 1)

        # createDeck
        util.invoke('createDeck', deck='test1')

        # deckNames (part 2)
        deckNames = util.invoke('deckNames')
        self.assertIn('test1', deckNames)

        # changeDeck
        note = {'deckName': 'test1', 'modelName': 'Basic', 'fields': {'Front': 'front', 'Back': 'back'}, 'tags': ['tag']}
        noteId = util.invoke('addNote', note=note)
        cardIds = util.invoke('findCards', query='deck:test1')
        util.invoke('changeDeck', cards=cardIds, deck='test2')

        # deckNames (part 3)
        deckNames = util.invoke('deckNames')
        self.assertIn('test2', deckNames)

        # deleteDecks
        util.invoke('deleteDecks', decks=['test1', 'test2'], cardsToo=True)

        # deckNames (part 4)
        deckNames = util.invoke('deckNames')
        self.assertNotIn('test1', deckNames)
        self.assertNotIn('test2', deckNames)

        # getDeckConfig
        deckConfig = util.invoke('getDeckConfig', deck='Default')
        self.assertEqual('Default', deckConfig['name'])

        # saveDeckConfig
        deckConfig = util.invoke('saveDeckConfig', config=deckConfig)

        # setDeckConfigId
        setDeckConfigId = util.invoke('setDeckConfigId', decks=['Default'], configId=1)
        self.assertTrue(setDeckConfigId)

        # cloneDeckConfigId (part 1)
        deckConfigId = util.invoke('cloneDeckConfigId', cloneFrom=1, name='test')
        self.assertTrue(deckConfigId)

        # removeDeckConfigId (part 1)
        removedDeckConfigId = util.invoke('removeDeckConfigId', configId=deckConfigId)
        self.assertTrue(removedDeckConfigId)

        # removeDeckConfigId (part 2)
        removedDeckConfigId = util.invoke('removeDeckConfigId', configId=deckConfigId)
        self.assertFalse(removedDeckConfigId)

        # cloneDeckConfigId (part 2)
        deckConfigId = util.invoke('cloneDeckConfigId', cloneFrom=deckConfigId, name='test')
        self.assertFalse(deckConfigId)

        # updateCompleteDeck
        util.invoke('updateCompleteDeck', data={
            'deck': 'test3',
            'cards': {
                '12': {
                    'id': 12, 'nid': 23, 'ord': 0,  'type': 0, 'queue': 0,
                    'due': 1186031, 'factor': 0, 'ivl': 0, 'reps': 0, 'lapses': 0, 'left': 0
                }
            },
            'notes': {
                '23': {
                    'id': 23, 'mid': 34, 'fields': ['frontValue', 'backValue'], 'tags': ['aTag']
                }
            },
            'models': {
                '34': {
                    'id': 34, 'fields': ['Front', 'Back'], 'templateNames': ['Card 1'], 'name': 'anotherModel',
                }
            }
        })
        deckNames = util.invoke('deckNames')
        self.assertIn('test3', deckNames)
        cardIDs = util.invoke('findCards', query='deck:test3')
        self.assertEqual(len(cardIDs), 1)
        self.assertEqual(cardIDs[0], 12)


if __name__ == '__main__':
    unittest.main()