From 195f1d90a471951ad418ad1fe444a5ee2adb1ac2 Mon Sep 17 00:00:00 2001 From: SKairinos Date: Thu, 25 Jan 2024 10:01:52 +0000 Subject: [PATCH] make is_admin optional --- codeforlife/tests/model_view_set.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/codeforlife/tests/model_view_set.py b/codeforlife/tests/model_view_set.py index ff85fc45..5c99fc9b 100644 --- a/codeforlife/tests/model_view_set.py +++ b/codeforlife/tests/model_view_set.py @@ -424,20 +424,23 @@ def login(self, **credentials): return user - def login_teacher(self, is_admin: bool, **credentials): + def login_teacher(self, is_admin: t.Optional[bool] = None, **credentials): + # pylint: disable=line-too-long """Log in a user and assert they are a teacher. Args: - is_admin: Whether or not the teacher is an admin. + is_admin: Whether or not the teacher is an admin. Set none if a teacher can be either or. Returns: The teacher-user. """ + # pylint: enable=line-too-long user = self.login(**credentials) assert user.teacher assert user.teacher.school - assert is_admin == user.teacher.is_admin + if is_admin is not None: + assert is_admin == user.teacher.is_admin return user def login_student(self, **credentials):