From 6b04b7bdf43ea2e9d86384a03a253ac9c131c96d Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Wed, 1 Dec 2010 19:36:37 +0300
Subject: [PATCH] diagostics for EA-24292 - CCE:
 PyElementGeneratorImpl.createCallExpression

---
 .../jetbrains/python/psi/impl/PyElementGeneratorImpl.java   | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/python/src/com/jetbrains/python/psi/impl/PyElementGeneratorImpl.java b/python/src/com/jetbrains/python/psi/impl/PyElementGeneratorImpl.java
index 4edd972cad29..6af5421c44b0 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyElementGeneratorImpl.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyElementGeneratorImpl.java
@@ -198,7 +198,11 @@ public class PyElementGeneratorImpl extends PyElementGenerator {
 
   public PyCallExpression createCallExpression(String functionName) {
     final PsiFile dummyFile = createDummyFile(LanguageLevel.getDefault(), functionName + "()");
-    return (PyCallExpression)dummyFile.getFirstChild().getFirstChild();
+    final PsiElement element = dummyFile.getFirstChild().getFirstChild();
+    if (!(element instanceof PyCallExpression)) {
+      throw new IllegalArgumentException("Invalid call expression text " + functionName);
+    }
+    return (PyCallExpression)element;
   }
 
   public PyImportStatement createImportStatementFromText(final String text) {