[Fix] Failed Unittest

Signed-off-by: Kent Huang <kent@infuseai.io>

recce/server.py

@@ -166,6 +166,7 @@ async def config_flag():
 async def mark_onboarding_completed():
     context = default_context()
     context.mark_onboarding_completed()
+    app.state.flag['show_onboarding_guide'] = False
 
 
 @app.get("/api/info")

recce/util/recce_cloud.py

@@ -127,10 +127,13 @@ def set_onboarding_state(self, state: str):
 
 
 def get_recce_cloud_onboarding_state(token: str) -> str:
-    recce_cloud = RecceCloud(token)
-    user_info = recce_cloud.get_user_info()
-    if user_info:
-        return user_info.get('onboarding_state')
+    try:
+        recce_cloud = RecceCloud(token)
+        user_info = recce_cloud.get_user_info()
+        if user_info:
+            return user_info.get('onboarding_state')
+    except Exception as e:
+        logger.debug(str(e))
     return 'undefined'
 
 

tests/test_cli.py

@@ -39,12 +39,14 @@ def test_cmd_server_with_cloud_without_token(self, mock_run):
         result = self.runner.invoke(cli_command_server, ['--cloud', '--password', 'unittest'])
         assert result.exit_code == 1
 
+    @patch('recce.util.recce_cloud.get_recce_cloud_onboarding_state')
     @patch('recce.cli.uvicorn.run')
     @patch('recce.cli.RecceStateLoader')
-    def test_cmd_server_with_cloud(self, mock_state_loader_class, mock_run):
+    def test_cmd_server_with_cloud(self, mock_state_loader_class, mock_run, mock_get_recce_cloud_onboarding_state):
         mock_state_loader = MagicMock(spec=RecceStateLoader)
         mock_state_loader.verify.return_value = True
         mock_state_loader.review_mode = True
+        mock_get_recce_cloud_onboarding_state.return_value = 'completed'
 
         mock_state_loader_class.return_value = mock_state_loader
         self.runner.invoke(cli_command_server, ['--cloud', '--password', 'unittest', '--cloud-token', 'unittest'])