Test file for Issue with subdirectories

UmpleModelTest.java

/*

 Copyright: All contributers to the Umple Project

 This file is made available subject to the open source license found at:
 http://umple.org/license

*/

package cruise.umple.compiler;


import static org.junit.Assert.*;

import java.io.*;
import java.util.*;
import java.util.stream.Collectors;

import org.junit.*;

import cruise.umple.util.SampleFileWriter;



public class UmpleModelTest
{

  UmpleFile uFile;
  UmpleModel model;

  @Before
  public void setUp()
  {
    SampleFileWriter.createFile("teacher.txt", "class Teacher {}");
    SampleFileWriter.createFile("student.txt", "class Student {}");

    SampleFileWriter.createFile("sub/Teacher2.ump", "class Teacher2{}");
    SampleFileWriter.createFile("sub/student2.ump", "class Student2 {}");
    SampleFileWriter.createFile("TestSymmetric.txt", "class Course { * self isMutuallyExclusiveWith;}");

    uFile = new UmpleFile("teacher.txt");
    model = new UmpleModel(new UmpleFile("teacher.txt"));
  }

  @After
  public void tearDown()
  {
    SampleFileWriter.destroy("teacher.txt");
    SampleFileWriter.destroy("student.txt");
    SampleFileWriter.destroy("TestSymmetric.txt");
    SampleFileWriter.destroy("Course.java");
    SampleFileWriter.destroy("sub/");
  }
  
   @Test
  public void umpleFileWithLinkedFiles() {
    uFile.addLinkedFiles("sub2/student2.ump");
    model = new UmpleModel(uFile);
    model.run();
    
    List<String> classNames = model.getUmpleClasses().stream().map(UmpleClass::getName).collect(Collectors.toList());
    assertTrue(classNames.contains("Student2"));
    assertTrue(classNames.contains("Teacher"));
  }
  
  @Test
  public void umpleFileWithLinkedFilesSubDir() {
    uFile = new UmpleFile("sub/Teacher2.ump");
    uFile.addLinkedFiles("sub/student2.ump");
    model = new UmpleModel(uFile);
    model.run();
    
    List<String> classNames = model.getUmpleClasses().stream().map(UmpleClass::getName).collect(Collectors.toList());
    assertTrue(classNames.contains("Student2"));
    assertTrue(classNames.contains("Teacher2"));
  }

}