plugins/maven/src/test/java/org/jetbrains/idea/maven/dom/MavenModelReadingAndWritingTest.java

/*
 * Copyright 2000-2009 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package org.jetbrains.idea.maven.dom;

import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.command.CommandProcessor;
import com.intellij.openapi.editor.Document;
import com.intellij.openapi.fileEditor.FileDocumentManager;
import com.intellij.openapi.vfs.VfsUtil;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiManager;
import com.intellij.psi.codeStyle.CodeStyleManager;
import com.intellij.util.IncorrectOperationException;
import org.jetbrains.idea.maven.MavenImportingTestCase;
import org.jetbrains.idea.maven.dom.model.MavenDomDependency;
import org.jetbrains.idea.maven.dom.model.MavenDomProjectModel;

public class MavenModelReadingAndWritingTest extends MavenImportingTestCase {
  @Override
  protected void setUpInWriteAction() throws Exception {
    super.setUpInWriteAction();

    importProject("<groupId>test</groupId>" +
                  "<artifactId>project</artifactId>" +
                  "<version>1</version>");
  }

  public void testReading() throws Exception {
    MavenDomProjectModel model = getDomModel();

    assertEquals("test", model.getGroupId().getStringValue());
    assertEquals("project", model.getArtifactId().getStringValue());
    assertEquals("1", model.getVersion().getStringValue());
  }

  public void testWriting() throws Exception {
    CommandProcessor.getInstance().executeCommand(myProject, new Runnable() {
      @Override
      public void run() {
        ApplicationManager.getApplication().runWriteAction(new Runnable() {
          public void run() {
            MavenDomProjectModel model = getDomModel();

            model.getGroupId().setStringValue("foo");
            model.getArtifactId().setStringValue("bar");
            model.getVersion().setStringValue("baz");

            formatAndSaveProjectPomDocument();
          }
        });
      }
    }, null, null);

    assertSameLines("<?xml version=\"1.0\"?>\r\n" +
                    "<project xmlns=\"http://maven.apache.org/POM/4.0.0\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\r\n" +
                    "         xsi:schemaLocation=\"http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd\">\r\n" +
                    "    <modelVersion>4.0.0</modelVersion>\r\n" +
                    "    <groupId>foo</groupId>\r\n" +
                    "    <artifactId>bar</artifactId>\r\n" +
                    "    <version>baz</version>\r\n" +
                    "</project>",
                    VfsUtil.loadText(myProjectPom));
  }

  public void testAddingADependency() throws Exception {
    CommandProcessor.getInstance().executeCommand(myProject, new Runnable() {
      @Override
      public void run() {
        ApplicationManager.getApplication().runWriteAction(new Runnable() {
          public void run() {
            MavenDomProjectModel model = getDomModel();

            MavenDomDependency d = model.getDependencies().addDependency();
            d.getGroupId().setStringValue("group");
            d.getArtifactId().setStringValue("artifact");
            d.getVersion().setStringValue("version");

            formatAndSaveProjectPomDocument();
          }
        });
      }
    }, null, null);

    assertSameLines("<?xml version=\"1.0\"?>\r\n" +
                    "<project xmlns=\"http://maven.apache.org/POM/4.0.0\" xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\"\r\n" +
                    "         xsi:schemaLocation=\"http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd\">\r\n" +
                    "    <modelVersion>4.0.0</modelVersion>\r\n" +
                    "    <groupId>test</groupId>\r\n" +
                    "    <artifactId>project</artifactId>\r\n" +
                    "    <version>1</version>\r\n" +
                    "    <dependencies>\r\n" +
                    "        <dependency>\r\n" +
                    "            <groupId>group</groupId>\r\n" +
                    "            <artifactId>artifact</artifactId>\r\n" +
                    "            <version>version</version>\r\n" +
                    "        </dependency>\r\n" +
                    "    </dependencies>\r\n" +
                    "</project>", VfsUtil.loadText(myProjectPom));
  }

  private MavenDomProjectModel getDomModel() {
    return MavenDomUtil.getMavenDomProjectModel(myProject, myProjectPom);
  }

  private void formatAndSaveProjectPomDocument() {
    try {
      PsiFile psiFile = PsiManager.getInstance(myProject).findFile(myProjectPom);
      CodeStyleManager.getInstance(myProject).reformat(psiFile);
      Document d = FileDocumentManager.getInstance().getDocument(myProjectPom);
      FileDocumentManager.getInstance().saveDocument(d);
    }
    catch (IncorrectOperationException e) {
      throw new RuntimeException(e);
    }
  }
}