Differential D7365 Diff 26648 swh/web/tests/create_test_users.py

Changeset View

Standalone View

swh/web/tests/create_test_users.py

	# Copyright (C) 2021-2022 The Software Heritage developers			# Copyright (C) 2021-2022 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution			# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU Affero General Public License version 3, or any later version			# License: GNU Affero General Public License version 3, or any later version
	# See top-level LICENSE file for more information			# See top-level LICENSE file for more information

	from typing import Dict, List, Tuple			from typing import Dict, List, Tuple

	from django.contrib.auth import get_user_model			from django.contrib.auth import get_user_model

	from swh.web.auth.utils import ADMIN_LIST_DEPOSIT_PERMISSION, SWH_AMBASSADOR_PERMISSION			from swh.web.auth.utils import (
				ADD_FORGE_MODERATOR_PERMISSION,
				ADMIN_LIST_DEPOSIT_PERMISSION,
				SWH_AMBASSADOR_PERMISSION,
				)
	from swh.web.tests.utils import create_django_permission			from swh.web.tests.utils import create_django_permission

	User = get_user_model()			User = get_user_model()


	users: Dict[str, Tuple[str, str, List[str]]] = {			users: Dict[str, Tuple[str, str, List[str]]] = {
	"user": ("user", "user@example.org", []),			"user": ("user", "user@example.org", []),
	"ambassador": (			"ambassador": (
	"ambassador",			"ambassador",
	"ambassador@example.org",			"ambassador@example.org",
	[SWH_AMBASSADOR_PERMISSION],			[SWH_AMBASSADOR_PERMISSION],
	),			),
				anlambertUnsubmitted Done Inline Actions I would rather name that user `add-forge-moderator`. anlambert: I would rather name that user `add-forge-moderator`.
	"deposit": ("deposit", "deposit@example.org", [ADMIN_LIST_DEPOSIT_PERMISSION],),			"deposit": ("deposit", "deposit@example.org", [ADMIN_LIST_DEPOSIT_PERMISSION],),
				"add-forge-moderator": (
				"add-forge-moderator",
				"moderator@example.org",
				[ADD_FORGE_MODERATOR_PERMISSION],
				),
	}			}


	for username, (password, email, permissions) in users.items():			for username, (password, email, permissions) in users.items():
	if not User.objects.filter(username=username).exists():			if not User.objects.filter(username=username).exists():
	user = User.objects.create_user(username, email, password)			user = User.objects.create_user(username, email, password)
	if permissions:			if permissions:
	for perm_name in permissions:			for perm_name in permissions:
	permission = create_django_permission(perm_name)			permission = create_django_permission(perm_name)
	user.user_permissions.add(permission)			user.user_permissions.add(permission)

	user.save()			user.save()