+{% endblock %}
+
+{% block footer %}
+ {{ block.super }}
+{% endblock %}
diff --git a/enterprise/utils.py b/enterprise/utils.py
index 291f7d139b..631eeefd48 100644
--- a/enterprise/utils.py
+++ b/enterprise/utils.py
@@ -1565,6 +1565,13 @@ def get_sso_orchestrator_configure_path():
return settings.ENTERPRISE_SSO_ORCHESTRATOR_CONFIGURE_PATH
+def get_sso_orchestrator_configure_edx_oauth_path():
+ """
+ Return the SSO orchestrator configure-edx-oauth endpoint path, or None if it is not defined.
+ """
+ return getattr(settings, "ENTERPRISE_SSO_ORCHESTRATOR_CONFIGURE_EDX_OAUTH_PATH", None)
+
+
def get_enterprise_worker_user():
"""
Return the user object of enterprise worker user.
diff --git a/tests/test_admin/test_view.py b/tests/test_admin/test_view.py
index 87288b60d3..ca2602881b 100644
--- a/tests/test_admin/test_view.py
+++ b/tests/test_admin/test_view.py
@@ -33,6 +33,7 @@
TransmitEnterpriseCoursesForm,
)
from enterprise.admin.utils import ValidationMessages
+from enterprise.api_client.sso_orchestrator import SsoOrchestratorClientError
from enterprise.constants import PAGE_SIZE
from enterprise.models import (
EnrollmentNotificationEmailTemplate,
@@ -2078,3 +2079,92 @@ def test_post_validation_errors(self):
)
]
}
+
+
+class BaseTestEnterpriseCustomerSetupAuthOrgIDView(BaseEnterpriseCustomerView):
+ """
+ Common functionality for EnterpriseCustomerTransmitCoursesView tests.
+ """
+
+ def setUp(self):
+ """
+ Test set up
+ """
+ super().setUp()
+ self.enterprise_customer.auth_org_id = None
+ self.enterprise_customer.save()
+ self.view_url = reverse(
+ 'admin:' + enterprise_admin.utils.UrlNames.SETUP_AUTH_ORG_ID,
+ args=(self.enterprise_customer.uuid,)
+ )
+
+
+@ddt.ddt
+@mark.django_db
+@override_settings(ROOT_URLCONF='test_utils.admin_urls')
+class TestEnterpriseCustomerSetupAuthOrgIDViewGet(BaseTestEnterpriseCustomerSetupAuthOrgIDView):
+ """
+ Tests for EnterpriseCustomerSetupAuthOrgIDView GET endpoint.
+ """
+
+ def _test_get_response(self, response):
+ """
+ Test view GET response for common parts.
+ """
+ assert response.status_code == 200
+ self._test_common_context(response.context)
+ assert response.context['enterprise_customer'] == self.enterprise_customer
+
+ def test_get_not_logged_in(self):
+ response = self.client.get(self.view_url)
+ assert response.status_code == 302
+
+ def test_get_links(self):
+ self._login()
+
+ response = self.client.get(self.view_url)
+ self._test_get_response(response)
+
+
+@ddt.ddt
+@mark.django_db
+@override_settings(ROOT_URLCONF='test_utils.admin_urls')
+class TestEnterpriseCustomerSetupAuthOrgIDViewPost(BaseTestEnterpriseCustomerSetupAuthOrgIDView):
+ """
+ Tests for EnterpriseCustomerSetupAuthOrgIDView POST endpoint.
+ """
+
+ def test_post_not_logged_in(self):
+ response = self.client.post(self.view_url, data={})
+ assert response.status_code == 302
+
+ @mock.patch('enterprise.api_client.sso_orchestrator.EnterpriseSSOOrchestratorApiClient.configure_edx_oauth')
+ def test_post_happy_path(self, mock_configure_edx_oauth):
+ fake_org_id = 'foobar'
+ mock_configure_edx_oauth.return_value = fake_org_id
+ self._login()
+ response = self.client.post(self.view_url)
+ mock_configure_edx_oauth.assert_called_once_with(self.enterprise_customer)
+ self.enterprise_customer.refresh_from_db()
+ assert self.enterprise_customer.auth_org_id == fake_org_id
+
+ # Now check that the redirect is correct and that the success message is set.
+ self.assertRedirects(response, self.view_url, fetch_redirect_response=False)
+ get_response = self.client.get(self.view_url)
+ actual_messages = {
+ (m.level, m.message) for m in get_response.context['messages']
+ }
+ expected_messages = {
+ (messages.SUCCESS, 'Successfully written the "Auth org id" field for this enterprise customer.'),
+ }
+ assert actual_messages == expected_messages
+
+ @mock.patch('enterprise.api_client.sso_orchestrator.EnterpriseSSOOrchestratorApiClient.configure_edx_oauth')
+ def test_post_api_raises_error(self, mock_configure_edx_oauth):
+ mock_configure_edx_oauth.side_effect = SsoOrchestratorClientError('foobar')
+ self._login()
+ response = self.client.post(self.view_url)
+ assert response.status_code == 500
+ mock_configure_edx_oauth.assert_called_once_with(self.enterprise_customer)
+ self.enterprise_customer.refresh_from_db()
+ assert self.enterprise_customer.auth_org_id is None
diff --git a/tests/test_enterprise/api_client/test_sso_orchestrator.py b/tests/test_enterprise/api_client/test_sso_orchestrator.py
index 683fe2fec6..02dbb39be5 100644
--- a/tests/test_enterprise/api_client/test_sso_orchestrator.py
+++ b/tests/test_enterprise/api_client/test_sso_orchestrator.py
@@ -11,12 +11,24 @@
from django.conf import settings
from enterprise.api_client import sso_orchestrator
-from enterprise.utils import get_sso_orchestrator_api_base_url, get_sso_orchestrator_configure_path
+from enterprise.utils import (
+ get_sso_orchestrator_api_base_url,
+ get_sso_orchestrator_configure_edx_oauth_path,
+ get_sso_orchestrator_configure_path,
+)
+from test_utils.factories import EnterpriseCustomerFactory
TEST_ENTERPRISE_ID = '1840e1dc-59cf-4a78-82c5-c5bbc0b5df0f'
TEST_ENTERPRISE_SSO_CONFIG_UUID = uuid4()
TEST_ENTERPRISE_NAME = 'Test Enterprise'
-SSO_ORCHESTRATOR_CONFIGURE_URL = urljoin(get_sso_orchestrator_api_base_url(), get_sso_orchestrator_configure_path())
+SSO_ORCHESTRATOR_CONFIGURE_URL = urljoin(
+ get_sso_orchestrator_api_base_url(),
+ get_sso_orchestrator_configure_path()
+)
+SSO_ORCHESTRATOR_CONFIGURE_EDX_OAUTH_URL = urljoin(
+ get_sso_orchestrator_api_base_url(),
+ get_sso_orchestrator_configure_edx_oauth_path()
+)
@responses.activate
@@ -51,3 +63,29 @@ def test_post_sso_configuration():
'name': TEST_ENTERPRISE_NAME,
'slug': TEST_ENTERPRISE_NAME
}
+
+
+@responses.activate
+def test_configure_edx_oauth():
+ """
+ Test the configure_edx_oauth method.
+ """
+ fake_enterprise_customer = EnterpriseCustomerFactory.stub()
+ fake_org_id = 'foobar'
+ responses.add(
+ responses.POST,
+ SSO_ORCHESTRATOR_CONFIGURE_EDX_OAUTH_URL,
+ json={'orgId': fake_org_id, 'status': 200},
+ )
+ client = sso_orchestrator.EnterpriseSSOOrchestratorApiClient()
+
+ # Call the method under test:
+ actual_response = client.configure_edx_oauth(enterprise_customer=fake_enterprise_customer)
+
+ assert actual_response == fake_org_id
+ responses.assert_call_count(count=1, url=SSO_ORCHESTRATOR_CONFIGURE_EDX_OAUTH_URL)
+
+ sent_body_params = json.loads(responses.calls[0].request.body)
+ assert sent_body_params['enterpriseName'] == fake_enterprise_customer.name
+ assert sent_body_params['enterpriseSlug'] == fake_enterprise_customer.slug
+ assert sent_body_params['enterpriseUuid'] == str(fake_enterprise_customer.uuid)