zuowei/erpnew
1
0
Fork 0
forked from erp-dev/erp
Code Pull Requests Activity

fix: limit param for tiia(tencent)

Browse Source
This commit is contained in:
colaftc
2026-01-20 17:45:09 +08:00
parent c170381383
commit a5e3312d5c
4 changed files with 134 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
api_v1/views/test_tiia_search_image_api.py
View File

@@ -43,6 +43,47 @@ class TiiaSearchImageAPITestCase(TestCase):
self.assertEqual(resp.json(), {'Candidates': [{'EntityId': '1', 'Score': 99}]})
mock_search.assert_called()
@patch('api_v1.views.tiia.search_image_url_in_tencent_tiia')
def test_search_image_pass_limit_offset(self, mock_search):
self.client.force_authenticate(self.user)
mock_search.return_value = {'Candidates': []}
resp = self.client.post(
'/api/v1/tiia/search-image/',
data={
'imageUrl': 'https://example.com/a.png',
'limit': 50,
'offset': 10,
'matchThreshold': 80,
},
format='json',
)
self.assertEqual(resp.status_code, status.HTTP_200_OK)
_, kwargs = mock_search.call_args
self.assertEqual(kwargs.get('limit'), 50)
self.assertEqual(kwargs.get('offset'), 10)
self.assertEqual(kwargs.get('match_threshold'), 80)
def test_search_image_limit_too_large(self):
self.client.force_authenticate(self.user)
resp = self.client.post(
'/api/v1/tiia/search-image/',
data={'imageUrl': 'https://example.com/a.png', 'limit': 101},
format='json',
)
self.assertEqual(resp.status_code, status.HTTP_400_BAD_REQUEST)
self.assertIn('error', resp.json())
def test_search_image_offset_negative(self):
self.client.force_authenticate(self.user)
resp = self.client.post(
'/api/v1/tiia/search-image/',
data={'imageUrl': 'https://example.com/a.png', 'offset': -1},
format='json',
)
self.assertEqual(resp.status_code, status.HTTP_400_BAD_REQUEST)
self.assertIn('error', resp.json())
@patch('api_v1.views.tiia.search_image_url_in_tencent_tiia')
def test_search_image_success_base64(self, mock_search):
self.client.force_authenticate(self.user)