Пример #1
0
def _mock_logging_defaults():
    # Pre-populate settings obj with defaults
    class MockSettings:
        pass
    mock_settings_obj = MockSettings()
    mock_settings_json = dict()
    for key in settings_registry.get_registered_settings(category_slug='logging'):
        value = settings_registry.get_setting_field(key).get_default()
        setattr(mock_settings_obj, key, value)
        mock_settings_json[key] = value
    setattr(mock_settings_obj, 'MAX_EVENT_RES_DATA', 700000)
    return mock_settings_obj, mock_settings_json
Пример #2
0
    def mock_settings(self, request):
        fixture_args = request.node.get_closest_marker('fixture_args')
        if fixture_args and 'autocreate' in fixture_args.kwargs:
            autocreate = fixture_args.kwargs['autocreate']
        else:
            autocreate = True

        class MockSettings:
            SAML_AUTO_CREATE_OBJECTS = autocreate
            SOCIAL_AUTH_SAML_ORGANIZATION_ATTR = {
                'saml_attr': 'memberOf',
                'saml_admin_attr': 'admins',
                'saml_auditor_attr': 'auditors',
                'remove': True,
                'remove_admins': True,
            }
            SOCIAL_AUTH_SAML_TEAM_ATTR = {
                'saml_attr':
                'groups',
                'remove':
                True,
                'team_org_map': [
                    {
                        'team': 'Blue',
                        'organization': 'Default1'
                    },
                    {
                        'team': 'Blue',
                        'organization': 'Default2'
                    },
                    {
                        'team': 'Blue',
                        'organization': 'Default3'
                    },
                    {
                        'team': 'Red',
                        'organization': 'Default1'
                    },
                    {
                        'team': 'Green',
                        'organization': 'Default1'
                    },
                    {
                        'team': 'Green',
                        'organization': 'Default3'
                    },
                    {
                        'team': 'Yellow',
                        'team_alias': 'Yellow_Alias',
                        'organization': 'Default4',
                        'organization_alias': 'Default4_Alias'
                    },
                ],
            }

        mock_settings_obj = MockSettings()
        for key in settings_registry.get_registered_settings(
                category_slug='logging'):
            value = settings_registry.get_setting_field(key).get_default()
            setattr(mock_settings_obj, key, value)
        setattr(mock_settings_obj, 'DEBUG', True)

        return mock_settings_obj