Adapt to changed YAML Psi to find the first KeyValue mapping

Author: cedricziel

src/main/java/fr/adrienbrault/idea/symfony2plugin/doctrine/metadata/util/DoctrineMetadataUtil.java

@@ -316,9 +316,9 @@ public static String findModelNameInScope(@NotNull PsiElement psiElement) {
             if(yamlDocument != null) {
                 YAMLValue topLevelValue = yamlDocument.getTopLevelValue();
                 if(topLevelValue instanceof YAMLMapping) {
-                    PsiElement firstChild = topLevelValue.getFirstChild();
-                    if(firstChild instanceof YAMLKeyValue) {
-                        String keyText = ((YAMLKeyValue) firstChild).getKeyText();
+                    YAMLKeyValue firstChild = PsiTreeUtil.findChildOfType(topLevelValue, YAMLKeyValue.class);
+                    if(firstChild != null) {
+                        String keyText = firstChild.getKeyText();
                         if(StringUtils.isNotBlank(keyText)) {
                             return keyText;
                         }

src/test/java/fr/adrienbrault/idea/symfony2plugin/tests/doctrine/metadata/util/DoctrineMetadataUtilTest.java

@@ -15,7 +15,6 @@
 import fr.adrienbrault.idea.symfony2plugin.util.PhpElementsUtil;
 import org.jetbrains.yaml.YAMLFileType;
 
-import java.io.File;
 import java.util.*;
 
 /**
@@ -236,7 +235,7 @@ public void testYamlFindModelNameInScope() {
         }) {
             myFixture.configureByText(YAMLFileType.YML, s);
             PsiElement psiElement = myFixture.getFile().findElementAt(myFixture.getCaretOffset());
-            assertEquals("Foo\\Foo", DoctrineMetadataUtil.findModelNameInScope(psiElement));
+            assertEquals(String.format("Fixture %s", s), "Foo\\Foo", DoctrineMetadataUtil.findModelNameInScope(psiElement));
         }
     }
 }