def testGetMyMessages(self):
status, result = views.getMyMessages(self.n1)
self.assertEqual(200, status)
ms1 = json.loads(result)
self.assertEqual(1, len(ms1))
self.assertEqual(self.n1, ms1[0]['recipient'])
self.assertEqual(self.n2, ms1[0]['sender'])
status, result = views.getMyMessages(self.n2)
self.assertEqual(200, status)
ms2 = json.loads(result)
self.assertEqual(2, len(ms2))
for m in ms2:
self.assertEqual(self.n2, m['recipient'])
self.assertEqual(self.n1, m['sender'])
status, result = views.getMyMessages(uuid.uuid4().hex)
self.assertEqual(404, status)
def testGetAllMessagesLimit(self):
n3, k3 = uuid.uuid4().hex, fakePublicKey()
u3 = User.create(n3, k3)
ms = [Message.create(u3, u3, uuid.uuid4().hex)
for i in range(LIMITS['MESSAGES']+1)]
status, result = views.getAllMessages()
self.assertEqual(200, status)
ms1 = json.loads(result)
self.assertEqual(LIMITS['MESSAGES'], len(ms1))
status, result = views.getMyMessages(n3)
self.assertEqual(200, status)
ms2 = json.loads(result)
self.assertEqual(LIMITS['MESSAGES'], len(ms2))
# Clean up
for m in ms:
m.delete()
u3.delete()
