# coding: utf-8 import tempfile import os import time from shared import getEmptyDeck, testDir # copying files to media folder def test_add(): d = getEmptyDeck() dir = tempfile.mkdtemp(prefix="anki") path = os.path.join(dir, u"foo.jpg") open(path, "w").write("hello") # new file, should preserve name assert d.media.addFile(path) == "foo.jpg" # adding the same file again should not create a duplicate assert d.media.addFile(path) == "foo.jpg" # but if it has a different md5, it should open(path, "w").write("world") assert d.media.addFile(path) == "foo (1).jpg" def test_strings(): d = getEmptyDeck() mf = d.media.filesInStr mid = d.models.models.keys()[0] assert mf(mid, "aoeu") == [] assert mf(mid, "aoeuao") == ["foo.jpg"] assert mf(mid, "aoeuao") == ["foo.jpg"] assert mf(mid, "aoeuao") == [ "foo.jpg", "bar.jpg"] assert mf(mid, "aoeuao") == ["foo.jpg"] assert mf(mid, "") == ["one", "two"] assert mf(mid, "aoeuao") == ["foo.jpg"] assert mf(mid, "aoeuao") == [ "foo.jpg", "fo"] assert mf(mid, "aou[sound:foo.mp3]aou") == ["foo.mp3"] sp = d.media.strip assert sp("aoeu") == "aoeu" assert sp("aoeu[sound:foo.mp3]aoeu") == "aoeuaoeu" assert sp("aoeu") == "aoeu" es = d.media.escapeImages assert es("aoeu") == "aoeu" assert es("") == "" assert es('') == '' def test_deckIntegration(): d = getEmptyDeck() # create a media dir d.media.dir() # put a file into it file = unicode(os.path.join(testDir, "support/fake.png")) d.media.addFile(file) # add a note which references it f = d.newNote() f['Front'] = u"one"; f['Back'] = u"" d.addNote(f) # and one which references a non-existent file f = d.newNote() f['Front'] = u"one"; f['Back'] = u"" d.addNote(f) # and add another file which isn't used open(os.path.join(d.media.dir(), "foo.jpg"), "wb").write("test") # check media ret = d.media.check() assert ret[0] == ["fake2.png"] assert ret[1] == ["foo.jpg"] def test_changes(): d = getEmptyDeck() assert d.media._changed() def added(): return d.media.db.execute("select fname from log where type = 0") assert not list(added()) assert not list(d.media.removed()) # add a file dir = tempfile.mkdtemp(prefix="anki") path = os.path.join(dir, u"foo.jpg") open(path, "w").write("hello") time.sleep(1) path = d.media.addFile(path) # should have been logged d.media.findChanges() assert list(added()) assert not list(d.media.removed()) # if we modify it, the cache won't notice time.sleep(1) open(path, "w").write("world") assert len(list(added())) == 1 assert not list(d.media.removed()) # but if we add another file, it will time.sleep(1) open(path+"2", "w").write("yo") d.media.findChanges() assert len(list(added())) == 2 assert not list(d.media.removed()) # deletions should get noticed too time.sleep(1) os.unlink(path+"2") d.media.findChanges() assert len(list(added())) == 1 assert len(list(d.media.removed())) == 1 def test_illegal(): d = getEmptyDeck() aString = u"a:b|cd\\e/f\0g*h" good = u"abcdefgh" assert d.media.stripIllegal(aString) == good for c in aString: bad = d.media.hasIllegal("somestring"+c+"morestring") if bad: assert(c not in good) else: assert(c in good)