86 lines
3.3 KiB
Kotlin
86 lines
3.3 KiB
Kotlin
package agzam.vyatsu
|
||
|
||
import agzam.vyatsu.data.remote.TeacherScheduleParser
|
||
import agzam.vyatsu.domain.model.LessonType
|
||
import java.time.LocalDate
|
||
import java.time.LocalTime
|
||
import org.junit.Assert.assertEquals
|
||
import org.junit.Assert.assertNull
|
||
import org.junit.Assert.assertTrue
|
||
import org.junit.Test
|
||
|
||
class TeacherScheduleParserTest {
|
||
|
||
private fun fixture(): String =
|
||
checkNotNull(javaClass.getResourceAsStream("/prepod.html"))
|
||
.readBytes().toString(Charsets.UTF_8)
|
||
|
||
@Test
|
||
fun teachersHeaderParsedInOrderAndDeduped() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
assertEquals(listOf("Лелекова Е.В.", "Иванов И.И.", "Петрова А.С."), parsed.teachers)
|
||
}
|
||
|
||
@Test
|
||
fun daysWithCorrectDates() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
assertEquals(2, parsed.days.size)
|
||
assertEquals(LocalDate.of(2026, 9, 14), parsed.days[0].date)
|
||
assertEquals(LocalDate.of(2026, 9, 15), parsed.days[1].date)
|
||
}
|
||
|
||
@Test
|
||
fun intervalTimesParsed() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
val day0 = parsed.days[0]
|
||
assertEquals(2, day0.intervals.size)
|
||
assertEquals(LocalTime.of(8, 20), day0.intervals[0].start)
|
||
assertEquals(LocalTime.of(9, 50), day0.intervals[0].end)
|
||
assertEquals(LocalTime.of(10, 0), day0.intervals[1].start)
|
||
assertEquals(LocalTime.of(11, 30), day0.intervals[1].end)
|
||
}
|
||
|
||
@Test
|
||
fun multiGroupCellSplitsIntoPerGroupLessons() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
val cell = parsed.days[0].intervals[0].lessons.filter { it.teacher == "Лелекова Е.В." }
|
||
assertEquals(2, cell.size)
|
||
val lesson0 = cell[0]
|
||
val lesson1 = cell[1]
|
||
assertEquals("Бб-1101-04-00", lesson0.group)
|
||
assertNull(lesson0.subgroup)
|
||
assertEquals("01 подгруппа", lesson1.subgroup)
|
||
assertEquals("4-302", lesson0.room)
|
||
assertEquals("Биоразнообразие", lesson0.subject)
|
||
assertEquals(LessonType.LECTURE, lesson0.type)
|
||
assertEquals("Лелекова Е.В.", lesson0.teacher)
|
||
}
|
||
|
||
@Test
|
||
fun emptyCellContributesNoLesson() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
val threeSlots = (0..6).map { slot ->
|
||
parsed.days[0].intervals.firstOrNull { it.start.hour == 8 && it.start.minute == 20 } ?: return@map true
|
||
}
|
||
val petrovLessons = parsed.days[0].intervals[0].lessons.filter { it.teacher == "Петрова А.С." }
|
||
assertEquals(0, petrovLessons.size)
|
||
assertTrue(threeSlots.isEmpty() || threeSlots.size == 7)
|
||
}
|
||
|
||
@Test
|
||
fun noRoomCellKeepsRawAndNullRoom() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
val cell = parsed.days[0].intervals[1].lessons.filter { it.teacher == "Лелекова Е.В." }
|
||
assertEquals(2, cell.size)
|
||
assertNull(cell[0].room)
|
||
assertTrue(cell[0].raw.contains("Бб-1101-04-00"))
|
||
assertNull(cell[0].subject)
|
||
}
|
||
|
||
@Test
|
||
fun exactTotalLessonCount() {
|
||
val parsed = TeacherScheduleParser.parse(fixture())
|
||
val total = parsed.days.sumOf { day -> day.intervals.sumOf { it.lessons.size } }
|
||
assertEquals(11, total)
|
||
}
|
||
} |