StudyMigrationTest.java

package com.jetbrains.edu.learning;

import com.intellij.testFramework.PlatformTestUtil;
import com.intellij.util.JdomKt;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.jetbrains.annotations.NotNull;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.TestName;

import java.io.IOException;
import java.nio.file.Path;
import java.nio.file.Paths;

import static com.intellij.testFramework.assertions.Assertions.assertThat;

public class StudyMigrationTest {
  @Rule public TestName name = new TestName();

  @Test
  public void testFromThirdToForth() throws JDOMException, IOException, StudySerializationUtils.StudyUnrecognizedFormatException {
    doTest(3);
  }

  @Test
  public void testAdaptive45() throws JDOMException, IOException, StudySerializationUtils.StudyUnrecognizedFormatException {
    doTest(4);
  }

  @Test
  public void testSubtasks45() throws JDOMException, IOException, StudySerializationUtils.StudyUnrecognizedFormatException {
    doTest(4);
  }

  @Test
  public void testTheory35To4() throws JDOMException, IOException, StudySerializationUtils.StudyUnrecognizedFormatException {
    doTest(4);
  }

  @Test
  public void testTheory351To4() throws JDOMException, IOException, StudySerializationUtils.StudyUnrecognizedFormatException {
    doTest(4);
  }

  private void doTest(int version) throws IOException, JDOMException, StudySerializationUtils.StudyUnrecognizedFormatException {
    final String name = PlatformTestUtil.getTestName(this.name.getMethodName(), true);
    final Path before = getTestDataPath().resolve(name + ".xml");
    final Path after = getTestDataPath().resolve(name + ".after.xml");
    Element element = JdomKt.loadElement(before);
    Element converted = element;
    switch (version) {
      case 1:
        converted = StudySerializationUtils.Xml.convertToSecondVersion(element);
        break;
      case 3:
        converted = StudySerializationUtils.Xml.convertToForthVersion(element);
        break;
      case 4:
        converted = StudySerializationUtils.Xml.convertToFifthVersion(element);
        break;
    }
    assertThat(converted).isEqualTo(after);
  }

  @NotNull
  protected Path getTestDataPath() {
    return Paths.get(PlatformTestUtil.getCommunityPath(), "python/educational-core/testData/migration");
  }
}