Skip to content

Commit 97326f6

Browse files
authored
chore(tests): factor out social attributes assertion (#1426)
This makes the assertion be shared between login and partial pipeline.
1 parent d3fd2e1 commit 97326f6

File tree

1 file changed

+10
-25
lines changed

1 file changed

+10
-25
lines changed

social_core/tests/backends/test_twitter_oauth2.py

Lines changed: 10 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,7 @@ class TwitterOAuth2Mixin(OAuth2Test, ABC):
8181

8282
expected_username = "twitter_username"
8383

84-
def test_login(self) -> None:
85-
user = self.do_login()
86-
87-
self.assertEqual(len(user.social), 1)
88-
89-
social = user.social[0]
84+
def _assert_social_attributes(self, social):
9085
self.assertEqual(social.uid, "1234567890123456789")
9186
self.assertEqual(social.extra_data["first_name"], "first")
9287
self.assertEqual(social.extra_data["last_name"], "last")
@@ -107,30 +102,20 @@ def test_login(self) -> None:
107102
self.assertEqual(social.extra_data["public_metrics"]["tweet_count"], 40)
108103
self.assertEqual(social.extra_data["public_metrics"]["listed_count"], 7)
109104

105+
def test_login(self) -> None:
106+
user = self.do_login()
107+
108+
self.assertEqual(len(user.social), 1)
109+
110+
social = user.social[0]
111+
self._assert_social_attributes(social)
112+
110113
def test_partial_pipeline(self) -> None:
111114
user = self.do_partial_pipeline()
112115
self.assertEqual(len(user.social), 1)
113116

114117
social = user.social[0]
115-
self.assertEqual(social.uid, "1234567890123456789")
116-
self.assertEqual(social.extra_data["first_name"], "first")
117-
self.assertEqual(social.extra_data["last_name"], "last")
118-
self.assertEqual(social.extra_data["fullname"], "first last")
119-
self.assertEqual(social.extra_data["created_at"], "2023-03-06T06:18:59.000Z")
120-
self.assertEqual(social.extra_data["verified"], False)
121-
self.assertEqual(social.extra_data["verified_type"], "none")
122-
self.assertEqual(social.extra_data["protected"], True)
123-
self.assertEqual(social.extra_data["description"], "description str")
124-
self.assertEqual(social.extra_data["url"], "https://social-core-test-url.com")
125-
self.assertEqual(social.extra_data["pinned_tweet_id"], "9876543210987654321")
126-
self.assertEqual(
127-
social.extra_data["profile_image_url"],
128-
"https://social-core-test-url.com/image.png",
129-
)
130-
self.assertEqual(social.extra_data["public_metrics"]["followers_count"], 69)
131-
self.assertEqual(social.extra_data["public_metrics"]["following_count"], 129)
132-
self.assertEqual(social.extra_data["public_metrics"]["tweet_count"], 40)
133-
self.assertEqual(social.extra_data["public_metrics"]["listed_count"], 7)
118+
self._assert_social_attributes(social)
134119

135120

136121
class TwitterOAuth2TestMissingOptionalValue(OAuth2Test, BaseAuthUrlTestMixin):

0 commit comments

Comments
 (0)