data-permission.integration.test.ts 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922
  1. import { describe, it, expect, beforeEach, vi, afterEach } from 'vitest';
  2. import { testClient } from 'hono/testing';
  3. import { IntegrationTestDatabase, setupIntegrationDatabaseHooksWithEntities } from '@d8d/shared-test-util';
  4. import { JWTUtil } from '@d8d/shared-utils';
  5. import { z } from '@hono/zod-openapi';
  6. import { createCrudRoutes } from '../../src/routes/generic-crud.routes';
  7. import { Entity, PrimaryGeneratedColumn, Column } from 'typeorm';
  8. // 测试用户实体
  9. @Entity()
  10. class TestUser {
  11. @PrimaryGeneratedColumn()
  12. id!: number;
  13. @Column('varchar')
  14. username!: string;
  15. @Column('varchar')
  16. password!: string;
  17. @Column('varchar')
  18. nickname!: string;
  19. @Column('varchar')
  20. registrationSource!: string;
  21. }
  22. // 测试实体类
  23. @Entity()
  24. class TestEntity {
  25. @PrimaryGeneratedColumn()
  26. id!: number;
  27. @Column('varchar')
  28. name!: string;
  29. @Column('int')
  30. userId!: number;
  31. @Column('int', { nullable: true })
  32. createdBy?: number;
  33. @Column('int', { nullable: true })
  34. updatedBy?: number;
  35. }
  36. // 测试带状态字段的实体类
  37. @Entity()
  38. class TestEntityWithStatus {
  39. @PrimaryGeneratedColumn()
  40. id!: number;
  41. @Column('varchar')
  42. name!: string;
  43. @Column('int')
  44. status!: number; // 1=可用,0=不可用
  45. @Column('int')
  46. userId!: number;
  47. }
  48. // 定义测试实体的Schema
  49. const createTestSchema = z.object({
  50. name: z.string().min(1, '名称不能为空'),
  51. userId: z.number().optional()
  52. });
  53. const updateTestSchema = z.object({
  54. name: z.string().min(1, '名称不能为空').optional()
  55. });
  56. const getTestSchema = z.object({
  57. id: z.number(),
  58. name: z.string(),
  59. userId: z.number(),
  60. createdBy: z.number().nullable().optional(),
  61. updatedBy: z.number().nullable().optional()
  62. });
  63. const listTestSchema = z.object({
  64. id: z.number(),
  65. name: z.string(),
  66. userId: z.number(),
  67. createdBy: z.number().nullable().optional(),
  68. updatedBy: z.number().nullable().optional()
  69. });
  70. // 设置集成测试钩子
  71. setupIntegrationDatabaseHooksWithEntities([TestUser, TestEntity, TestEntityWithStatus])
  72. describe('共享CRUD数据权限控制集成测试', () => {
  73. let client: any;
  74. let testToken1: string;
  75. let testToken2: string;
  76. let testUser1: TestUser;
  77. let testUser2: TestUser;
  78. let mockAuthMiddleware: any;
  79. beforeEach(async () => {
  80. // 获取数据源
  81. const dataSource = await IntegrationTestDatabase.getDataSource();
  82. // 创建测试用户1
  83. const userRepository = dataSource.getRepository(TestUser);
  84. testUser1 = userRepository.create({
  85. username: `test_user_1_${Date.now()}`,
  86. password: 'test_password',
  87. nickname: '测试用户1',
  88. registrationSource: 'web'
  89. });
  90. await userRepository.save(testUser1);
  91. // 创建测试用户2
  92. testUser2 = userRepository.create({
  93. username: `test_user_2_${Date.now()}`,
  94. password: 'test_password',
  95. nickname: '测试用户2',
  96. registrationSource: 'web'
  97. });
  98. await userRepository.save(testUser2);
  99. // 生成测试用户的token
  100. testToken1 = JWTUtil.generateToken({
  101. id: testUser1.id,
  102. username: testUser1.username,
  103. roles: [{name:'user'}]
  104. });
  105. testToken2 = JWTUtil.generateToken({
  106. id: testUser2.id,
  107. username: testUser2.username,
  108. roles: [{name:'user'}]
  109. });
  110. // 创建模拟认证中间件
  111. mockAuthMiddleware = async (c: any, next: any) => {
  112. const authHeader = c.req.header('Authorization');
  113. if (authHeader && authHeader.startsWith('Bearer ')) {
  114. const token = authHeader.substring(7);
  115. try {
  116. // 简单模拟用户解析
  117. if (token === testToken1) {
  118. c.set('user', { id: testUser1.id, username: testUser1.username });
  119. } else if (token === testToken2) {
  120. c.set('user', { id: testUser2.id, username: testUser2.username });
  121. }
  122. } catch (error) {
  123. // token解析失败
  124. }
  125. } else {
  126. // 没有认证信息,返回401
  127. return c.json({ code: 401, message: '认证失败' }, 401);
  128. }
  129. await next();
  130. };
  131. // 创建测试路由 - 启用数据权限控制
  132. const testRoutes = createCrudRoutes({
  133. entity: TestEntity,
  134. createSchema: createTestSchema,
  135. updateSchema: updateTestSchema,
  136. getSchema: getTestSchema,
  137. listSchema: listTestSchema,
  138. middleware: [mockAuthMiddleware],
  139. dataPermission: {
  140. enabled: true,
  141. userIdField: 'userId'
  142. }
  143. });
  144. client = testClient(testRoutes);
  145. });
  146. describe('GET / - 列表查询权限过滤', () => {
  147. it('应该只返回当前用户的数据', async () => {
  148. // 创建测试数据
  149. const dataSource = await IntegrationTestDatabase.getDataSource();
  150. const testRepository = dataSource.getRepository(TestEntity);
  151. // 为用户1创建数据
  152. const user1Data1 = testRepository.create({
  153. name: '用户1的数据1',
  154. userId: testUser1.id
  155. });
  156. await testRepository.save(user1Data1);
  157. const user1Data2 = testRepository.create({
  158. name: '用户1的数据2',
  159. userId: testUser1.id
  160. });
  161. await testRepository.save(user1Data2);
  162. // 为用户2创建数据
  163. const user2Data = testRepository.create({
  164. name: '用户2的数据',
  165. userId: testUser2.id
  166. });
  167. await testRepository.save(user2Data);
  168. // 用户1查询列表
  169. const response = await client.index.$get({
  170. query: {
  171. page: 1,
  172. pageSize: 10
  173. }
  174. }, {
  175. headers: {
  176. 'Authorization': `Bearer ${testToken1}`
  177. }
  178. });
  179. console.debug('列表查询响应状态:', response.status);
  180. if (response.status !== 200) {
  181. const errorData = await response.json();
  182. console.debug('列表查询错误信息:', errorData);
  183. }
  184. expect(response.status).toBe(200);
  185. if (response.status === 200) {
  186. const data = await response.json();
  187. expect(data).toHaveProperty('data');
  188. expect(Array.isArray(data.data)).toBe(true);
  189. expect(data.data).toHaveLength(2); // 应该只返回用户1的2条数据
  190. // 验证所有返回的数据都属于用户1
  191. data.data.forEach((item: any) => {
  192. expect(item.userId).toBe(testUser1.id);
  193. });
  194. }
  195. });
  196. it('应该拒绝未认证用户的访问', async () => {
  197. const response = await client.index.$get({
  198. query: {
  199. page: 1,
  200. pageSize: 10
  201. }
  202. });
  203. expect(response.status).toBe(401);
  204. });
  205. });
  206. describe('POST / - 创建操作权限验证', () => {
  207. it('应该成功创建属于当前用户的数据', async () => {
  208. const createData = {
  209. name: '测试创建数据',
  210. userId: testUser1.id // 用户ID与当前用户匹配
  211. };
  212. const response = await client.index.$post({
  213. json: createData
  214. }, {
  215. headers: {
  216. 'Authorization': `Bearer ${testToken1}`
  217. }
  218. });
  219. console.debug('创建数据响应状态:', response.status);
  220. expect(response.status).toBe(201);
  221. if (response.status === 201) {
  222. const data = await response.json();
  223. expect(data).toHaveProperty('id');
  224. expect(data.name).toBe(createData.name);
  225. expect(data.userId).toBe(testUser1.id);
  226. }
  227. });
  228. it('应该拒绝创建不属于当前用户的数据', async () => {
  229. const createData = {
  230. name: '测试创建数据',
  231. userId: testUser2.id // 用户ID与当前用户不匹配
  232. };
  233. const response = await client.index.$post({
  234. json: createData
  235. }, {
  236. headers: {
  237. 'Authorization': `Bearer ${testToken1}`
  238. }
  239. });
  240. console.debug('创建无权数据响应状态:', response.status);
  241. expect(response.status).toBe(403); // 权限验证失败返回403 Forbidden
  242. if (response.status === 403) {
  243. const data = await response.json();
  244. expect(data.message).toContain('无权');
  245. }
  246. });
  247. });
  248. describe('GET /:id - 获取详情权限验证', () => {
  249. it('应该成功获取属于当前用户的数据详情', async () => {
  250. // 先创建测试数据
  251. const dataSource = await IntegrationTestDatabase.getDataSource();
  252. const testRepository = dataSource.getRepository(TestEntity);
  253. const testData = testRepository.create({
  254. name: '测试数据详情',
  255. userId: testUser1.id
  256. });
  257. await testRepository.save(testData);
  258. const response = await client[':id'].$get({
  259. param: { id: testData.id }
  260. }, {
  261. headers: {
  262. 'Authorization': `Bearer ${testToken1}`
  263. }
  264. });
  265. console.debug('获取详情响应状态:', response.status);
  266. if (response.status !== 200) {
  267. const errorData = await response.json();
  268. console.debug('获取详情错误信息:', errorData);
  269. }
  270. expect(response.status).toBe(200);
  271. if (response.status === 200) {
  272. const data = await response.json();
  273. expect(data.id).toBe(testData.id);
  274. expect(data.name).toBe(testData.name);
  275. expect(data.userId).toBe(testUser1.id);
  276. }
  277. });
  278. it('应该拒绝获取不属于当前用户的数据详情', async () => {
  279. // 先创建属于用户2的数据
  280. const dataSource = await IntegrationTestDatabase.getDataSource();
  281. const testRepository = dataSource.getRepository(TestEntity);
  282. const testData = testRepository.create({
  283. name: '用户2的数据',
  284. userId: testUser2.id
  285. });
  286. await testRepository.save(testData);
  287. // 用户1尝试获取用户2的数据
  288. const response = await client[':id'].$get({
  289. param: { id: testData.id }
  290. }, {
  291. headers: {
  292. 'Authorization': `Bearer ${testToken1}`
  293. }
  294. });
  295. console.debug('获取无权详情响应状态:', response.status);
  296. expect(response.status).toBe(404); // GET操作中,权限错误返回404而不是403
  297. });
  298. it('应该处理不存在的资源', async () => {
  299. const response = await client[':id'].$get({
  300. param: { id: 999999 }
  301. }, {
  302. headers: {
  303. 'Authorization': `Bearer ${testToken1}`
  304. }
  305. });
  306. expect(response.status).toBe(404);
  307. });
  308. });
  309. describe('PUT /:id - 更新操作权限验证', () => {
  310. it('应该成功更新属于当前用户的数据', async () => {
  311. // 先创建测试数据
  312. const dataSource = await IntegrationTestDatabase.getDataSource();
  313. const testRepository = dataSource.getRepository(TestEntity);
  314. const testData = testRepository.create({
  315. name: '原始数据',
  316. userId: testUser1.id
  317. });
  318. await testRepository.save(testData);
  319. const updateData = {
  320. name: '更新后的数据'
  321. };
  322. const response = await client[':id'].$put({
  323. param: { id: testData.id },
  324. json: updateData
  325. }, {
  326. headers: {
  327. 'Authorization': `Bearer ${testToken1}`
  328. }
  329. });
  330. console.debug('更新数据响应状态:', response.status);
  331. expect(response.status).toBe(200);
  332. if (response.status === 200) {
  333. const data = await response.json();
  334. expect(data.name).toBe(updateData.name);
  335. expect(data.userId).toBe(testUser1.id);
  336. }
  337. });
  338. it('应该拒绝更新不属于当前用户的数据', async () => {
  339. // 先创建属于用户2的数据
  340. const dataSource = await IntegrationTestDatabase.getDataSource();
  341. const testRepository = dataSource.getRepository(TestEntity);
  342. const testData = testRepository.create({
  343. name: '用户2的数据',
  344. userId: testUser2.id
  345. });
  346. await testRepository.save(testData);
  347. const updateData = {
  348. name: '尝试更新的数据'
  349. };
  350. // 用户1尝试更新用户2的数据
  351. const response = await client[':id'].$put({
  352. param: { id: testData.id },
  353. json: updateData
  354. }, {
  355. headers: {
  356. 'Authorization': `Bearer ${testToken1}`
  357. }
  358. });
  359. console.debug('更新无权数据响应状态:', response.status);
  360. expect(response.status).toBe(403); // 权限验证失败返回403 Forbidden
  361. if (response.status === 403) {
  362. const data = await response.json();
  363. expect(data.message).toContain('无权');
  364. }
  365. });
  366. });
  367. describe('DELETE /:id - 删除操作权限验证', () => {
  368. it('应该成功删除属于当前用户的数据', async () => {
  369. // 先创建测试数据
  370. const dataSource = await IntegrationTestDatabase.getDataSource();
  371. const testRepository = dataSource.getRepository(TestEntity);
  372. const testData = testRepository.create({
  373. name: '待删除数据',
  374. userId: testUser1.id
  375. });
  376. await testRepository.save(testData);
  377. const response = await client[':id'].$delete({
  378. param: { id: testData.id }
  379. }, {
  380. headers: {
  381. 'Authorization': `Bearer ${testToken1}`
  382. }
  383. });
  384. console.debug('删除数据响应状态:', response.status);
  385. expect(response.status).toBe(204);
  386. // 验证数据确实被删除
  387. const deletedData = await testRepository.findOne({
  388. where: { id: testData.id }
  389. });
  390. expect(deletedData).toBeNull();
  391. });
  392. it('应该拒绝删除不属于当前用户的数据', async () => {
  393. // 先创建属于用户2的数据
  394. const dataSource = await IntegrationTestDatabase.getDataSource();
  395. const testRepository = dataSource.getRepository(TestEntity);
  396. const testData = testRepository.create({
  397. name: '用户2的数据',
  398. userId: testUser2.id
  399. });
  400. await testRepository.save(testData);
  401. // 用户1尝试删除用户2的数据
  402. const response = await client[':id'].$delete({
  403. param: { id: testData.id }
  404. }, {
  405. headers: {
  406. 'Authorization': `Bearer ${testToken1}`
  407. }
  408. });
  409. console.debug('删除无权数据响应状态:', response.status);
  410. expect(response.status).toBe(403); // 权限验证失败返回403 Forbidden
  411. if (response.status === 403) {
  412. const data = await response.json();
  413. expect(data.message).toContain('无权');
  414. }
  415. // 验证数据没有被删除
  416. const existingData = await testRepository.findOne({
  417. where: { id: testData.id }
  418. });
  419. expect(existingData).not.toBeNull();
  420. });
  421. });
  422. describe('禁用数据权限控制的情况', () => {
  423. it('当数据权限控制禁用时应该允许跨用户访问', async () => {
  424. // 创建禁用数据权限控制的路由
  425. const noPermissionRoutes = createCrudRoutes({
  426. entity: TestEntity,
  427. createSchema: createTestSchema,
  428. updateSchema: updateTestSchema,
  429. getSchema: getTestSchema,
  430. listSchema: listTestSchema,
  431. middleware: [mockAuthMiddleware],
  432. dataPermission: {
  433. enabled: false, // 禁用权限控制
  434. userIdField: 'userId'
  435. }
  436. });
  437. const noPermissionClient = testClient(noPermissionRoutes);
  438. // 创建属于用户2的数据
  439. const dataSource = await IntegrationTestDatabase.getDataSource();
  440. const testRepository = dataSource.getRepository(TestEntity);
  441. const testData = testRepository.create({
  442. name: '用户2的数据',
  443. userId: testUser2.id
  444. });
  445. await testRepository.save(testData);
  446. // 用户1应该能够访问用户2的数据(权限控制已禁用)
  447. const response = await noPermissionClient[':id'].$get({
  448. param: { id: testData.id }
  449. }, {
  450. headers: {
  451. 'Authorization': `Bearer ${testToken1}`
  452. }
  453. });
  454. console.debug('禁用权限控制时的响应状态:', response.status);
  455. if (response.status !== 200) {
  456. try {
  457. const errorData = await response.json();
  458. console.debug('禁用权限控制时的错误信息:', errorData);
  459. } catch (e) {
  460. const text = await response.text();
  461. console.debug('禁用权限控制时的响应文本:', text);
  462. }
  463. }
  464. expect(response.status).toBe(200);
  465. if (response.status === 200) {
  466. const data = await response.json();
  467. expect(data.id).toBe(testData.id);
  468. expect(data.userId).toBe(testUser2.id);
  469. }
  470. });
  471. it('当不传递dataPermission配置时应该允许跨用户访问', async () => {
  472. // 创建不传递数据权限控制的路由
  473. const noPermissionRoutes = createCrudRoutes({
  474. entity: TestEntity,
  475. createSchema: createTestSchema,
  476. updateSchema: updateTestSchema,
  477. getSchema: getTestSchema,
  478. listSchema: listTestSchema,
  479. middleware: [mockAuthMiddleware]
  480. // 不传递 dataPermission 配置
  481. });
  482. const noPermissionClient = testClient(noPermissionRoutes);
  483. // 创建属于用户2的数据
  484. const dataSource = await IntegrationTestDatabase.getDataSource();
  485. const testRepository = dataSource.getRepository(TestEntity);
  486. const testData = testRepository.create({
  487. name: '用户2的数据(无权限配置)',
  488. userId: testUser2.id
  489. });
  490. await testRepository.save(testData);
  491. // 用户1应该能够访问用户2的数据(没有权限控制配置)
  492. console.debug('测试数据ID(无权限配置):', testData.id);
  493. const response = await noPermissionClient[':id'].$get({
  494. param: { id: testData.id }
  495. }, {
  496. headers: {
  497. 'Authorization': `Bearer ${testToken1}`
  498. }
  499. });
  500. console.debug('无权限配置时的响应状态:', response.status);
  501. expect(response.status).toBe(200);
  502. if (response.status === 200) {
  503. const data = await response.json();
  504. expect(data.id).toBe(testData.id);
  505. expect(data.userId).toBe(testUser2.id);
  506. }
  507. });
  508. });
  509. describe('默认过滤条件配置', () => {
  510. it('应该支持listFilters和detailFilters分别配置', async () => {
  511. // 定义Schema
  512. const createTestSchemaWithStatus = z.object({
  513. name: z.string().min(1, '名称不能为空'),
  514. status: z.number().int().min(0).max(1),
  515. userId: z.number().optional()
  516. });
  517. const updateTestSchemaWithStatus = z.object({
  518. name: z.string().min(1, '名称不能为空').optional(),
  519. status: z.number().int().min(0).max(1).optional()
  520. });
  521. const getTestSchemaWithStatus = z.object({
  522. id: z.number(),
  523. name: z.string(),
  524. status: z.number(),
  525. userId: z.number()
  526. });
  527. const listTestSchemaWithStatus = z.object({
  528. id: z.number(),
  529. name: z.string(),
  530. status: z.number(),
  531. userId: z.number()
  532. });
  533. // 创建带有listFilters和detailFilters的路由
  534. const filteredRoutes = createCrudRoutes({
  535. entity: TestEntityWithStatus,
  536. createSchema: createTestSchemaWithStatus,
  537. updateSchema: updateTestSchemaWithStatus,
  538. getSchema: getTestSchemaWithStatus,
  539. listSchema: listTestSchemaWithStatus,
  540. middleware: [mockAuthMiddleware],
  541. // 列表查询:只返回状态为1的数据
  542. listFilters: { status: 1 },
  543. // 详情查询:没有过滤,可以访问任何状态的数据
  544. detailFilters: undefined
  545. });
  546. const filteredClient = testClient(filteredRoutes);
  547. // 创建测试数据
  548. const dataSource = await IntegrationTestDatabase.getDataSource();
  549. const testRepository = dataSource.getRepository(TestEntityWithStatus);
  550. // 创建可用状态的数据
  551. const availableData = testRepository.create({
  552. name: '可用数据',
  553. status: 1,
  554. userId: testUser1.id
  555. });
  556. await testRepository.save(availableData);
  557. // 创建不可用状态的数据
  558. const unavailableData = testRepository.create({
  559. name: '不可用数据',
  560. status: 0,
  561. userId: testUser1.id
  562. });
  563. await testRepository.save(unavailableData);
  564. // 测试列表查询:应该只返回可用状态的数据
  565. const listResponse = await filteredClient.index.$get({
  566. query: {
  567. page: 1,
  568. pageSize: 10
  569. }
  570. }, {
  571. headers: {
  572. 'Authorization': `Bearer ${testToken1}`
  573. }
  574. });
  575. expect(listResponse.status).toBe(200);
  576. const listData = await listResponse.json();
  577. // 类型检查:确保是成功响应
  578. if ('data' in listData) {
  579. expect(listData.data).toHaveLength(1); // 只返回可用状态的数据
  580. expect(listData.data[0].id).toBe(availableData.id);
  581. expect(listData.data[0].status).toBe(1);
  582. } else {
  583. throw new Error('列表查询失败: ' + JSON.stringify(listData));
  584. }
  585. // 测试详情查询:可以访问不可用状态的数据(detailFilters为空)
  586. const detailResponse = await filteredClient[':id'].$get({
  587. param: { id: unavailableData.id }
  588. }, {
  589. headers: {
  590. 'Authorization': `Bearer ${testToken1}`
  591. }
  592. });
  593. expect(detailResponse.status).toBe(200);
  594. const detailData = await detailResponse.json();
  595. // 类型检查:确保是成功响应
  596. if ('id' in detailData) {
  597. expect(detailData.id).toBe(unavailableData.id);
  598. expect(detailData.status).toBe(0);
  599. } else {
  600. throw new Error('详情查询失败: ' + JSON.stringify(detailData));
  601. }
  602. });
  603. it('应该支持向后兼容的defaultFilters', async () => {
  604. // 定义Schema
  605. const createTestSchemaWithStatus = z.object({
  606. name: z.string().min(1, '名称不能为空'),
  607. status: z.number().int().min(0).max(1),
  608. userId: z.number().optional()
  609. });
  610. const updateTestSchemaWithStatus = z.object({
  611. name: z.string().min(1, '名称不能为空').optional(),
  612. status: z.number().int().min(0).max(1).optional()
  613. });
  614. const getTestSchemaWithStatus = z.object({
  615. id: z.number(),
  616. name: z.string(),
  617. status: z.number(),
  618. userId: z.number()
  619. });
  620. const listTestSchemaWithStatus = z.object({
  621. id: z.number(),
  622. name: z.string(),
  623. status: z.number(),
  624. userId: z.number()
  625. });
  626. // 创建只使用defaultFilters的路由(向后兼容)
  627. const defaultFilteredRoutes = createCrudRoutes({
  628. entity: TestEntityWithStatus,
  629. createSchema: createTestSchemaWithStatus,
  630. updateSchema: updateTestSchemaWithStatus,
  631. getSchema: getTestSchemaWithStatus,
  632. listSchema: listTestSchemaWithStatus,
  633. middleware: [mockAuthMiddleware],
  634. // 只使用defaultFilters(旧方式)
  635. defaultFilters: { status: 1 }
  636. });
  637. const defaultFilteredClient = testClient(defaultFilteredRoutes);
  638. // 创建测试数据
  639. const dataSource = await IntegrationTestDatabase.getDataSource();
  640. const testRepository = dataSource.getRepository(TestEntityWithStatus);
  641. // 创建可用状态的数据
  642. const availableData = testRepository.create({
  643. name: '可用数据(defaultFilters)',
  644. status: 1,
  645. userId: testUser1.id
  646. });
  647. await testRepository.save(availableData);
  648. // 创建不可用状态的数据
  649. const unavailableData = testRepository.create({
  650. name: '不可用数据(defaultFilters)',
  651. status: 0,
  652. userId: testUser1.id
  653. });
  654. await testRepository.save(unavailableData);
  655. // 测试列表查询:应该只返回可用状态的数据
  656. const listResponse = await defaultFilteredClient.index.$get({
  657. query: {
  658. page: 1,
  659. pageSize: 10
  660. }
  661. }, {
  662. headers: {
  663. 'Authorization': `Bearer ${testToken1}`
  664. }
  665. });
  666. expect(listResponse.status).toBe(200);
  667. const listData = await listResponse.json();
  668. // 类型检查:确保是成功响应
  669. if ('data' in listData) {
  670. expect(listData.data).toHaveLength(1); // 只返回可用状态的数据
  671. expect(listData.data[0].id).toBe(availableData.id);
  672. expect(listData.data[0].status).toBe(1);
  673. } else {
  674. throw new Error('列表查询失败: ' + JSON.stringify(listData));
  675. }
  676. // 测试详情查询:不应该过滤不可用状态的数据(defaultFilters只应用于列表查询,不应用于详情查询)
  677. const detailResponse = await defaultFilteredClient[':id'].$get({
  678. param: { id: unavailableData.id }
  679. }, {
  680. headers: {
  681. 'Authorization': `Bearer ${testToken1}`
  682. }
  683. });
  684. expect(detailResponse.status).toBe(200); // 可以访问不可用状态的数据(defaultFilters不应用于详情查询)
  685. const detailData = await detailResponse.json();
  686. // 类型检查:确保是成功响应
  687. if ('id' in detailData) {
  688. expect(detailData.id).toBe(unavailableData.id);
  689. expect(detailData.status).toBe(0);
  690. } else {
  691. throw new Error('详情查询失败: ' + JSON.stringify(detailData));
  692. }
  693. });
  694. it('应该支持listFilters和detailFilters的优先级高于defaultFilters', async () => {
  695. // 定义Schema
  696. const createTestSchemaWithStatus = z.object({
  697. name: z.string().min(1, '名称不能为空'),
  698. status: z.number().int().min(0).max(1),
  699. userId: z.number().optional()
  700. });
  701. const updateTestSchemaWithStatus = z.object({
  702. name: z.string().min(1, '名称不能为空').optional(),
  703. status: z.number().int().min(0).max(1).optional()
  704. });
  705. const getTestSchemaWithStatus = z.object({
  706. id: z.number(),
  707. name: z.string(),
  708. status: z.number(),
  709. userId: z.number()
  710. });
  711. const listTestSchemaWithStatus = z.object({
  712. id: z.number(),
  713. name: z.string(),
  714. status: z.number(),
  715. userId: z.number()
  716. });
  717. // 创建同时有defaultFilters、listFilters和detailFilters的路由
  718. const mixedFilteredRoutes = createCrudRoutes({
  719. entity: TestEntityWithStatus,
  720. createSchema: createTestSchemaWithStatus,
  721. updateSchema: updateTestSchemaWithStatus,
  722. getSchema: getTestSchemaWithStatus,
  723. listSchema: listTestSchemaWithStatus,
  724. middleware: [mockAuthMiddleware],
  725. // defaultFilters(旧方式,应该被忽略)
  726. defaultFilters: { status: 0 }, // 默认过滤状态为0的数据
  727. // listFilters(新方式,优先级更高)
  728. listFilters: { status: 1 }, // 列表过滤状态为1的数据
  729. // detailFilters(新方式,优先级更高)
  730. detailFilters: undefined // 详情查询不过滤
  731. });
  732. const mixedFilteredClient = testClient(mixedFilteredRoutes);
  733. // 创建测试数据
  734. const dataSource = await IntegrationTestDatabase.getDataSource();
  735. const testRepository = dataSource.getRepository(TestEntityWithStatus);
  736. // 创建可用状态的数据
  737. const availableData = testRepository.create({
  738. name: '可用数据(混合过滤)',
  739. status: 1,
  740. userId: testUser1.id
  741. });
  742. await testRepository.save(availableData);
  743. // 创建不可用状态的数据
  744. const unavailableData = testRepository.create({
  745. name: '不可用数据(混合过滤)',
  746. status: 0,
  747. userId: testUser1.id
  748. });
  749. await testRepository.save(unavailableData);
  750. // 测试列表查询:应该使用listFilters(status: 1),而不是defaultFilters(status: 0)
  751. const listResponse = await mixedFilteredClient.index.$get({
  752. query: {
  753. page: 1,
  754. pageSize: 10
  755. }
  756. }, {
  757. headers: {
  758. 'Authorization': `Bearer ${testToken1}`
  759. }
  760. });
  761. expect(listResponse.status).toBe(200);
  762. const listData = await listResponse.json();
  763. // 类型检查:确保是成功响应
  764. if ('data' in listData) {
  765. expect(listData.data).toHaveLength(1); // 只返回可用状态的数据(status: 1)
  766. expect(listData.data[0].id).toBe(availableData.id);
  767. expect(listData.data[0].status).toBe(1);
  768. } else {
  769. throw new Error('列表查询失败: ' + JSON.stringify(listData));
  770. }
  771. // 测试详情查询:应该使用detailFilters(空),可以访问不可用状态的数据
  772. const detailResponse = await mixedFilteredClient[':id'].$get({
  773. param: { id: unavailableData.id }
  774. }, {
  775. headers: {
  776. 'Authorization': `Bearer ${testToken1}`
  777. }
  778. });
  779. expect(detailResponse.status).toBe(200); // 可以访问不可用状态的数据
  780. const detailData = await detailResponse.json();
  781. // 类型检查:确保是成功响应
  782. if ('id' in detailData) {
  783. expect(detailData.id).toBe(unavailableData.id);
  784. expect(detailData.status).toBe(0);
  785. } else {
  786. throw new Error('详情查询失败: ' + JSON.stringify(detailData));
  787. }
  788. });
  789. });
  790. });