✅ test: add unit test for src/server/routers/lambda/aiModel.ts #6085
+163
−0
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Trigger Info
Summary
This PR introduces unit tests for the
aiModelRouter
module, ensuring comprehensive coverage of its functionalities. The following key features are tested:Batch Operations:
batchToggleAiModels
: Verifies toggling the enabled state of multiple AI models.batchUpdateAiModels
: Ensures batch updates of AI models are handled correctly.Model Management:
clearModelsByProvider
: Tests clearing models associated with a specific provider.clearRemoteModels
: Validates clearing remote models for a given provider.createAiModel
: Confirms the creation of a new AI model and returns the created model's ID.getAiModelById
: Ensures retrieval of an AI model by its ID.removeAiModel
: Verifies the deletion of an AI model by ID and provider.Model State and Order:
toggleModelEnabled
: Tests toggling the enabled state of a specific model.updateAiModel
: Validates updating a model's attributes.updateAiModelOrder
: Ensures the correct reordering of models based on a provided sort map.Mocks are used for dependencies such as
AiModelModel
,UserModel
,AiInfraRepos
,KeyVaultsGateKeeper
, andgetServerGlobalConfig
to isolate the tests and focus on the router's logic. Each test case ensures the corresponding method is called with the expected arguments.Tip
You can
@gru-agent
and leave your feedback. TestGru will make adjustments based on your inputTip
You can
@gru-agent rebase
to rebase the PR.Tip
You can
@gru-agent redo
to reset or rebase before redoing the PR.Tip
To modify the test code yourself, click here Edit Test Code