diff --git a/buisnesslogic/src/main/java/com/plannaplan/services/CourseService.java b/buisnesslogic/src/main/java/com/plannaplan/services/CourseService.java index e5933cc..446faea 100644 --- a/buisnesslogic/src/main/java/com/plannaplan/services/CourseService.java +++ b/buisnesslogic/src/main/java/com/plannaplan/services/CourseService.java @@ -24,4 +24,8 @@ public class CourseService { public void save(Course course) { this.repo.save(course); } + + public int getCoursesAmmount(){ + return (int)this.repo.count(); + } } \ No newline at end of file diff --git a/buisnesslogic/src/main/java/com/plannaplan/services/GroupService.java b/buisnesslogic/src/main/java/com/plannaplan/services/GroupService.java index dee884b..e899dd8 100644 --- a/buisnesslogic/src/main/java/com/plannaplan/services/GroupService.java +++ b/buisnesslogic/src/main/java/com/plannaplan/services/GroupService.java @@ -27,4 +27,8 @@ public class GroupService { public void save(Groups group) { this.repo.save(group); } + + public int getGroupsAmmount(){ + return (int)this.repo.count(); + } } \ No newline at end of file diff --git a/buisnesslogic/src/main/java/com/plannaplan/services/LecturerService.java b/buisnesslogic/src/main/java/com/plannaplan/services/LecturerService.java index 2eb1415..650fd42 100644 --- a/buisnesslogic/src/main/java/com/plannaplan/services/LecturerService.java +++ b/buisnesslogic/src/main/java/com/plannaplan/services/LecturerService.java @@ -18,4 +18,8 @@ public class LecturerService { public void save(Lecturer lecturer) { repo.save(lecturer); } + + public int getLecturersAmmount(){ + return (int)this.repo.count(); + } } \ No newline at end of file diff --git a/buisnesslogic/src/test/java/com/plannaplan/ConfiguratorTest.java b/buisnesslogic/src/test/java/com/plannaplan/ConfiguratorTest.java index c12a293..102c5a1 100644 --- a/buisnesslogic/src/test/java/com/plannaplan/ConfiguratorTest.java +++ b/buisnesslogic/src/test/java/com/plannaplan/ConfiguratorTest.java @@ -5,14 +5,14 @@ import org.springframework.boot.test.context.SpringBootTest; import org.springframework.test.context.ContextConfiguration; import org.springframework.test.context.junit4.SpringRunner; -import static org.assertj.core.api.Assertions.assertThat; import static org.junit.Assert.assertTrue; import java.io.InputStream; -import com.plannaplan.App; -import com.plannaplan.Configurator; import com.plannaplan.models.ConfigData; +import com.plannaplan.services.CourseService; +import com.plannaplan.services.GroupService; +import com.plannaplan.services.LecturerService; import org.junit.Test; import org.junit.runner.RunWith; @@ -22,15 +22,29 @@ import org.junit.runner.RunWith; @ContextConfiguration public class ConfiguratorTest { + private static String FILE_NAME = "Zajecia.xlsx"; + @Autowired private Configurator restTemplate; + @Autowired + private CourseService courseService; + + @Autowired + private GroupService groupService; + + @Autowired + private LecturerService lecturerService; + @Test - public void greetingShouldReturnDefaultMessage() throws Exception { - final InputStream inputStream = getClass().getClassLoader().getResourceAsStream("Zajecia.xlsx"); + public void shouldImportDataFromFileToDatabase() throws Exception { + final InputStream inputStream = getClass().getClassLoader().getResourceAsStream(FILE_NAME); final ConfigData data = new ConfigData(null, null, inputStream); this.restTemplate.config(data); - assertTrue(true); - + int courses_ammount = this.courseService.getCoursesAmmount(); + int groups_ammount = this.groupService.getGroupsAmmount(); + int lecturers_ammount = this.lecturerService.getLecturersAmmount(); + + assertTrue(courses_ammount > 0 && groups_ammount > 0 && lecturers_ammount > 0); } } \ No newline at end of file