1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
|
import unittest
import openpaperwork_core
import paperwork_backend.model.util
class TestUtil(unittest.TestCase):
def setUp(self):
self.core = openpaperwork_core.Core(auto_load_dependencies=True)
self.core.load("openpaperwork_core.fs.fake")
self.core.init()
class FakeModule(object):
class Plugin(openpaperwork_core.PluginBase):
def doc_get_nb_pages_by_url(s, doc_url):
doc_url = doc_url[len("file:///"):]
return len(self.fs.fs[doc_url])
self.core._load_module("fake_module", FakeModule())
self.core.init()
self.fs = self.core.get_by_name("openpaperwork_core.fs.fake")
def test_delete_page_file(self):
self.fs.fs = {
"some_doc": {
"paper.1.words": "abcdef",
"paper.2.words": "ghijkl",
"paper.3.words": "mnopqr",
"paper.4.words": "stuvwx",
},
}
r = paperwork_backend.model.util.delete_page_file(
self.core, "paper.{}.words", "file:///some_doc", 1
)
self.assertTrue(r)
self.assertEqual(
self.fs.fs,
{
"some_doc": {
"paper.1.words": "abcdef",
"paper.2.words": "mnopqr",
"paper.3.words": "stuvwx",
},
}
)
def test_move_page_file(self):
self.fs.fs = {
"source_doc": {
"paper.1.words": "Aabcdef",
"paper.2.words": "Aghijkl",
"paper.3.words": "Amnopqr",
"paper.4.words": "Astuvwx",
},
"dest_doc": {
"paper.1.words": "Babcdef",
"paper.2.words": "Bghijkl",
"paper.3.words": "Bmnopqr",
"paper.4.words": "Bstuvwx",
},
}
r = paperwork_backend.model.util.move_page_file(
self.core, "paper.{}.words",
"file:///source_doc", 2,
"file:///dest_doc", 1
)
self.assertTrue(r)
self.assertEqual(
self.fs.fs,
{
"source_doc": {
"paper.1.words": "Aabcdef",
"paper.2.words": "Aghijkl",
"paper.3.words": "Astuvwx",
},
"dest_doc": {
"paper.1.words": "Babcdef",
"paper.2.words": "Amnopqr",
"paper.3.words": "Bghijkl",
"paper.4.words": "Bmnopqr",
"paper.5.words": "Bstuvwx",
},
}
)
def test_move_page_file_same_doc(self):
self.fs.fs = {
"source_doc": {
"paper.1.words": "Aabcdef",
"paper.2.words": "Aghijkl",
"paper.3.words": "Amnopqr",
"paper.4.words": "Astuvwx",
},
}
r = paperwork_backend.model.util.move_page_file(
self.core, "paper.{}.words",
"file:///source_doc", 2,
"file:///source_doc", 1
)
self.assertTrue(r)
self.assertEqual(
self.fs.fs,
{
"source_doc": {
"paper.1.words": "Aabcdef",
"paper.2.words": "Amnopqr",
"paper.3.words": "Aghijkl",
"paper.4.words": "Astuvwx",
},
}
)
|