TDD 学习整理#

1
package com.example.demo.service;
2

3
import com.example.demo.dto.TrainingRequest;
4
import com.example.demo.dto.TrainingResult;
5
import com.example.demo.entity.TrainingTask;
6
import com.example.demo.mapper.TrainingTaskMapper;
7
import org.junit.jupiter.api.Test;
8
import org.mockito.ArgumentCaptor;
9

10
import java.util.concurrent.CompletableFuture;
11

12
import static org.junit.jupiter.api.Assertions.*;
13
import static org.mockito.Mockito.*;
14

15
public class TrainingServiceTddTest {
16

17
    @Test
18
    void startTraining_shouldFillDefaults_persistTask_andTriggerExecutor() {
19
        TrainingTaskMapper taskMapper = mock(TrainingTaskMapper.class);
20
        TrainingExecutor executor = mock(TrainingExecutor.class);
21

22
        when(executor.executeTraining(anyString(), anyInt()))
23
                .thenReturn(CompletableFuture.completedFuture(0.0));
24

25
        TrainingService service = new TrainingService();
26
        service.taskMapper = taskMapper;
27
        service.trainingExecutor = executor;
28

29
        TrainingRequest req = new TrainingRequest();
30
        req.setAlgorithm("PPO");
31
        req.setEpisodes(null);
32
        req.setLearningRate(null);
33

34
        Long userId = 1001L;
35

36
        TrainingResult result = service.startTraining(req, userId);
37

38
        assertNotNull(result.getTaskId());
39
        assertEquals("PENDING", result.getStatus());
40

41
        ArgumentCaptor<TrainingTask> captor = ArgumentCaptor.forClass(TrainingTask.class);
42
        verify(taskMapper, times(1)).insert(captor.capture());
43

44
        TrainingTask inserted = captor.getValue();
45
        assertEquals("PPO", inserted.getAlgorithm());
46
        assertEquals(1000, inserted.getEpisodes());
47
        assertEquals(0.001, inserted.getLearningRate(), 1e-9);
48
        assertEquals(userId, inserted.getUserId());
49

50
        verify(executor, times(1)).executeTraining(result.getTaskId(), 1000);
51
    }
52
}

1
public TrainingResult startTraining(TrainingRequest request, Long userId) {
2
    String taskId = UUID.randomUUID().toString().substring(0, 8);
3

4
    int episodes = request.getEpisodes() != null ? request.getEpisodes() : 1000;
5
    double learningRate = request.getLearningRate() != null ? request.getLearningRate() : 0.001;
6

7
    TrainingTask task = new TrainingTask(taskId, request.getAlgorithm(), episodes, learningRate);
8
    task.setUserId(userId);
9

10
    taskMapper.insert(task);
11
    trainingExecutor.executeTraining(taskId, episodes);
12

13
    return new TrainingResult(taskId, "PENDING", 0.0, "任务已提交，请稍后查询状态");
14
}

1
private int resolveEpisodes(TrainingRequest request) {
2
    return request.getEpisodes() != null ? request.getEpisodes() : 1000;
3
}
4

5
private double resolveLearningRate(TrainingRequest request) {
6
    return request.getLearningRate() != null ? request.getLearningRate() : 0.001;
7
}