anki/pylib/tests/test_collection.py
Damien Elmes ce243c2cae Simplify note adding and the deck/notetype choosers
The existing code was really difficult to reason about:

- The default notetype depended on the selected deck, and vice versa,
and this logic was buried in the deck and notetype choosing screens,
and models.py.
- Changes to the notetype were not passed back directly, but were fired
via a hook, which changed any screen in the app that had a notetype
selector.

It also wasn't great for performance, as the most recent deck and tags
were embedded in the notetype, which can be expensive to save and sync
for large notetypes.

To address these points:

- The current deck for a notetype, and notetype for a deck, are now
stored in separate config variables, instead of directly in the deck
or notetype. These are cheap to read and write, and we'll be able to
sync them individually in the future once config syncing is updated in
the future. I seem to recall some users not wanting the tag saving
behaviour, so I've dropped that for now, but if people end up missing
it, it would be simple to add as an extra auxiliary config variable.
- The logic for getting the starting deck and notetype has been moved
into the backend. It should be the same as the older Python code, with
one exception: when "change deck depending on notetype" is enabled in
the preferences, it will start with the current notetype ("curModel"),
instead of first trying to get a deck-specific notetype.
- ModelChooser has been duplicated into notetypechooser.py, and it
has been updated to solely be concerned with keeping track of a selected
notetype - it no longer alters global state.
2021-03-10 11:53:27 +10:00

172 lines
4.5 KiB
Python

# coding: utf-8
import os
import tempfile
from anki import Collection as aopen
from anki.dbproxy import emulate_named_args
from anki.lang import TR, without_unicode_isolation
from anki.stdmodels import addBasicModel, get_stock_notetypes
from anki.utils import isWin
from tests.shared import assertException, getEmptyCol
def test_create_open():
(fd, path) = tempfile.mkstemp(suffix=".anki2", prefix="test_attachNew")
try:
os.close(fd)
os.unlink(path)
except OSError:
pass
col = aopen(path)
# for open()
newPath = col.path
newMod = col.mod
col.close()
del col
# reopen
col = aopen(newPath)
assert col.mod == newMod
col.close()
# non-writeable dir
if isWin:
dir = "c:\root.anki2"
else:
dir = "/attachroot.anki2"
assertException(Exception, lambda: aopen(dir))
# reuse tmp file from before, test non-writeable file
os.chmod(newPath, 0)
assertException(Exception, lambda: aopen(newPath))
os.chmod(newPath, 0o666)
os.unlink(newPath)
def test_noteAddDelete():
col = getEmptyCol()
# add a note
note = col.newNote()
note["Front"] = "one"
note["Back"] = "two"
n = col.addNote(note)
assert n == 1
# test multiple cards - add another template
m = col.models.current()
mm = col.models
t = mm.newTemplate("Reverse")
t["qfmt"] = "{{Back}}"
t["afmt"] = "{{Front}}"
mm.addTemplate(m, t)
mm.save(m)
assert col.cardCount() == 2
# creating new notes should use both cards
note = col.newNote()
note["Front"] = "three"
note["Back"] = "four"
n = col.addNote(note)
assert n == 2
assert col.cardCount() == 4
# check q/a generation
c0 = note.cards()[0]
assert "three" in c0.q()
# it should not be a duplicate
assert not note.duplicate_or_empty()
# now let's make a duplicate
note2 = col.newNote()
note2["Front"] = "one"
note2["Back"] = ""
assert note2.duplicate_or_empty()
# empty first field should not be permitted either
note2["Front"] = " "
assert note2.duplicate_or_empty()
def test_fieldChecksum():
col = getEmptyCol()
note = col.newNote()
note["Front"] = "new"
note["Back"] = "new2"
col.addNote(note)
assert col.db.scalar("select csum from notes") == int("c2a6b03f", 16)
# changing the val should change the checksum
note["Front"] = "newx"
note.flush()
assert col.db.scalar("select csum from notes") == int("302811ae", 16)
def test_addDelTags():
col = getEmptyCol()
note = col.newNote()
note["Front"] = "1"
col.addNote(note)
note2 = col.newNote()
note2["Front"] = "2"
col.addNote(note2)
# adding for a given id
col.tags.bulk_add([note.id], "foo")
note.load()
note2.load()
assert "foo" in note.tags
assert "foo" not in note2.tags
# should be canonified
col.tags.bulk_add([note.id], "foo aaa")
note.load()
assert note.tags[0] == "aaa"
assert len(note.tags) == 2
def test_timestamps():
col = getEmptyCol()
assert len(col.models.all_names_and_ids()) == len(get_stock_notetypes(col))
for i in range(100):
addBasicModel(col)
assert len(col.models.all_names_and_ids()) == 100 + len(get_stock_notetypes(col))
def test_furigana():
col = getEmptyCol()
mm = col.models
m = mm.current()
# filter should work
m["tmpls"][0]["qfmt"] = "{{kana:Front}}"
mm.save(m)
n = col.newNote()
n["Front"] = "foo[abc]"
col.addNote(n)
c = n.cards()[0]
assert c.q().endswith("abc")
# and should avoid sound
n["Front"] = "foo[sound:abc.mp3]"
n.flush()
assert "anki:play" in c.q(reload=True)
# it shouldn't throw an error while people are editing
m["tmpls"][0]["qfmt"] = "{{kana:}}"
mm.save(m)
c.q(reload=True)
def test_translate():
col = getEmptyCol()
no_uni = without_unicode_isolation
assert (
col.tr(TR.CARD_TEMPLATE_RENDERING_FRONT_SIDE_PROBLEM)
== "Front template has a problem:"
)
assert no_uni(col.tr(TR.STATISTICS_REVIEWS, reviews=1)) == "1 review"
assert no_uni(col.tr(TR.STATISTICS_REVIEWS, reviews=2)) == "2 reviews"
def test_db_named_args(capsys):
sql = "select a, 2+:test5 from b where arg =:foo and x = :test5"
args = []
kwargs = dict(test5=5, foo="blah")
s, a = emulate_named_args(sql, args, kwargs)
assert s == "select a, 2+?1 from b where arg =?2 and x = ?1"
assert a == [5, "blah"]
# swallow the warning
_ = capsys.readouterr()