diff --git a/tjdests/test.py b/tjdests/test.py index d4a669c..7cda356 100644 --- a/tjdests/test.py +++ b/tjdests/test.py @@ -11,6 +11,7 @@ class TJDestsTestCase(TestCase): make_student: bool = False, make_senior: bool = False, make_superuser: bool = False, + publish_data: bool = False, ) -> User: """ Log in as the specified user. @@ -21,6 +22,7 @@ class TJDestsTestCase(TestCase): make_student: Whether to make this user a student. make_senior: Whether to make this user a senior. make_superuser: Whether to make this user a superuser. + publish_data: Whether to publish this user's data. Return: The user. """ @@ -32,6 +34,7 @@ class TJDestsTestCase(TestCase): "is_superuser": make_superuser, "is_senior": make_senior, "accepted_terms": accept_tos, + "publish_data": publish_data, }, )[0] self.client.force_login(user)