Add test for edit_text

jsmnbom · jsmnbom · commit 9928a1eefcc9 · 2016-09-25T16:11:19.000+02:00
diff --git a/tests/test_message.py b/tests/test_message.py
@@ -76,6 +76,15 @@ def test_forward(self):
         self.assertTrue(self.is_json(message.to_json()))
         self.assertEqual(message.text, 'Testing class method')
 
+    @flaky(3, 1)
+    def test_edit_text(self):
+        """Test for Message.edit_text"""
+        message = self._bot.sendMessage(self._chat_id, '.')
+        message = message.edit_text('Testing class method')
+
+        self.assertTrue(self.is_json(message.to_json()))
+        self.assertEqual(message.text, 'Testing class method')
+
 
 if __name__ == '__main__':
     unittest.main()
