def test_is_enterprise_learner(self):
with mock.patch('django.core.cache.cache.set') as mock_cache_set:
EnterpriseCustomerUserFactory.create(active=True,
user_id=self.user.id)
self.assertTrue(is_enterprise_learner(self.user))
self.assertTrue(mock_cache_set.called)
def test_get_enterprise_learner_portal_no_branding_config(self):
"""
Test that only an enabled enterprise portal is returned,
and that it matches the customer UUID provided in the request,
even if no branding config is associated with the customer.
"""
enterprise_customer_user = EnterpriseCustomerUserFactory.create(
active=True, user_id=self.user.id)
enterprise_customer_user.enterprise_customer.enable_learner_portal = True
enterprise_customer_user.enterprise_customer.save()
request = mock.MagicMock(session={}, user=self.user)
# Indicate the "preferred" customer in the request
request.GET = {
'enterprise_customer':
enterprise_customer_user.enterprise_customer.uuid
}
portal = get_enterprise_learner_portal(request)
self.assertDictEqual(
portal, {
'name':
enterprise_customer_user.enterprise_customer.name,
'slug':
enterprise_customer_user.enterprise_customer.slug,
'logo':
enterprise_customer_user.enterprise_customer.
safe_branding_configuration.safe_logo_url,
})
def test_is_enterprise_customer_user(self):
"""
Verify that if user is an enterprise learner.
"""
# Create users from factory
user = UserFactory(username='******', email='*****@*****.**')
other_user = UserFactory(username='******', email='*****@*****.**')
customer_idp = EnterpriseCustomerIdentityProviderFactory.create(
provider_id='the-provider',
)
customer = customer_idp.enterprise_customer
EnterpriseCustomerUserFactory.create(
enterprise_customer=customer,
user_id=user.id,
)
assert is_enterprise_customer_user('the-provider', user)
assert not is_enterprise_customer_user('the-provider', other_user)
def test_unlink_enterprise_user_from_idp(self, mock_customer_from_request, mock_registry):
customer_idp = EnterpriseCustomerIdentityProviderFactory.create(
provider_id='the-provider',
)
customer = customer_idp.enterprise_customer
customer_user = EnterpriseCustomerUserFactory.create( # lint-amnesty, pylint: disable=unused-variable
enterprise_customer=customer,
user_id=self.user.id,
)
mock_customer_from_request.return_value = {
'uuid': customer.uuid,
}
mock_registry.get_enabled_by_backend_name.return_value = [
mock.Mock(provider_id='the-provider')
]
request = mock.Mock()
unlink_enterprise_user_from_idp(request, self.user, idp_backend_name='the-backend-name')
assert 0 == EnterpriseCustomerUser.objects.filter(user_id=self.user.id).count()
def test_user_enterprise(self, mock_get_programs_by_type):
mock_get_programs_by_type.return_value = [self.program]
EnterpriseCustomerUserFactory.create(user_id=self.user.id)
assert not show_user_demographics(user=self.user)
