added tests for PyUtil.addElementToStatementList

python/testData/quickFixes/AddMethodQuickFixTest/addInitAfterDocstring.py

@@ -0,0 +1,5 @@
+__author__ = 'ktisha'
+class <caret>A(object):
+    """ class docstring"""
+    def foo(self):
+        self.b = 1

python/testData/quickFixes/AddMethodQuickFixTest/addInitAfterDocstring_after.py

@@ -0,0 +1,9 @@
+__author__ = 'ktisha'
+class <caret>A(object):
+    """ class docstring"""
+
+    def __init__(self):
+        pass
+
+    def foo(self):
+        self.b = 1

python/testData/quickFixes/AddMethodQuickFixTest/addMethodReplacePass.py

@@ -0,0 +1,5 @@
+class A:
+    pass
+
+a = A()
+a.<caret><warning descr="Unresolved attribute reference 'y' for class 'A'">y</warning>()

python/testData/quickFixes/AddMethodQuickFixTest/addMethodReplacePass_after.py

@@ -0,0 +1,6 @@
+class A:
+    def y(self):
+        pass
+
+a = A()
+a.y()

python/testSrc/com/jetbrains/python/quickFixes/AddMethodQuickFixTest.java

@@ -13,6 +13,14 @@ public class AddMethodQuickFixTest extends PyQuickFixTestCase {
     doInspectionTest(PyClassHasNoInitInspection.class, PyBundle.message("QFIX.NAME.add.method.$0.to.class.$1", "__init__", "A"));
   }
 
+  public void testAddInitAfterDocstring() {
+    doInspectionTest(PyClassHasNoInitInspection.class, PyBundle.message("QFIX.NAME.add.method.$0.to.class.$1", "__init__", "A"));
+  }
+
+  public void testAddMethodReplacePass() {
+    doInspectionTest(PyUnresolvedReferencesInspection.class, PyBundle.message("QFIX.NAME.add.method.$0.to.class.$1", "y", "A"));
+  }
+
   public void testAddMethodFromInstance() {
     doInspectionTest(PyUnresolvedReferencesInspection.class, PyBundle.message("QFIX.NAME.add.method.$0.to.class.$1", "y", "A"));
   }