[PR #21502] feat: refactor: test_dataset unit tests #21499 #29689

Closed
opened 2026-02-21 20:46:03 -05:00 by yindo · 0 comments
Owner

Original Pull Request: https://github.com/langgenius/dify/pull/21502

State: closed
Merged: Yes


Important

  1. Make sure you have read our contribution guidelines
  2. Ensure there is an associated issue and you have been assigned to it
  3. Use the correct syntax to link this PR: Fixes #<issue number>.

Summary

Fix https://github.com/langgenius/dify/issues/21499

refactor unit tests for dataset:

  • test_dataset_permission.py
  • test_dataset_service_batch_update_document_status.py
  • test_dataset_service_update_dataset.py

Key Changes:

  • Added DataFactory for creating mock objects with customizable attributes.
  • Enhanced test cases to cover
  • Replaced redundant mock setup code with reusable factory methods.
  • Improved helper methods for asserting permission checks and database query behavior.

These changes improve test clarity and maintainability while ensuring robust validation of the DatasetService logic.

Screenshots

Before After
Redundant mock setup in each test Centralized mock creation via factory
Limited coverage of scenarios Comprehensive coverage of all cases

Checklist

  • This change requires a documentation update, included: Dify Document
  • I understand that this PR may be closed in case there was no previous discussion or issues. (This doesn't apply to typos!)
  • I've added a test for each change that was introduced, and I tried as much as possible to make a single atomic change.
  • I've updated the documentation accordingly.
  • I ran dev/reformat(backend) and cd web && npx lint-staged(frontend) to appease the lint gods.
**Original Pull Request:** https://github.com/langgenius/dify/pull/21502 **State:** closed **Merged:** Yes --- > [!IMPORTANT] > > 1. Make sure you have read our [contribution guidelines](https://github.com/langgenius/dify/blob/main/CONTRIBUTING.md) > 2. Ensure there is an associated issue and you have been assigned to it > 3. Use the correct syntax to link this PR: `Fixes #<issue number>`. ## Summary Fix https://github.com/langgenius/dify/issues/21499 refactor unit tests for dataset: - test_dataset_permission.py - test_dataset_service_batch_update_document_status.py - test_dataset_service_update_dataset.py ### Key Changes: - Added `DataFactory` for creating mock objects with customizable attributes. - Enhanced test cases to cover - Replaced redundant mock setup code with reusable factory methods. - Improved helper methods for asserting permission checks and database query behavior. These changes improve test clarity and maintainability while ensuring robust validation of the `DatasetService` logic. ## Screenshots | Before | After | |--------|-------| | Redundant mock setup in each test | Centralized mock creation via factory | | Limited coverage of scenarios | Comprehensive coverage of all cases | ## Checklist - [ ] This change requires a documentation update, included: [Dify Document](https://github.com/langgenius/dify-docs) - [x] I understand that this PR may be closed in case there was no previous discussion or issues. (This doesn't apply to typos!) - [x] I've added a test for each change that was introduced, and I tried as much as possible to make a single atomic change. - [x] I've updated the documentation accordingly. - [x] I ran `dev/reformat`(backend) and `cd web && npx lint-staged`(frontend) to appease the lint gods.
yindo added the pull-request label 2026-02-21 20:46:03 -05:00
yindo closed this issue 2026-02-21 20:46:03 -05:00
Sign in to join this conversation.
1 Participants
Notifications
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: langgenius/dify#29689