Fix error messages in test cases (#26)

Author: hanzjk

libs/amp-instrumentation/tests/test_cli.py

@@ -101,8 +101,7 @@ def test_bootstrap_directory_not_found(self, tmp_path):
 
         # Check that error message was printed to stderr
         stderr_output = mock_stderr.getvalue()
-        assert "Error: Bootstrap directory not found" in stderr_output
-        assert "Package may not be properly installed" in stderr_output
+        assert "Error: Package installation is incomplete" in stderr_output
         assert "pip install --force-reinstall" in stderr_output
 
 
@@ -131,10 +130,8 @@ def test_warns_when_sitecustomize_exists(self, tmp_path):
 
                 stderr_output = mock_stderr.getvalue()
                 # Verify warning was printed
-                assert "Warning: Found existing sitecustomize.py" in stderr_output
-                assert (
-                    "This may conflict with WSO2 AMP instrumentation" in stderr_output
-                )
+                assert "Warning: Found sitecustomize.py" in stderr_output
+                assert "may conflict with instrumentation" in stderr_output
         finally:
             os.chdir(original_cwd)
 

libs/amp-instrumentation/tests/test_initialization.py

@@ -30,7 +30,7 @@ def test_missing_variable_raises_error(self, clean_environment):
         with pytest.raises(initialization.ConfigurationError) as exc_info:
             initialization._get_required_env_var("MISSING_VAR")
         assert "MISSING_VAR" in str(exc_info.value)
-        assert "not set or is empty" in str(exc_info.value)
+        assert "is required but not set" in str(exc_info.value)
 
     def test_empty_variable_raises_error(self, clean_environment):
         """Test that empty variable raises ConfigurationError."""
@@ -54,9 +54,7 @@ def test_successful_initialization(self, clean_environment, mock_traceloop):
         """Test successful initialization with all required env vars."""
         # Set required environment variables
         os.environ[env_vars.AMP_AGENT_NAME] = "test-app"
-        os.environ[env_vars.AMP_OTEL_ENDPOINT] = (
-            "https://otel.example.com"
-        )
+        os.environ[env_vars.AMP_OTEL_ENDPOINT] = "https://otel.example.com"
         os.environ[env_vars.AMP_AGENT_API_KEY] = "test-key"
 
         # Reset initialization state

libs/amp-instrumentation/tests/test_sitecustomize.py

@@ -44,8 +44,8 @@ def test_sitecustomize_initialization_failure_exits_with_error():
     assert result.returncode == 1, "Expected exit code 1 when initialization fails"
 
     # Should print error message to stderr
-    assert "ERROR: WSO2 AMP instrumentation failed" in result.stderr
-    assert "Check your environment variables and configuration" in result.stderr
+    assert "Error: Environment variable" in result.stderr
+    assert "is required but not set" in result.stderr
 
 
 def test_sitecustomize_successful_initialization():