src/test/java/com/example/mydemo1/ChunkServiceTest.java - master - marie.sidorkina/eye_tracking_java

src/test/java/com/example/mydemo1/ChunkServiceTest.java
81 строка3 KB
Marie
add tests
14 янв 2026, 11:17
14 янв 2026, 11:17c2fa01e
package com.example.mydemo1;

import com.example.mydemo1.dto.ChunkCreateRequest;
import com.example.mydemo1.dto.GazeCreateRequest;
import com.example.mydemo1.model.*;
import com.example.mydemo1.repository.SessionRepository;
import com.example.mydemo1.repository.UserRepository;
import com.example.mydemo1.service.ChunkService;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;

import java.time.Instant;
import java.util.List;

import static org.junit.jupiter.api.Assertions.*;

@SpringBootTest
class ChunkServiceTest {

    @Autowired
    private ChunkService chunkService;

    @Autowired
    private SessionRepository sessionRepository;

    @Autowired
    private UserRepository userRepository;

    @Test
    void createChunk_savesChunkAndGazes() {
        User user = new User();
        user.setUsername("alice");
        user.setEmail("alice@example.com");
        user.setPassword("secret");
        user = userRepository.save(user);

        Session session = new Session();
        session.setName("test session");
        session.setStartedAt(Instant.now());
        session.setUser(user);
        session = sessionRepository.save(session);

        ChunkCreateRequest req = new ChunkCreateRequest();
        req.setSessionId(session.getId());
        req.setTimestamp(Instant.now());
        req.setElementFrom("p1");
        req.setElementTo("p1");

        GazeCreateRequest g1 = new GazeCreateRequest();
        g1.setTimestamp(Instant.now());
        g1.setX(100.0);
        g1.setY(200.0);

        GazeCreateRequest g2 = new GazeCreateRequest();
        g2.setTimestamp(Instant.now().plusMillis(50));
        g2.setX(110.0);
        g2.setY(201.0);

        req.setGazes(List.of(g1, g2));

        Chunk saved = chunkService.createChunk(session, req);

        assertNotNull(saved.getId(), "Chunk saved must have id");
        assertNotNull(saved.getSession(), "Chunk saved must have session not null");
        assertEquals(session.getId(), saved.getSession().getId(), "Session id should be the same for session and chunk");

        assertNotNull(saved.getGazes(), "Chunk saved must contain gazes");
        assertEquals(2, saved.getGazes().size(), "Chunk must contain 2 gazes");

        Session finalSession = session;
        saved.getGazes().forEach(g -> {
            assertNotNull(g.getId(), "Gaze id must not be null");
            assertNotNull(g.getTimestamp(),  "Gaze timestamp must not be null");
            assertNotNull(g.getX(), "Gaze x must not be null");
            assertNotNull(g.getY(), "Gaze y must not be null");
            assertNotNull(g.getSession(), "Gaze session must not be null");
            assertEquals(finalSession.getId(), g.getSession().getId(), "Session id should be the same for session and gaze");
        });
    }
}