anki-connect/tests/test_decks.py

#!/usr/bin/env python

import unittest
import util


class TestDeckNames(unittest.TestCase):
    def runTest(self):
        result = util.invoke('deckNames')
        self.assertIn('Default', result)


class TestDeckNamesAndIds(unittest.TestCase):
    def runTest(self):
        result = util.invoke('deckNamesAndIds')
        self.assertIn('Default', result)
        self.assertEqual(result['Default'], 1)


class TestCreateDeck(unittest.TestCase):
    def tearDown(self):
        util.invoke('deleteDecks', {'decks': ['test']})


    def runTest(self):
        util.invoke('createDeck', {'deck': 'test'})
        self.assertIn('test', util.invoke('deckNames'))


class TestDeleteDecks(unittest.TestCase):
    def setUp(self):
        util.invoke('createDeck', {'deck': 'test'})


    def runTest(self):
        util.invoke('deleteDecks', {'decks': ['test']})
        self.assertNotIn('test', util.invoke('deckNames'))


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