def test_vote(self): """test link comment count""" list = ListFactory() list.vote() list.save() self.assertEqual(list.upvotes, 1) from datetime import datetime hour = datetime.now().hour hours = { 0: 'votes_h00', 1: 'votes_h00', 2: 'votes_h02', 3: 'votes_h02', 4: 'votes_h04', 5: 'votes_h04', 6: 'votes_h06', 7: 'votes_h06', 8: 'votes_h08', 9: 'votes_h08', 10: 'votes_h10', 11: 'votes_h10', 12: 'votes_h12', 13: 'votes_h12', 14: 'votes_h14', 15: 'votes_h14', 16: 'votes_h16', 17: 'votes_h16', 18: 'votes_h18', 19: 'votes_h18', 20: 'votes_h20', 21: 'votes_h20', 22: 'votes_h22', 23: 'votes_h22' } self.assertEqual(getattr(list, hours[hour]), 1)
def test_vote(self): """test link comment count""" list = ListFactory() list.vote() list.save() self.assertEqual(list.upvotes, 1) from datetime import datetime hour = datetime.now().hour hours = { 0: 'votes_h00', 1: 'votes_h00', 2: 'votes_h02', 3: 'votes_h02', 4: 'votes_h04', 5: 'votes_h04', 6: 'votes_h06', 7: 'votes_h06', 8: 'votes_h08', 9: 'votes_h08', 10: 'votes_h10', 11: 'votes_h10', 12: 'votes_h12', 13: 'votes_h12', 14: 'votes_h14', 15: 'votes_h14', 16: 'votes_h16', 17: 'votes_h16', 18: 'votes_h18', 19: 'votes_h18', 20: 'votes_h20', 21: 'votes_h20', 22: 'votes_h22', 23: 'votes_h22'} self.assertEqual(getattr(list, hours[hour]), 1)
def test_votes(self): profile2 = ProfileFactory() user2 = profile2.user alist = ListFactory() alist.owner = user2 alist.save() self.assertEqual(alist.get_votes(), 0) alist.vote(self.user) self.assertEqual(alist.get_votes(), 1) alist.unvote(self.user) self.assertEqual(alist.get_votes(), 0)