login_client_test.py

from unittest import TestCase

from linode_api4 import OAuthScopes


class OAuthScopesTest(TestCase):
    def test_parse_scopes_none(self):
        """
        Tests parsing no scopes
        """
        scopes = OAuthScopes.parse("")
        self.assertEqual(scopes, [])

    def test_parse_scopes_single(self):
        """
        Tests parsing a single scope
        """
        scopes = OAuthScopes.parse("linodes:read_only")
        self.assertEqual(scopes, [OAuthScopes.Linodes.read_only])

    def test_parse_scopes_many(self):
        """
        Tests parsing many scopes
        """
        scopes = OAuthScopes.parse("linodes:read_only domains:read_write")
        self.assertEqual(
            scopes,
            [OAuthScopes.Linodes.read_only, OAuthScopes.Domains.read_write],
        )

    def test_parse_scopes_many_comma_delimited(self):
        """
        Tests parsing many scopes that are comma-delimited (which preserves old behavior)
        """
        scopes = OAuthScopes.parse(
            "nodebalancers:read_write,stackscripts:*,events:read_only"
        )
        self.assertEqual(
            scopes,
            [
                OAuthScopes.NodeBalancers.read_write,
                OAuthScopes.StackScripts.all,
                OAuthScopes.Events.read_only,
            ],
        )

    def test_parse_scopes_all(self):
        """
        Tests parsing * scopes
        """
        scopes = OAuthScopes.parse("*")
        self.assertEqual(
            scopes,
            [getattr(c, "all") for c in OAuthScopes._scope_families.values()],
        )

    def test_parse_scopes_databases_and_vpc(self):
        """
        Tests parsing documented databases and vpc scopes
        """
        scopes = OAuthScopes.parse(
            "databases:read_only,databases:read_write,vpc:read_write,vpc:*"
        )
        self.assertEqual(
            scopes,
            [
                OAuthScopes.Databases.read_only,
                OAuthScopes.Databases.read_write,
                OAuthScopes.VPC.read_write,
                OAuthScopes.VPC.all,
            ],
        )
        self.assertEqual(OAuthScopes.parse("vpc:read_only"), [])