From eecb83fc6685ce312b85f30efabd42598ae3d73d Mon Sep 17 00:00:00 2001
From: Christopher Setzer <37311998+cmsetzer@users.noreply.github.com>
Date: Thu, 5 Sep 2024 14:00:21 -0800
Subject: [PATCH] Fix test failing due to would_be_org_admin widget change

---
 perma_web/perma/tests/test_views_user_management.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/perma_web/perma/tests/test_views_user_management.py b/perma_web/perma/tests/test_views_user_management.py
index f878f349d..58dd3b5d2 100644
--- a/perma_web/perma/tests/test_views_user_management.py
+++ b/perma_web/perma/tests/test_views_user_management.py
@@ -2310,6 +2310,7 @@ def test_new_firm_success(self):
             'sign_up_firm',
             data={
                 'e-address': self.randomize_capitalization(existing_user['email']),
+                'would_be_org_admin': firm_user_form['would_be_org_admin'],
                 **firm_organization_form,
                 **firm_usage_form,
             },
@@ -2324,6 +2325,7 @@ def test_new_firm_success(self):
             'sign_up_firm',
             data={
                 'e-address': firm_user_form['raw_email'],
+                'would_be_org_admin': firm_user_form['would_be_org_admin'],
                 **firm_organization_form,
                 **firm_usage_form,
             },
@@ -2338,6 +2340,7 @@ def test_new_firm_success(self):
             'sign_up_firm',
             data={
                 'e-address': firm_user_form['raw_email'],
+                'would_be_org_admin': firm_user_form['would_be_org_admin'],
                 **firm_organization_form,
                 **firm_usage_form,
                 'create_account': True,
@@ -2360,6 +2363,7 @@ def test_new_firm_success(self):
             'sign_up_firm',
             data={
                 'e-address': firm_user_form['raw_email'],
+                'would_be_org_admin': firm_user_form['would_be_org_admin'],
                 **firm_organization_form,
                 **firm_usage_form,
                 'create_account': True,
@@ -2380,6 +2384,7 @@ def test_new_firm_success(self):
             'sign_up_firm',
             data={
                 'e-address': self.randomize_capitalization(existing_user['email']),
+                'would_be_org_admin': firm_user_form['would_be_org_admin'],
                 **firm_organization_form,
                 **firm_usage_form,
                 'create_account': True,
@@ -2433,7 +2438,7 @@ def test_new_firm_failure(self):
             data={},
             form_keys=['organization_form', 'usage_form', 'user_form'],
             user='test_user@example.com',
-            error_keys=['email'],
+            error_keys=['email', 'would_be_org_admin'],
         )
         self.assertEqual(len(mail.outbox), 0)