test: missing entity classes and non-jsonapi entities

vinceh121 · Nov 1, 2023 · 7bf1e6c · 7bf1e6c
1 parent 8084aef
commit 7bf1e6c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/wrapper/src/main/java/me/vinceh121/jskolengo/SkolengoConstants.java b/wrapper/src/main/java/me/vinceh121/jskolengo/SkolengoConstants.java
@@ -5,6 +5,8 @@
 import me.vinceh121.jskolengo.entities.SkoAppConfig;
 import me.vinceh121.jskolengo.entities.StudentUserInfo;
 import me.vinceh121.jskolengo.entities.agenda.Agenda;
+import me.vinceh121.jskolengo.entities.agenda.Audio;
+import me.vinceh121.jskolengo.entities.agenda.CorrectedWork;
 import me.vinceh121.jskolengo.entities.agenda.Homework;
 import me.vinceh121.jskolengo.entities.agenda.Lesson;
 import me.vinceh121.jskolengo.entities.agenda.LessonContent;
@@ -38,6 +40,8 @@ public final class SkolengoConstants {
 		SkoAppConfig.class,
 		StudentUserInfo.class,
 		Agenda.class,
+		Audio.class,
+		CorrectedWork.class,
 		Homework.class,
 		Lesson.class,
 		LessonContent.class,

diff --git a/wrapper/src/test/java/me/vinceh121/jskolengo/test/TestMapper.java b/wrapper/src/test/java/me/vinceh121/jskolengo/test/TestMapper.java
@@ -3,12 +3,15 @@
 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.List;
+import java.util.stream.Collectors;
 
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 import org.reflections.Reflections;
 import org.reflections.scanners.Scanners;
 
+import com.github.jasminb.jsonapi.annotations.Type;
+
 import me.vinceh121.jskolengo.SkolengoConstants;
 import me.vinceh121.jskolengo.entities.AbstractSkolengoEntity;
 
@@ -19,8 +22,12 @@ class TestMapper {
 	void testClassList() {
 		Reflections ref = new Reflections("me.vinceh121.jskolengo.entities", Scanners.SubTypes);
 		List<Class<? extends AbstractSkolengoEntity>> expected
-				= new ArrayList<>(ref.getSubTypesOf(AbstractSkolengoEntity.class));
+				= new ArrayList<>(ref.getSubTypesOf(AbstractSkolengoEntity.class)).stream()
+						.filter(c -> c.isAnnotationPresent(Type.class))
+						.collect(Collectors.toList());
+
 		List<Class<? extends AbstractSkolengoEntity>> actual = new ArrayList<>();
+
 		for (Class<?> o : SkolengoConstants.ENTITY_CLASSES) {
 			actual.add((Class<? extends AbstractSkolengoEntity>) o);
 		}