From eab2de69ba2df4e10e9e3dbaa5ad9ebe478ac248 Mon Sep 17 00:00:00 2001 From: Harmon Date: Fri, 21 Jun 2019 12:00:22 -0500 Subject: [PATCH] Update test for API.send_direct_message and API.destroy_direct_message --- tests/test_api.py | 16 +++++++--------- 1 file changed, 7 insertions(+), 9 deletions(-) diff --git a/tests/test_api.py b/tests/test_api.py index 92ca6d4..5db0362 100644 --- a/tests/test_api.py +++ b/tests/test_api.py @@ -160,18 +160,16 @@ class TweepyAPITests(TweepyTestCase): @tape.use_cassette('testsendanddestroydirectmessage.json') def testsendanddestroydirectmessage(self): + me = self.api.me() + # send - sent_dm = self.api.send_direct_message(username, text='test message') - self.assertEqual(sent_dm.text, 'test message') - self.assertEqual(sent_dm.sender.screen_name, username) - self.assertEqual(sent_dm.recipient.screen_name, username) + sent_dm = self.api.send_direct_message(me.id, text='test message') + self.assertEqual(sent_dm.message_create['message_data']['text'], 'test message') + self.assertEqual(int(sent_dm.message_create['sender_id']), me.id) + self.assertEqual(int(sent_dm.message_create['target']['recipient_id']), me.id) # destroy - destroyed_dm = self.api.destroy_direct_message(sent_dm.id) - self.assertEqual(destroyed_dm.text, sent_dm.text) - self.assertEqual(destroyed_dm.id, sent_dm.id) - self.assertEqual(destroyed_dm.sender.screen_name, username) - self.assertEqual(destroyed_dm.recipient.screen_name, username) + self.api.destroy_direct_message(sent_dm.id) @tape.use_cassette('testcreatedestroyfriendship.json') def testcreatedestroyfriendship(self): -- 2.25.1