def test_get_work_by_user(self):
data = self.build_data(True, True)
tagType = TagType(label='one')
db.session.add(tagType)
tagType = TagType(label='two')
db.session.add(tagType)
db.session.commit()
data['user_id'] = 1
work.add_work(data)
selected_work = json.loads(work.get_by_user(1))
self.assertTrue(selected_work['works'][0]['title'] == "A Tale of Two Poor Students")
def test_add_work(self):
data = self.build_data(False, False)
data['user_id'] = 1
new_id = work.add_work(data)
selected_work = Work.query.filter_by(id=1).first()
self.assertTrue(new_id==1)
self.assertTrue(selected_work.is_complete == 0)
def test_add_bookmark(self):
data = self.build_data(True, False, True)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
bookmark.add_bookmark(data)
new_bookmark = Bookmark.query.filter_by(id=1).first()
self.assertTrue(new_bookmark.description == "This was fine I guess...")
def test_add_bookmark(self):
data = self.build_data(True, False, True)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
bookmark.add_bookmark(data)
new_bookmark = Bookmark.query.filter_by(id=1).first()
self.assertTrue(new_bookmark.description == "This was fine I guess...")
def test_get_bookmarks_by_curator(self):
data = self.build_data(True, False, False)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
work_fetched = Work.query.filter_by(id=work_id).first()
bookmark_id = bookmark.add_bookmark(data)
fetched_bookmark = bookmark.get_bookmarks_by_curator(1)
self.assertTrue(fetched_bookmark['bookmarks'][0]["rating"] == 3)
def test_get_bookmark(self):
data = self.build_data(True, False, False)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
work_fetched = Work.query.filter_by(id=work_id).first()
bookmark_id = bookmark.add_bookmark(data)
new_bookmark = bookmark.get_bookmark(1, 1)
self.assertTrue(new_bookmark["curator_title"] == "A Fic I Read")
def post_work():
user_id = auth.auth_from_data(request)
if user_id > 0:
request.json['user_id'] = user_id
work_id = work.add_work(request.json)
return json.dumps({"work_id": work_id})
else:
abort(400)
def test_get_bookmarks_by_curator(self):
data = self.build_data(True, False, False)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
work_fetched = Work.query.filter_by(id=work_id).first()
bookmark_id = bookmark.add_bookmark(data)
fetched_bookmark = bookmark.get_bookmarks_by_curator(1)
self.assertTrue(fetched_bookmark['bookmarks'][0]["rating"] == 3)
def test_get_bookmark(self):
data = self.build_data(True, False, False)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
work_fetched = Work.query.filter_by(id=work_id).first()
bookmark_id = bookmark.add_bookmark(data)
new_bookmark = bookmark.get_bookmark(1, 1)
self.assertTrue(new_bookmark["curator_title"] == "A Fic I Read")
def test_update_bookmark(self):
data = self.build_data(True, False, True)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
bookmark_id = bookmark.add_bookmark(data)
data["id"] = bookmark_id
data["description"] = "This was bad actually!!!"
bookmark.update_bookmark(data)
updated_bookmark = Bookmark.query.filter_by(id=bookmark_id).first()
self.assertTrue(updated_bookmark.description == "This was bad actually!!!")
def test_update_bookmark(self):
data = self.build_data(True, False, True)
work_data = self.build_dummary_work_data(True, True)
work_id = work.add_work(work_data)
bookmark_id = bookmark.add_bookmark(data)
data["id"] = bookmark_id
data["description"] = "This was bad actually!!!"
bookmark.update_bookmark(data)
updated_bookmark = Bookmark.query.filter_by(id=bookmark_id).first()
self.assertTrue(
updated_bookmark.description == "This was bad actually!!!")
def test_export_work(self):
data = self.build_data(False, True)
data['user_id'] = 1
new_id = work.add_work(data)
selected_work = Work.query.filter_by(id=1).first()
work_export.create_epub(selected_work)