Adds a test for property setter

pylint-dev · Feb 10, 2021 · 7179392 · 7179392
1 parent 61a0347
commit 7179392
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tests/unittest_inference.py b/tests/unittest_inference.py
@@ -5494,14 +5494,19 @@ class A:
         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(self):
     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)