From 7516ce91ad9be6b9dd8c97ed150ac4ed852194ba Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@intellij.com>
Date: Thu, 9 Jul 2009 13:54:46 +0400
Subject: [PATCH] QuickFixTest fixed

---
 python/src/com/jetbrains/python/actions/AddMethodQuickFix.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/python/src/com/jetbrains/python/actions/AddMethodQuickFix.java b/python/src/com/jetbrains/python/actions/AddMethodQuickFix.java
index 6a5b8913f8eb..fa0f844eaff2 100644
--- a/python/src/com/jetbrains/python/actions/AddMethodQuickFix.java
+++ b/python/src/com/jetbrains/python/actions/AddMethodQuickFix.java
@@ -1,5 +1,6 @@
 package com.jetbrains.python.actions;
 
+import com.intellij.codeInsight.CodeInsightUtilBase;
 import com.intellij.codeInsight.template.TemplateBuilder;
 import com.intellij.codeInsight.template.TemplateBuilderFactory;
 import com.intellij.codeInspection.LocalQuickFix;
@@ -88,6 +89,7 @@ public class AddMethodQuickFix implements LocalQuickFix {
   }
 
   private static void showTemplateBuilder(PyFunction method) {
+    method = CodeInsightUtilBase.forcePsiPostprocessAndRestoreElement(method);
     TemplateBuilder builder = TemplateBuilderFactory.getInstance().createTemplateBuilder(method);
     PyParameter[] parameters = method.getParameterList().getParameters();
     for (int i = 1; i < parameters.length; i++) {