mirror of
https://github.com/ankitects/anki.git
synced 2025-09-19 14:32:22 -04:00
95 lines
2.9 KiB
Python
95 lines
2.9 KiB
Python
# coding: utf-8
|
|
|
|
import nose, os, shutil
|
|
from tests.shared import assertException
|
|
|
|
from anki.errors import *
|
|
from anki import DeckStorage
|
|
from anki.importing import anki10, csv, mnemosyne10
|
|
from anki.stdmodels import BasicModel
|
|
from anki.facts import Fact
|
|
from anki.sync import SyncClient, SyncServer
|
|
|
|
from anki.db import *
|
|
|
|
testDir = os.path.dirname(__file__)
|
|
|
|
def test_csv():
|
|
deck = DeckStorage.Deck()
|
|
deck.addModel(BasicModel())
|
|
file = unicode(os.path.join(testDir, "importing/text-2fields.txt"))
|
|
i = csv.TextImporter(deck, file)
|
|
i.doImport()
|
|
# two problems - missing front, dupe front
|
|
assert len(i.log) == 2
|
|
assert i.total == 5
|
|
deck.s.close()
|
|
|
|
def test_csv_tags():
|
|
deck = DeckStorage.Deck()
|
|
deck.addModel(BasicModel())
|
|
file = unicode(os.path.join(testDir, "importing/text-tags.txt"))
|
|
i = csv.TextImporter(deck, file)
|
|
i.doImport()
|
|
facts = deck.s.query(Fact).all()
|
|
assert len(facts) == 1
|
|
assert facts[0].tags == "baz qux"
|
|
deck.s.close()
|
|
|
|
def test_mnemosyne10():
|
|
deck = DeckStorage.Deck()
|
|
deck.addModel(BasicModel())
|
|
file = unicode(os.path.join(testDir, "importing/test.mem"))
|
|
i = mnemosyne10.Mnemosyne10Importer(deck, file)
|
|
i.doImport()
|
|
assert i.total == 5
|
|
deck.s.close()
|
|
|
|
def test_anki10():
|
|
# though these are not modified, sqlite updates the mtime, so copy to tmp
|
|
# first
|
|
file_ = unicode(os.path.join(testDir, "importing/test10.anki"))
|
|
file = "/tmp/test10.anki"
|
|
shutil.copy(file_, file)
|
|
file2_ = unicode(os.path.join(testDir, "importing/test10-2.anki"))
|
|
file2 = "/tmp/test10-2.anki"
|
|
shutil.copy(file2_, file2)
|
|
deck = DeckStorage.Deck()
|
|
i = anki10.Anki10Importer(deck, file)
|
|
i.doImport()
|
|
assert i.total == 2
|
|
deck.s.rollback()
|
|
deck.close()
|
|
# import a deck into itself - 10-2 is the same as test10, but with one
|
|
# card answered and another deleted. nothing should be synced to client
|
|
deck = DeckStorage.Deck(file)
|
|
i = anki10.Anki10Importer(deck, file2)
|
|
i.doImport()
|
|
assert i.total == 0
|
|
deck.s.rollback()
|
|
|
|
def test_anki10_modtime():
|
|
deck1 = DeckStorage.Deck()
|
|
deck2 = DeckStorage.Deck()
|
|
client = SyncClient(deck1)
|
|
server = SyncServer(deck2)
|
|
client.setServer(server)
|
|
deck1.addModel(BasicModel())
|
|
f = deck1.newFact()
|
|
f['Front'] = u"foo"; f['Back'] = u"bar"
|
|
deck1.addFact(f)
|
|
assert deck1.cardCount == 1
|
|
assert deck2.cardCount == 0
|
|
client.sync()
|
|
assert deck1.cardCount == 1
|
|
assert deck2.cardCount == 1
|
|
file_ = unicode(os.path.join(testDir, "importing/test10-3.anki"))
|
|
file = "/tmp/test10-3.anki"
|
|
shutil.copy(file_, file)
|
|
i = anki10.Anki10Importer(deck1, file)
|
|
i.doImport()
|
|
client.sync()
|
|
assert i.total == 1
|
|
assert deck2.s.scalar("select count(*) from cards") == 2
|
|
assert deck2.s.scalar("select count(*) from facts") == 2
|
|
assert deck2.s.scalar("select count(*) from models") == 2
|