Enhance _check_reference to unwrap ObjectProxy for assertions

Author: LuKrO2011

src/pynguin/assertion/assertiontraceobserver.py

@@ -23,6 +23,7 @@
 import pynguin.testcase.testcase as tc
 import pynguin.testcase.variablereference as vr
 import pynguin.utils.generic.genericaccessibleobject as gao
+import pynguin.utils.typetracing as tt
 from pynguin.analyses.typesystem import ANY, TypeInfo
 from pynguin.utils.type_utils import (
     is_assertable,
@@ -120,7 +121,8 @@ def _handle(  # noqa: C901
         trace = self._assertion_local_state.trace
 
         if not statement.ret_val.is_none_type():
-            if is_primitive_type(type(exec_ctx.get_reference_value(statement.ret_val))):
+            ret_value = tt.unwrap(exec_ctx.get_reference_value(statement.ret_val))
+            if is_primitive_type(type(ret_value)):
                 # Primitives won't change, so we only check them once.
                 self._check_reference(module_provider, exec_ctx, statement.ret_val, position, trace)
             elif type(exec_ctx.get_reference_value(statement.ret_val)).__module__ != "builtins":
@@ -155,7 +157,7 @@ def _handle(  # noqa: C901
 
         # Check fields of classes whose constructors were used.
         for seen_type in [
-            type(exec_ctx.get_reference_value(ref))
+            type(tt.unwrap(exec_ctx.get_reference_value(ref)))
             for ref in self._assertion_local_state.watch_list
         ]:
             if (
@@ -208,6 +210,7 @@ def _check_reference(
             max_depth: The maximum recursion depth.
         """
         value = exec_ctx.get_reference_value(ref)
+        value = tt.unwrap(value)
         if isinstance(value, float):
             trace.add_entry(position, ass.FloatAssertion(ref, value))
             return

tests/assertion/test_assertiontraceobserver.py

@@ -8,7 +8,9 @@
 from unittest import mock
 from unittest.mock import MagicMock
 
+import pynguin.assertion.assertion as ass
 import pynguin.assertion.assertiontraceobserver as ato
+import pynguin.utils.typetracing as tt
 from pynguin.testcase.execution import ExecutionContext, TestCaseExecutor
 from pynguin.testcase.statement import Statement
 
@@ -47,3 +49,46 @@ def test_after_test_case_execution():
         trace_mock.clone.return_value = clone
         observer.after_test_case_execution(MagicMock(), MagicMock(), result)
         assert result.assertion_trace == clone
+
+
+def test_check_reference_unwraps_object_proxy():
+    """Regression test: _check_reference must unwrap ObjectProxy to generate assertions."""
+    observer = ato.RemoteAssertionTraceObserver()
+
+    wrapped_value = "test_string"
+    proxy = tt.ObjectProxy(wrapped_value)
+
+    exec_ctx = MagicMock()
+    exec_ctx.get_reference_value.return_value = proxy
+    module_provider = MagicMock()
+    ref = MagicMock()
+
+    trace = observer._assertion_local_state.trace
+    observer._check_reference(module_provider, exec_ctx, ref, position=0, trace=trace)
+
+    assertions = list(trace.trace.get(0, []))
+    assert len(assertions) == 1
+    assert isinstance(assertions[0], ass.ObjectAssertion)
+    assert assertions[0].object == wrapped_value
+
+
+def test_check_reference_unwraps_nested_object_proxy():
+    """Regression test: _check_reference handles nested ObjectProxy correctly."""
+    observer = ato.RemoteAssertionTraceObserver()
+
+    inner_value = 42
+    inner_proxy = tt.ObjectProxy(inner_value)
+    outer_proxy = tt.ObjectProxy(inner_proxy)
+
+    exec_ctx = MagicMock()
+    exec_ctx.get_reference_value.return_value = outer_proxy
+    module_provider = MagicMock()
+    ref = MagicMock()
+
+    trace = observer._assertion_local_state.trace
+    observer._check_reference(module_provider, exec_ctx, ref, position=0, trace=trace)
+
+    assertions = list(trace.trace.get(0, []))
+    assert len(assertions) == 1
+    assert isinstance(assertions[0], ass.ObjectAssertion)
+    assert assertions[0].object == inner_value