On branch 001-label-backend-spec
Changes to be committed: new file: src/main/java/com/label/common/shiro/BearerToken.java new file: src/main/java/com/label/common/shiro/ShiroConfig.java new file: src/main/java/com/label/common/shiro/TokenFilter.java new file: src/main/java/com/label/common/shiro/TokenPrincipal.java new file: src/main/java/com/label/common/shiro/UserRealm.java modified: src/main/java/com/label/common/statemachine/DatasetStatus.java new file: src/test/java/com/label/AbstractIntegrationTest.java new file: src/test/java/com/label/unit/StateMachineTest.java new file: src/test/resources/db/init.sql
This commit is contained in:
wh
87
src/test/java/com/label/AbstractIntegrationTest.java
Normal file
87
src/test/java/com/label/AbstractIntegrationTest.java
Normal file
@@ -0,0 +1,87 @@
|
||||
package com.label;
|
||||
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.boot.test.web.server.LocalServerPort;
|
||||
import org.springframework.jdbc.core.JdbcTemplate;
|
||||
import org.springframework.test.context.DynamicPropertyRegistry;
|
||||
import org.springframework.test.context.DynamicPropertySource;
|
||||
import org.testcontainers.containers.GenericContainer;
|
||||
import org.testcontainers.containers.PostgreSQLContainer;
|
||||
import org.testcontainers.junit.jupiter.Container;
|
||||
import org.testcontainers.junit.jupiter.Testcontainers;
|
||||
import org.testcontainers.utility.DockerImageName;
|
||||
import org.testcontainers.utility.MountableFile;
|
||||
|
||||
/**
|
||||
* Base class for all integration tests.
|
||||
*
|
||||
* Starts real PostgreSQL 16 and Redis 7 containers (shared across test class instances).
|
||||
* Executes sql/init.sql to initialize schema and seed data.
|
||||
*
|
||||
* DESIGN:
|
||||
* - @Container with static fields → containers are shared across test methods (faster)
|
||||
* - @DynamicPropertySource → overrides datasource/redis properties at runtime
|
||||
* - @BeforeEach cleanData() → truncates business tables (not sys_company/sys_user) between tests
|
||||
*/
|
||||
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
|
||||
@Testcontainers
|
||||
public abstract class AbstractIntegrationTest {
|
||||
|
||||
@LocalServerPort
|
||||
protected int port;
|
||||
|
||||
@Autowired
|
||||
protected JdbcTemplate jdbcTemplate;
|
||||
|
||||
@SuppressWarnings("resource")
|
||||
@Container
|
||||
protected static final PostgreSQLContainer<?> postgres =
|
||||
new PostgreSQLContainer<>(DockerImageName.parse("postgres:16-alpine"))
|
||||
.withDatabaseName("label_db")
|
||||
.withUsername("label")
|
||||
.withPassword("label_password")
|
||||
.withCopyFileToContainer(
|
||||
MountableFile.forClasspathResource("db/init.sql"),
|
||||
"/docker-entrypoint-initdb.d/init.sql");
|
||||
|
||||
@SuppressWarnings("resource")
|
||||
@Container
|
||||
protected static final GenericContainer<?> redis =
|
||||
new GenericContainer<>(DockerImageName.parse("redis:7-alpine"))
|
||||
.withExposedPorts(6379)
|
||||
.withCommand("redis-server", "--requirepass", "test_redis_password");
|
||||
|
||||
@DynamicPropertySource
|
||||
static void configureProperties(DynamicPropertyRegistry registry) {
|
||||
registry.add("spring.datasource.url", postgres::getJdbcUrl);
|
||||
registry.add("spring.datasource.username", postgres::getUsername);
|
||||
registry.add("spring.datasource.password", postgres::getPassword);
|
||||
registry.add("spring.data.redis.host", redis::getHost);
|
||||
registry.add("spring.data.redis.port", () -> redis.getMappedPort(6379));
|
||||
registry.add("spring.data.redis.password", () -> "test_redis_password");
|
||||
}
|
||||
|
||||
/**
|
||||
* Clean only business data between tests to keep schema intact.
|
||||
* Keep sys_company and sys_user since init.sql seeds them.
|
||||
*/
|
||||
@BeforeEach
|
||||
void cleanData() {
|
||||
jdbcTemplate.execute("TRUNCATE TABLE video_process_job, annotation_task_history, " +
|
||||
"sys_operation_log, sys_config, export_batch, training_dataset, " +
|
||||
"annotation_result, annotation_task, source_data RESTART IDENTITY CASCADE");
|
||||
// Re-insert global sys_config entries that were truncated
|
||||
jdbcTemplate.execute("INSERT INTO sys_config (company_id, config_key, config_value) VALUES " +
|
||||
"(NULL, 'token_ttl_seconds', '7200'), " +
|
||||
"(NULL, 'model_default', 'glm-4'), " +
|
||||
"(NULL, 'video_frame_interval', '30') " +
|
||||
"ON CONFLICT DO NOTHING");
|
||||
}
|
||||
|
||||
/** Helper: get base URL for REST calls */
|
||||
protected String baseUrl(String path) {
|
||||
return "http://localhost:" + port + path;
|
||||
}
|
||||
}
|
||||
265
src/test/java/com/label/unit/StateMachineTest.java
Normal file
265
src/test/java/com/label/unit/StateMachineTest.java
Normal file
@@ -0,0 +1,265 @@
|
||||
package com.label.unit;
|
||||
|
||||
import com.label.common.exception.BusinessException;
|
||||
import com.label.common.statemachine.*;
|
||||
import org.junit.jupiter.api.DisplayName;
|
||||
import org.junit.jupiter.api.Nested;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import static org.assertj.core.api.Assertions.*;
|
||||
|
||||
/**
|
||||
* Unit tests for all state machine enums and StateValidator.
|
||||
* No Spring context needed - pure unit tests.
|
||||
*/
|
||||
@DisplayName("状态机单元测试")
|
||||
class StateMachineTest {
|
||||
|
||||
// ===== SourceStatus =====
|
||||
@Nested
|
||||
@DisplayName("SourceStatus 状态机")
|
||||
class SourceStatusTest {
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：PENDING → EXTRACTING（文本/图片直接提取）")
|
||||
void pendingToExtracting() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.PENDING, SourceStatus.EXTRACTING)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：PENDING → PREPROCESSING（视频上传）")
|
||||
void pendingToPreprocessing() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.PENDING, SourceStatus.PREPROCESSING)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：PREPROCESSING → PENDING（视频预处理完成）")
|
||||
void preprocessingToPending() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.PREPROCESSING, SourceStatus.PENDING)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：EXTRACTING → QA_REVIEW（提取审批通过）")
|
||||
void extractingToQaReview() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.EXTRACTING, SourceStatus.QA_REVIEW)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：QA_REVIEW → APPROVED（QA 审批通过）")
|
||||
void qaReviewToApproved() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.QA_REVIEW, SourceStatus.APPROVED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：APPROVED → PENDING 抛出异常")
|
||||
void approvedToPendingFails() {
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.APPROVED, SourceStatus.PENDING)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：PENDING → APPROVED（跳过中间状态）抛出异常")
|
||||
void pendingToApprovedFails() {
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(SourceStatus.TRANSITIONS, SourceStatus.PENDING, SourceStatus.APPROVED)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
}
|
||||
|
||||
// ===== TaskStatus =====
|
||||
@Nested
|
||||
@DisplayName("TaskStatus 状态机")
|
||||
class TaskStatusTest {
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：UNCLAIMED → IN_PROGRESS（领取）")
|
||||
void unclaimedToInProgress() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.UNCLAIMED, TaskStatus.IN_PROGRESS)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：IN_PROGRESS → SUBMITTED（提交）")
|
||||
void inProgressToSubmitted() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.IN_PROGRESS, TaskStatus.SUBMITTED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：IN_PROGRESS → UNCLAIMED（放弃）")
|
||||
void inProgressToUnclaimed() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.IN_PROGRESS, TaskStatus.UNCLAIMED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：IN_PROGRESS → IN_PROGRESS（ADMIN 强制转移，持有人变更）")
|
||||
void inProgressToInProgress() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.IN_PROGRESS, TaskStatus.IN_PROGRESS)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：SUBMITTED → APPROVED（审批通过）")
|
||||
void submittedToApproved() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.SUBMITTED, TaskStatus.APPROVED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：SUBMITTED → REJECTED（审批驳回）")
|
||||
void submittedToRejected() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.SUBMITTED, TaskStatus.REJECTED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：REJECTED → IN_PROGRESS（标注员重领）")
|
||||
void rejectedToInProgress() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.REJECTED, TaskStatus.IN_PROGRESS)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：APPROVED → IN_PROGRESS 抛出异常")
|
||||
void approvedToInProgressFails() {
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.APPROVED, TaskStatus.IN_PROGRESS)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：UNCLAIMED → SUBMITTED（跳过 IN_PROGRESS）抛出异常")
|
||||
void unclaimedToSubmittedFails() {
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(TaskStatus.TRANSITIONS, TaskStatus.UNCLAIMED, TaskStatus.SUBMITTED)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
}
|
||||
|
||||
// ===== DatasetStatus =====
|
||||
@Nested
|
||||
@DisplayName("DatasetStatus 状态机")
|
||||
class DatasetStatusTest {
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：PENDING_REVIEW → APPROVED")
|
||||
void pendingReviewToApproved() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(DatasetStatus.TRANSITIONS, DatasetStatus.PENDING_REVIEW, DatasetStatus.APPROVED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：PENDING_REVIEW → REJECTED")
|
||||
void pendingReviewToRejected() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(DatasetStatus.TRANSITIONS, DatasetStatus.PENDING_REVIEW, DatasetStatus.REJECTED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：REJECTED → PENDING_REVIEW（重新提交）")
|
||||
void rejectedToPendingReview() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(DatasetStatus.TRANSITIONS, DatasetStatus.REJECTED, DatasetStatus.PENDING_REVIEW)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：APPROVED → REJECTED 抛出异常")
|
||||
void approvedToRejectedFails() {
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(DatasetStatus.TRANSITIONS, DatasetStatus.APPROVED, DatasetStatus.REJECTED)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
}
|
||||
|
||||
// ===== VideoJobStatus =====
|
||||
@Nested
|
||||
@DisplayName("VideoJobStatus 状态机")
|
||||
class VideoJobStatusTest {
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：PENDING → RUNNING")
|
||||
void pendingToRunning() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.PENDING, VideoJobStatus.RUNNING)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：RUNNING → SUCCESS")
|
||||
void runningToSuccess() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.RUNNING, VideoJobStatus.SUCCESS)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：RUNNING → RETRYING（失败且未超重试次数）")
|
||||
void runningToRetrying() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.RUNNING, VideoJobStatus.RETRYING)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：RUNNING → FAILED（失败且超过最大重试）")
|
||||
void runningToFailed() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.RUNNING, VideoJobStatus.FAILED)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("合法转换：RETRYING → RUNNING（AI 重试）")
|
||||
void retryingToRunning() {
|
||||
assertThatCode(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.RETRYING, VideoJobStatus.RUNNING)
|
||||
).doesNotThrowAnyException();
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：FAILED → PENDING 不在状态机内（ADMIN 手动触发，不走 StateValidator）")
|
||||
void failedToPendingNotInStateMachine() {
|
||||
// FAILED → PENDING is intentionally NOT in TRANSITIONS (ADMIN manual reset via special API)
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.FAILED, VideoJobStatus.PENDING)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
|
||||
@Test
|
||||
@DisplayName("非法转换：SUCCESS → RUNNING 抛出异常")
|
||||
void successToRunningFails() {
|
||||
assertThatThrownBy(() ->
|
||||
StateValidator.assertTransition(VideoJobStatus.TRANSITIONS, VideoJobStatus.SUCCESS, VideoJobStatus.RUNNING)
|
||||
).isInstanceOf(BusinessException.class)
|
||||
.extracting("code").isEqualTo("INVALID_STATE_TRANSITION");
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user