fixed tests not working

This commit is contained in:
ajurna
2019-02-07 15:17:28 +00:00
parent bc6504088f
commit e52ba3f393
2 changed files with 27 additions and 29 deletions

View File

@@ -117,6 +117,9 @@ class DirFile:
self.label = generate_label(comic, status)
self.type = 'book'
def __repr__(self):
return f'<DirFile: {self.name}: {self.type}>'
def generate_directory(user, directory=False):
"""
@@ -147,6 +150,7 @@ def generate_directory(user, directory=False):
df.populate_directory(directory_obj, user)
files.append(df)
dir_list.remove(directory_obj.name)
with atomic():
for file_obj in file_list_obj:
df = DirFile()
@@ -169,6 +173,8 @@ def generate_directory(user, directory=False):
df = DirFile()
df.populate_comic(book, user)
files.append(df)
files.sort(key=lambda x: x.name)
files.sort(key=lambda x: x.type, reverse=True)
return files