Differential D5696 Diff 20443 swh/web/tests/create_test_users.py

Changeset View

Standalone View

swh/web/tests/create_test_users.py

	# Copyright (C) 2021 The Software Heritage developers			# Copyright (C) 2021 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 django.contrib.auth import get_user_model			from django.contrib.auth import get_user_model

				from swh.web.auth.utils import SWH_AMBASSADOR_PERMISSION
				from swh.web.tests.utils import create_django_permission

	User = get_user_model()			User = get_user_model()

	username = "user"
	password = "user"
	email = "user@swh-web.org"

				users: Dict[str, Tuple[str, str, List[str]]] = {
				"user": ("user", "user@swh-web.org", []),
				"ambassador": ("ambassador", "ambassador@swh-web.org", [SWH_AMBASSADOR_PERMISSION]),
				}

				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.objects.create_user(username, email, password)			user = User.objects.create_user(username, email, password)
				if permissions:
				for perm_name in permissions:
				permission = create_django_permission(perm_name)
				user.user_permissions.add(permission)

				user.save()
				anlambertUnsubmitted Done Inline Actions you can remove that assert as the code is already tested anlambert: you can remove that assert as the code is already tested