chore: patch 3rd party calls on tests (!1009) · Merge requests · GitLab.org / ModelOps / AI Assisted (formerly Applied ML) / Code Suggestions / AI Gateway · GitLab

Alejandro Rodríguez requested to merge mock-test-calls into main Jun 27, 2024

What does this merge request do and why?

chore: patch 3rd party calls on tests

This is a follow-up to !986 (merged) where we introduced unittest.Mock to avoid 3rd party provider calls. We missed a couple of spots, which became apparent on this CI failure where real requests are leaking through https://gitlab.com/gitlab-org/modelops/applied-ml/code-suggestions/ai-assist/-/jobs/7205043458 We patch those test cases properly, and also delete some unneeded imports after the aforementioned refactor.

Related to #511 (closed)

How to set up and validate locally

Numbered steps to set up and validate the change are strongly suggested.

Merge request checklist

Tests added for new functionality. If not, please raise an issue to follow up.
Documentation added/updated, if needed.