1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138
|
import unittest
from unittest import mock
from ...management.branding import Branding
class TestBranding(unittest.TestCase):
def test_init_with_optionals(self):
branding = Branding(
domain="domain", token="jwttoken", telemetry=False, timeout=(10, 2)
)
self.assertEqual(branding.client.options.timeout, (10, 2))
telemetry = branding.client.base_headers.get("Auth0-Client", None)
self.assertEqual(telemetry, None)
@mock.patch("auth0.management.branding.RestClient")
def test_get(self, mock_rc):
api = mock_rc.return_value
branding = Branding(domain="domain", token="jwttoken")
branding.get()
api.get.assert_called_with(
"https://domain/api/v2/branding",
)
@mock.patch("auth0.management.branding.RestClient")
def test_update(self, mock_rc):
api = mock_rc.return_value
api.patch.return_value = {}
branding = Branding(domain="domain", token="jwttoken")
branding.update({"a": "b", "c": "d"})
api.patch.assert_called_with(
"https://domain/api/v2/branding", data={"a": "b", "c": "d"}
)
@mock.patch("auth0.management.branding.RestClient")
def test_get_template_universal_login(self, mock_rc):
api = mock_rc.return_value
branding = Branding(domain="domain", token="jwttoken")
branding.get_template_universal_login()
api.get.assert_called_with(
"https://domain/api/v2/branding/templates/universal-login",
)
@mock.patch("auth0.management.branding.RestClient")
def test_delete_template_universal_login(self, mock_rc):
api = mock_rc.return_value
branding = Branding(domain="domain", token="jwttoken")
branding.delete_template_universal_login()
api.delete.assert_called_with(
"https://domain/api/v2/branding/templates/universal-login",
)
@mock.patch("auth0.rest.requests.request")
def test_update_template_universal_login(self, mock_rc):
mock_rc.return_value.status_code = 200
mock_rc.return_value.text = "{}"
branding = Branding(domain="domain", token="jwttoken")
branding.update_template_universal_login({"a": "b", "c": "d"})
mock_rc.assert_called_with(
"PUT",
"https://domain/api/v2/branding/templates/universal-login",
json={"template": {"a": "b", "c": "d"}},
headers=mock.ANY,
timeout=5.0,
)
@mock.patch("auth0.management.branding.RestClient")
def test_get_default_branding_theme(self, mock_rc):
api = mock_rc.return_value
api.get.return_value = {}
branding = Branding(domain="domain", token="jwttoken")
branding.get_default_branding_theme()
api.get.assert_called_with(
"https://domain/api/v2/branding/themes/default",
)
@mock.patch("auth0.management.branding.RestClient")
def test_get_branding_theme(self, mock_rc):
api = mock_rc.return_value
api.get.return_value = {}
branding = Branding(domain="domain", token="jwttoken")
branding.get_branding_theme("theme_id")
api.get.assert_called_with(
"https://domain/api/v2/branding/themes/theme_id",
)
@mock.patch("auth0.management.branding.RestClient")
def test_delete_branding_theme(self, mock_rc):
api = mock_rc.return_value
api.delete.return_value = {}
branding = Branding(domain="domain", token="jwttoken")
branding.delete_branding_theme("theme_id")
api.delete.assert_called_with(
"https://domain/api/v2/branding/themes/theme_id",
)
@mock.patch("auth0.management.branding.RestClient")
def test_update_branding_theme(self, mock_rc):
api = mock_rc.return_value
api.patch.return_value = {}
branding = Branding(domain="domain", token="jwttoken")
branding.update_branding_theme("theme_id", {})
api.patch.assert_called_with(
"https://domain/api/v2/branding/themes/theme_id",
data={},
)
@mock.patch("auth0.management.branding.RestClient")
def test_create_branding_theme(self, mock_rc):
api = mock_rc.return_value
api.post.return_value = {}
branding = Branding(domain="domain", token="jwttoken")
branding.create_branding_theme({})
api.post.assert_called_with(
"https://domain/api/v2/branding/themes",
data={},
)
|