def test_get_message(self): m = Message() m.content = 'test' m.add_or_update() m = Message.find_by_id(m.id) self.assertEqual(m.content, 'test') self.assertEqual(m.id, 1)
def test_to_json(self): m = Message() m.content = 'test' m.add_or_update() schema = MessageResponseSchema() m_json = schema.dump(m) expected_keys = ['id', 'content', 'palindrome', 'date_created', 'date_modified'] self.assertEqual(sorted(m_json.keys()), sorted(expected_keys))
def test_delete_message(self): m = Message() m.content = 'test' m.add_or_update() m = Message.find_by_id(m.id) m.delete() m = Message.find_by_id(m.id) self.assertIsNone(m)
def test_create_message(self): m = Message() m.content = 'test' m.add_or_update() self.assertTrue( (datetime.utcnow() - m.date_created).total_seconds() < 3) self.assertTrue( (datetime.utcnow() - m.date_modified).total_seconds() < 3) self.assertEqual(m.content, 'test') self.assertEqual(m.id, 1)
def test_update_message(self): m = Message() m.content = 'test' m.add_or_update() m = Message.find_by_id(m.id) m.content = 'test updated' m.add_or_update() m = Message.find_by_id(m.id) print(m.date_created) print(m.date_modified) self.assertEqual(m.content, 'test updated') self.assertEqual(m.id, 1)