authentik.sources.oauth.tests.test_type_twitter

Twitter Type tests

View Source

 1"""Twitter Type tests"""
 2
 3from django.test import TestCase
 4
 5from authentik.sources.oauth.models import OAuthSource
 6from authentik.sources.oauth.types.twitter import TwitterType
 7
 8# https://developer.twitter.com/en/docs/twitter-api/users/lookup/api-reference/get-users-me
 9TWITTER_USER = {"data": {"id": "2244994945", "name": "TwitterDev", "username": "Twitter Dev"}}
10
11
12class TestTypeGitHub(TestCase):
13    """OAuth Source tests"""
14
15    def setUp(self):
16        self.source = OAuthSource.objects.create(
17            name="test",
18            slug="test",
19            provider_type="openidconnect",
20            authorization_url="",
21            profile_url="",
22            consumer_key="",
23        )
24
25    def test_enroll_context(self):
26        """Test Twitter Enrollment context"""
27        ak_context = TwitterType().get_base_user_properties(source=self.source, info=TWITTER_USER)
28        self.assertEqual(ak_context["username"], TWITTER_USER["data"]["username"])
29        self.assertEqual(ak_context["email"], None)
30        self.assertEqual(ak_context["name"], TWITTER_USER["data"]["name"])

TWITTER_USER = {'data': {'id': '2244994945', 'name': 'TwitterDev', 'username': 'Twitter Dev'}}

class TestTypeGitHub(django.test.testcases.TestCase): View Source

13class TestTypeGitHub(TestCase):
14    """OAuth Source tests"""
15
16    def setUp(self):
17        self.source = OAuthSource.objects.create(
18            name="test",
19            slug="test",
20            provider_type="openidconnect",
21            authorization_url="",
22            profile_url="",
23            consumer_key="",
24        )
25
26    def test_enroll_context(self):
27        """Test Twitter Enrollment context"""
28        ak_context = TwitterType().get_base_user_properties(source=self.source, info=TWITTER_USER)
29        self.assertEqual(ak_context["username"], TWITTER_USER["data"]["username"])
30        self.assertEqual(ak_context["email"], None)
31        self.assertEqual(ak_context["name"], TWITTER_USER["data"]["name"])

OAuth Source tests

def setUp(self): View Source

16    def setUp(self):
17        self.source = OAuthSource.objects.create(
18            name="test",
19            slug="test",
20            provider_type="openidconnect",
21            authorization_url="",
22            profile_url="",
23            consumer_key="",
24        )

Hook method for setting up the test fixture before exercising it.

def test_enroll_context(self): View Source

26    def test_enroll_context(self):
27        """Test Twitter Enrollment context"""
28        ak_context = TwitterType().get_base_user_properties(source=self.source, info=TWITTER_USER)
29        self.assertEqual(ak_context["username"], TWITTER_USER["data"]["username"])
30        self.assertEqual(ak_context["email"], None)
31        self.assertEqual(ak_context["name"], TWITTER_USER["data"]["name"])

Test Twitter Enrollment context