Anki/tests/test_models.py

# coding: utf-8

from tests.shared import getEmptyDeck
from anki.models import Model
from anki.utils import stripHTML

def test_modelDelete():
    deck = getEmptyDeck()
    f = deck.newFact()
    f['Front'] = u'1'
    f['Back'] = u'2'
    deck.addFact(f)
    assert deck.cardCount() == 1
    deck.delModel(deck.conf['currentModelId'])
    assert deck.cardCount() == 0

def test_modelCopy():
    deck = getEmptyDeck()
    m = deck.currentModel()
    m2 = m.copy()
    assert m2.name == "Basic copy"
    assert m2.id != m.id
    assert len(m2.fields) == 2
    assert len(m.fields) == 2
    assert len(m2.fields) == len(m.fields)
    assert len(m.templates) == 2
    assert len(m2.templates) == 2

def test_fields():
    d = getEmptyDeck()
    f = d.newFact()
    f['Front'] = u'1'
    f['Back'] = u'2'
    d.addFact(f)
    m = d.currentModel()
    # make sure renaming a field updates the templates
    m.renameField(m.fields[0], "NewFront")
    assert m.templates[0]['qfmt'] == "{{NewFront}}"
    # add a field
    f = m.newField()
    f['name'] = "foo"
    m.addField(f)
    assert d.getFact(m.fids()[0]).fields == ["1", "2", ""]
    # rename it
    m.renameField(f, "bar")
    assert d.getFact(m.fids()[0])['bar'] == ''
    # delete back
    m.delField(m.fields[1])
    assert d.getFact(m.fids()[0]).fields == ["1", ""]
    # move 0 -> 1
    m.moveField(m.fields[0], 1)
    assert d.getFact(m.fids()[0]).fields == ["", "1"]
    # move 1 -> 0
    m.moveField(m.fields[1], 0)
    assert d.getFact(m.fids()[0]).fields == ["1", ""]
    # add another and put in middle
    f = m.newField()
    f['name'] = "baz"
    m.addField(f)
    f = d.getFact(m.fids()[0])
    f['baz'] = "2"
    f.flush()
    assert d.getFact(m.fids()[0]).fields == ["1", "", "2"]
    # move 2 -> 1
    m.moveField(m.fields[2], 1)
    assert d.getFact(m.fids()[0]).fields == ["1", "2", ""]
    # move 0 -> 2
    m.moveField(m.fields[0], 2)
    assert d.getFact(m.fids()[0]).fields == ["2", "", "1"]
    # move 0 -> 1
    m.moveField(m.fields[0], 1)
    assert d.getFact(m.fids()[0]).fields == ["", "2", "1"]

def test_templates():
    d = getEmptyDeck()
    m = d.currentModel()
    m.templates[1]['actv'] = True
    m.flush()
    f = d.newFact()
    f['Front'] = u'1'
    f['Back'] = u'2'
    d.addFact(f)
    assert d.cardCount() == 2
    (c, c2) = f.cards()
    # first card should have first ord
    assert c.ord == 0
    assert c2.ord == 1
    # switch templates
    m.moveTemplate(c.template(), 1)
    c.load(); c2.load()
    assert c.ord == 1
    assert c2.ord == 0
    # removing a template should delete its cards
    m.delTemplate(m.templates[0])
    assert d.cardCount() == 1
    # and should have updated the other cards' ordinals
    c = f.cards()[0]
    assert c.ord == 0
    stripHTML(c.q()) == "2"

def test_text():
    d = getEmptyDeck()
    m = d.currentModel()
    m.templates[0]['qfmt'] = "{{text:Front}}"
    m.flush()
    f = d.newFact()
    f['Front'] = u'hello<b>world'
    d.addFact(f)
    assert "helloworld" in f.cards()[0].q()

def test_cloze():
    d = getEmptyDeck()
    d.conf['currentModelId'] = 2
    f = d.newFact()
    assert f.model().name == "Cloze"
    # a cloze model with no clozes is empty
    f['Text'] = u'nothing'
    assert d.addFact(f) == 0
    # try with one cloze
    f['Text'] = "hello {{c1::world}}"
    assert d.addFact(f) == 1
    assert "hello <b>...</b>" in f.cards()[0].q()
    assert "hello <b>world</b>" in f.cards()[0].a()
    # and with a comment
    f = d.newFact()
    f['Text'] = "hello {{c1::world::typical}}"
    assert d.addFact(f) == 1
    assert "<b>...(typical)</b>" in f.cards()[0].q()
    assert "<b>world</b>" in f.cards()[0].a()
    # and with 2 clozes
    f = d.newFact()
    f['Text'] = "hello {{c1::world}} {{c2::bar}}"
    assert d.addFact(f) == 2
    (c1, c2) = f.cards()
    assert "<b>...</b> bar" in c1.q()
    assert "<b>world</b> bar" in c1.a()
    assert "world <b>...</b>" in c2.q()
    assert "world <b>bar</b>" in c2.a()
    # clozes should be supported in sections too
    m = d.currentModel()
    m.templates[0]['qfmt'] = "{{#cloze:1:Text}}{{Notes}}{{/cloze:1:Text}}"
    m.flush()
    f = d.newFact()
    f['Text'] = "hello"
    f['Notes'] = "world"
    assert d.addFact(f) == 0
    f['Text'] = "hello {{c1::foo}}"
    assert d.addFact(f) == 1

def test_modelChange():
    deck = getEmptyDeck()
    basic = deck.getModel(1)
    cloze = deck.getModel(2)
    # enable second template and add a fact
    basic.templates[1]['actv'] = True
    basic.flush()
    f = deck.newFact()
    f['Front'] = u'f'
    f['Back'] = u'b'
    deck.addFact(f)
    # switch fields
    map = {0: 1, 1: 0}
    basic.changeModel([f.id], basic, map, None)
    f.load()
    assert f['Front'] == 'b'
    assert f['Back'] == 'f'
    # switch cards
    c0 = f.cards()[0]
    c1 = f.cards()[1]
    assert stripHTML(c0.q()) == "b"
    assert stripHTML(c1.q()) == "f"
    assert c0.ord == 0
    assert c1.ord == 1
    basic.changeModel([f.id], basic, None, map)
    f.load(); c0.load(); c1.load()
    assert stripHTML(c0.q()) == "f"
    assert stripHTML(c1.q()) == "b"
    assert c0.ord == 1
    assert c1.ord == 0
    # .cards() returns cards in order
    assert f.cards()[0].id == c1.id
    # delete first card
    map = {0: None, 1: 1}
    basic.changeModel([f.id], basic, None, map)
    f.load()
    c0.load()
    try:
        c1.load()
        assert 0
    except TypeError:
        pass
    assert len(f.cards()) == 1
    # an unmapped field becomes blank
    assert f['Front'] == 'b'
    assert f['Back'] == 'f'
    basic.changeModel([f.id], basic, map, None)
    f.load()
    assert f['Front'] == ''
    assert f['Back'] == 'f'
    # another fact to try model conversion
    f = deck.newFact()
    f['Front'] = u'f2'
    f['Back'] = u'b2'
    deck.addFact(f)
    assert basic.useCount() == 2
    assert cloze.useCount() == 0
    map = {0: 0, 1: 1}
    basic.changeModel([f.id], cloze, map, map)
    f.load()
    assert f['Text'] == "f2"
    assert f['Notes'] == "b2"
    assert len(f.cards()) == 2
    assert "b2" in f.cards()[0].a()