def view_most_liked():
user = None
if current_user.is_authenticated:
user = current_user
return render_template('view_all.html',
header="Most Liked",
media=get_most_liked(user, limit=None))
def test_get_most_liked_limit(self):
self.logged_in.post('/upload', data=self.upload_jpg)
media1 = Media.query.filter_by(id=1).first()
self.logged_in.post('/upload', data=self.upload_mp4)
media2 = Media.query.filter_by(id=2).first()
media2.like(self.user)
result = get_most_liked(user=self.user, limit=1)
self.assertEquals(len(result), 1)
self.assertEquals(result, [media2])
def test_get_most_liked_default(self):
self.logged_in.post('/upload', data=self.upload_jpg)
media1 = Media.query.filter_by(id=1).first()
self.logged_in.post('/upload', data=self.upload_mp4)
media2 = Media.query.filter_by(id=2).first()
media2.like(self.user)
test = [media2, media1]
result = get_most_liked()
self.assertEquals(result, test)
def index():
user = None
if current_user.is_authenticated:
user = current_user
recent_uploads = get_most_recent(user=user)
most_viewed = get_most_viewed(user=user)
most_liked = get_most_liked(user=user)
return render_template('index.html',
recent_uploads=recent_uploads,
most_viewed=most_viewed,
most_liked=most_liked)
def test_get_most_liked_private(self):
self.upload_jpg['public'] = False
self.logged_in.post('/upload', data=self.upload_jpg)
media1 = Media.query.filter_by(id=1).first()
media1.like(self.user)
self.logged_in.post('/upload', data=self.upload_mp4)
media2 = Media.query.filter_by(id=2).first()
test = [media2]
result = get_most_liked()
self.assertEquals(result, test)
def test_get_most_liked_all(self):
self.logged_in.post('/upload', data=self.upload_jpg)
media1 = Media.query.filter_by(id=1).first()
self.logged_in.post('/upload', data=self.upload_mp4)
media2 = Media.query.filter_by(id=2).first()
self.logged_in.post('/upload', data=self.extra1)
media3 = Media.query.filter_by(id=3).first()
self.logged_in.post('/upload', data=self.extra2)
media4 = Media.query.filter_by(id=4).first()
self.logged_in.post('/upload', data=self.extra3)
media5 = Media.query.filter_by(id=5).first()
media5.like(self.user)
expected = [media5, media1, media2, media3, media4]
result = get_most_liked(user=self.user, limit=None)
self.assertEquals(result, expected)