Add test for edit_text

python-telegram-bot · Sep 25, 2016 · 9928a1e · 9928a1e
1 parent 0e2c366
commit 9928a1e
Showing 1 changed file with 9 additions and 0 deletions.
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()