Adds a test for property setter

1 files changed, 10 insertions, 1 deletions

diff --git a/tests/unittest_inference.py b/tests/unittest_inference.py
index e580ee2a..c107eadb 100644
--- a/tests/unittest_inference.py
+++ b/tests/unittest_inference.py
@@ -5494,14 +5494,19 @@ def test_property_inference():
         def test(self):
             return 42
 
+        @test.setter
+        def test(self, value):
+            return "banco"
+
     A.test #@
     A().test #@
     A.test.fget(A) #@
+    A.test.fset(A, "a_value") #@
     A.test.setter #@
     A.test.getter #@
     A.test.deleter #@
     """
-    prop, prop_result, prop_fget_result, prop_setter, prop_getter, prop_deleter = extract_node(
+    prop, prop_result, prop_fget_result, prop_fset_result, prop_setter, prop_getter, prop_deleter = extract_node(
         code
     )
 
@@ -5519,6 +5524,10 @@ def test_property_inference():
     assert isinstance(inferred, nodes.Const)
     assert inferred.value == 42
 
+    inferred = next(prop_fset_result.infer())
+    assert isinstance(inferred, nodes.Const)
+    assert inferred.value == "banco"
+
     for prop_func in prop_setter, prop_getter, prop_deleter:
         inferred = next(prop_func.infer())
         assert isinstance(inferred, nodes.FunctionDef)