From 4eb031165835a76854fedcceef85a1a629ca3afc Mon Sep 17 00:00:00 2001 From: ikaroskun Date: Tue, 29 Oct 2024 15:30:12 +0800 Subject: [PATCH] feat(dm): :white_check_mark: update tests for block user dm --- pytwitter/api.py | 2 -- tests/apis/test_users.py | 21 +++++++++++++++++++++ 2 files changed, 21 insertions(+), 2 deletions(-) diff --git a/pytwitter/api.py b/pytwitter/api.py index 9a83189..08eeecb 100644 --- a/pytwitter/api.py +++ b/pytwitter/api.py @@ -2048,7 +2048,6 @@ def block_user_dm(self, target_user_id: str): data = self._parse_response(resp) return data - def unblock_user_dm(self, target_user_id: str): """ Allows the authenticated user to unblock direct messages (DMs) from the target user. @@ -2062,7 +2061,6 @@ def unblock_user_dm(self, target_user_id: str): data = self._parse_response(resp) return data - def get_user_muting( self, user_id: str, diff --git a/tests/apis/test_users.py b/tests/apis/test_users.py index 1a4a975..e2b798d 100644 --- a/tests/apis/test_users.py +++ b/tests/apis/test_users.py @@ -153,6 +153,27 @@ def test_block_and_unblock_user(api_with_user): assert not resp["data"]["blocking"] +@responses.activate +def test_block_and_unblock_user_dm(api_with_user): + target_user_id = "123456" + + responses.add( + method=responses.POST, + url=f"https://api.twitter.com/2/users/{target_user_id}/dm/block", + json={"data": {"blocked": True}}, + ) + resp = api_with_user.block_user_dm(target_user_id=target_user_id) + assert resp["data"]["blocked"] + + responses.add( + method=responses.POST, + url=f"https://api.twitter.com/2/users/{target_user_id}/dm/unblock", + json={"data": {"blocked": False}}, + ) + resp = api_with_user.unblock_user_dm(target_user_id=target_user_id) + assert not resp["data"]["blocked"] + + @responses.activate def test_get_blocking_users(api_with_user, helpers): users_data = helpers.load_json_data(