def test_get_all_limit(self): msg = "message{0}" sender = "sender{0}" receiver = "receiver{0}" number_of_messages = 100 # From 90 to 60 (both included) offset = 10 count = 30 for i in range(number_of_messages): m = RMessage(msg.format(i), sender.format(i), receiver.format(i)) m.save() messages = RMessage.get_all(offset, count) self.assertEquals(count, len(messages)) number_of_messages = number_of_messages - offset - 1 # Because we start in 0 for i in messages: self.assertEquals(msg.format(number_of_messages), i.message) self.assertEquals(sender.format(number_of_messages), i.by) self.assertEquals(receiver.format(number_of_messages), i.to) number_of_messages = number_of_messages - 1
def show_messages(request): if request.method == "POST": form = MessageForm(data=request.POST) if form.is_valid(): data = form.cleaned_data message = RMessage(data['message'], data['by'], data['to']) message.save() messages.success(request, "Your new message is posted") # Create again the form form = MessageForm() else: form = MessageForm() # Get messages information rmessages = RMessage.get_all(0, MESSAGES_PER_PAGE) context = { "form": form, "rmessages": rmessages } return render_to_response('messagewall/messages.html', context, context_instance=RequestContext(request))
def test_get_count(self): number_of_messages = 100 for i in range(number_of_messages): m = RMessage("test", "test", "test") m.save() self.assertEquals(number_of_messages, RMessage.count())
def test_get_all_one(self): m = RMessage("message", "sender", "receiver") m.save() messages = RMessage.get_all() self.assertEquals(1, len(messages)) self.assertEquals("message", messages[0].message) self.assertEquals("sender", messages[0].by) self.assertEquals("receiver", messages[0].to)
def test_save(self): msg = RMessage("Microsoft is crap", "Richard Stallman", "Bill Gates") msg.save() r = utils.get_redis_connection() res = r.hgetall(RMessage.MESSAGE_KEY.format(msg.id)) self.assertEquals(1, r.llen(RMessage.MESSAGES_KEY)) self.assertEquals(msg.date, int(res['date'])) self.assertEquals(msg.id, res['id']) self.assertEquals(msg.by, res['by']) self.assertEquals(msg.to, res['to']) self.assertEquals(msg.message, res['message'])
def test_get_all(self): msg = "message{0}" sender = "sender{0}" receiver = "receiver{0}" number_of_messages = 100 for i in range(number_of_messages): m = RMessage(msg.format(i), sender.format(i), receiver.format(i)) m.save() messages = RMessage.get_all() for i in messages: number_of_messages = number_of_messages - 1 self.assertEquals(msg.format(number_of_messages), i.message) self.assertEquals(sender.format(number_of_messages), i.by) self.assertEquals(receiver.format(number_of_messages), i.to)