sentry/tests/acceptance/test_organization_settings.py

from __future__ import absolute_import

from sentry.testutils import AcceptanceTestCase


class OrganizationSettingsTest(AcceptanceTestCase):
    def setUp(self):
        super(OrganizationSettingsTest, self).setUp()
        self.user = self.create_user('foo@example.com')
        self.org = self.create_organization(
            name='Rowdy Tiger',
            owner=None,
        )
        self.team = self.create_team(organization=self.org, name='Mariachi Band')
        self.project = self.create_project(
            organization=self.org,
            teams=[self.team],
            name='Bengal',
        )
        self.create_member(
            user=self.user,
            organization=self.org,
            role='owner',
            teams=[self.team],
        )
        self.login_as(self.user)
        self.path = u'/organizations/{}/settings/'.format(self.org.slug)

    def load_organization_helper(self, snapshot_name=None):
        self.browser.wait_until_not('.loading-indicator')
        if snapshot_name is not None:
            self.browser.snapshot('organization settings -- ' + snapshot_name)
        assert self.browser.element_exists('.ref-organization-settings')

    def renders_2fa_setting(self):
        return self.browser.element_exists('#require2FA')

    def test_disabled_2fa_feature(self):
        user_owner = self.create_user('owner@example.com')
        organization = self.create_organization(name="Example", owner=user_owner)
        self.login_as(user_owner)
        path = '/organizations/%s/settings/' % organization.slug

        self.browser.get(path)
        self.load_organization_helper()
        assert not self.renders_2fa_setting()

    def test_renders_2fa_setting_for_owner(self):
        user_owner = self.create_user('owner@example.com')
        organization = self.create_organization(name="Example", owner=user_owner)
        self.login_as(user_owner)
        path = '/organizations/%s/settings/' % organization.slug

        with self.feature('organizations:require-2fa'):
            self.browser.get(path)
            self.load_organization_helper()
            assert self.renders_2fa_setting()

    def test_renders_2fa_setting_for_manager(self):
        user_manager = self.create_user('manager@gexample.com')
        organization = self.create_organization(
            name="Example", owner=self.create_user('owner@example.com'))
        self.create_member(organization=organization, user=user_manager, role='manager')
        self.login_as(user_manager)
        path = '/organizations/%s/settings/' % organization.slug

        with self.feature('organizations:require-2fa'):
            self.browser.get(path)
            self.load_organization_helper()
            assert self.renders_2fa_setting()

    def test_setting_2fa_without_2fa_enabled(self):
        user_owner = self.create_user('owner@example.com')
        organization = self.create_organization(name="Example", owner=user_owner)
        self.login_as(user_owner)
        path = '/organizations/%s/settings/' % organization.slug

        with self.feature('organizations:require-2fa'):
            self.browser.get(path)
            self.browser.wait_until_not('.loading-indicator')
            assert not self.browser.element_exists('.ref-organization-settings .error')
            self.browser.click('#require2FA')

            self.browser.wait_until('.modal')
            self.browser.click('.modal [data-test-id="confirm-modal"]')
            self.browser.wait_until_not('.modal')
            self.browser.wait_until('.ref-toast.ref-error')
            self.load_organization_helper("setting 2fa without 2fa enabled")