@@ -123,7 +123,7 @@ async def test_bootstrap_admin_user_already_exists(self, mock_settings, mock_db_
123123 async def test_bootstrap_admin_user_success (self , mock_settings , mock_db_session , mock_email_auth_service , mock_admin_user ):
124124 """Test successful admin user creation."""
125125 mock_email_auth_service .get_user_by_email .return_value = None
126- mock_email_auth_service .create_user .return_value = mock_admin_user
126+ mock_email_auth_service .create_platform_admin .return_value = mock_admin_user
127127
128128 with (
129129 patch ("mcpgateway.bootstrap_db.settings" , mock_settings ),
@@ -135,22 +135,19 @@ async def test_bootstrap_admin_user_success(self, mock_settings, mock_db_session
135135 mock_utc_now .return_value = "2024-01-01T00:00:00Z"
136136 await bootstrap_admin_user ()
137137
138- mock_email_auth_service .create_user .assert_called_once_with (
138+ mock_email_auth_service .create_platform_admin .assert_called_once_with (
139139 email = mock_settings .platform_admin_email ,
140140 password = mock_settings .platform_admin_password .get_secret_value (),
141141 full_name = mock_settings .platform_admin_full_name ,
142- is_admin = True ,
143142 )
144- assert mock_admin_user .email_verified_at == "2024-01-01T00:00:00Z"
145- assert mock_db_session .commit .call_count == 2
146- mock_logger .info .assert_any_call (f"Platform admin user created successfully: { mock_settings .platform_admin_email } " )
143+ mock_logger .info .assert_any_call (f"Creating platform admin user: { mock_settings .platform_admin_email } " )
147144
148145 @pytest .mark .asyncio
149146 async def test_bootstrap_admin_user_with_personal_team (self , mock_settings , mock_db_session , mock_email_auth_service , mock_admin_user ):
150147 """Test admin user creation with personal team auto-creation."""
151148 mock_settings .auto_create_personal_teams = True
152149 mock_email_auth_service .get_user_by_email .return_value = None
153- mock_email_auth_service .create_user .return_value = mock_admin_user
150+ mock_email_auth_service .create_platform_admin .return_value = mock_admin_user
154151
155152 with patch ("mcpgateway.bootstrap_db.settings" , mock_settings ):
156153 with patch ("mcpgateway.bootstrap_db.SessionLocal" , return_value = mock_db_session ):
@@ -159,7 +156,8 @@ async def test_bootstrap_admin_user_with_personal_team(self, mock_settings, mock
159156 with patch ("mcpgateway.bootstrap_db.logger" ) as mock_logger :
160157 await bootstrap_admin_user ()
161158
162- mock_logger .info .assert_any_call ("Personal team automatically created for admin user" )
159+ # Verify that the user creation was attempted
160+ mock_email_auth_service .create_platform_admin .assert_called_once ()
163161
164162 @pytest .mark .asyncio
165163 async def test_bootstrap_admin_user_exception (self , mock_settings , mock_db_session , mock_email_auth_service ):
0 commit comments