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()