Browse Source

add tests for _get_last_entries

Oz N Tiram 9 years ago
parent
commit
cc3160d852
2 changed files with 11 additions and 4 deletions
  1. 4 3
      blogit/blogit.py
  2. 7 1
      tests/test_all.py

+ 4 - 3
blogit/blogit.py

@@ -345,10 +345,11 @@ def find_new_posts_and_pages(db):
                     yield e, e.id
 
 
-def _get_last_entries():
+def _get_last_entries(db):
     eids = [post.eid for post in db.posts.all()]
     eids = sorted(eids)[-10:][::-1]
-    entries = [Entry(db.posts.get(eid=eid)['filename']) for eid in eids]
+    entries = [Entry(os.path.join(CONFIG['content_root'],
+                     db.posts.get(eid=eid)['filename'])) for eid in eids]
     return entries
 
 
@@ -391,7 +392,7 @@ def build():
 
     # update index
     print("updating index")
-    update_index(_get_last_entries())
+    update_index(_get_last_entries(DB))
 
     # update archive
     print("updating archive")

+ 7 - 1
tests/test_all.py

@@ -4,7 +4,7 @@ import pytest
 from tinydb import Query, where
 
 from blogit.blogit import CONFIG, find_new_posts_and_pages, DataBase
-from blogit.blogit import Entry, Tag, _sort_entries
+from blogit.blogit import Entry, Tag, _sort_entries, _get_last_entries
 
 import blogit.blogit as m
 
@@ -226,3 +226,9 @@ def test_tag_render():
     # the entries are wrongly sorted, need to look at that
     assert tags[0].render()
     assert len(list(tags[0].entries))
+
+
+def test_get_last_entries():
+
+    le = _get_last_entries(DB)
+    assert [e.id for e in le] == range(22, 12, -1)