|
10 | 10 | Infinite, |
11 | 11 | Integer, |
12 | 12 | InvalidJsonError, |
| 13 | + InvalidNumberOfStringsToGenerate, |
13 | 14 | InvalidTokenError, |
14 | 15 | MissingOrMalformedTokenError, |
15 | 16 | NotFoundError, |
16 | 17 | QuotaExceededError, |
17 | 18 | Term, |
18 | 19 | TimeoutExceededError, |
19 | 20 | TimeoutTooLargeError, |
20 | | - TooManyStringsToGenerateError, |
21 | 21 | TooManyTermsError, |
22 | 22 | UnauthorizedError, |
23 | 23 | ) |
@@ -160,10 +160,10 @@ async def test_error_handling_timeout_exceeded(async_client): |
160 | 160 | @pytest.mark.asyncio |
161 | 161 | async def test_error_handling_too_many_strings_to_generate(async_client): |
162 | 162 | error_400 = ApiException(status=400) |
163 | | - error_400.body = '{"success": false, "error": "Too many strings", "errorCode": "TooManyStringsToGenerate"}' |
| 163 | + error_400.body = '{"success": false, "error": "Too many strings", "errorCode": "InvalidNumberOfStringsToGenerate"}' |
164 | 164 | async_client._generate_api.strings.side_effect = error_400 |
165 | | - with pytest.raises(TooManyStringsToGenerateError): |
166 | | - await async_client.generate_strings(Term.regex("abc"), 1000) |
| 165 | + with pytest.raises(InvalidNumberOfStringsToGenerate): |
| 166 | + await async_client.generate_strings(Term.regex("abc"), 100, 0) |
167 | 167 |
|
168 | 168 |
|
169 | 169 | @pytest.mark.asyncio |
@@ -375,5 +375,5 @@ async def test_generate_strings(async_client): |
375 | 375 | mock_response = MagicMock() |
376 | 376 | mock_response.data.value = ["", "a", "aa"] |
377 | 377 | async_client._generate_api.strings.return_value = mock_response |
378 | | - result = await async_client.generate_strings(term, 3) |
| 378 | + result = await async_client.generate_strings(term, 3, 0) |
379 | 379 | assert result == ["", "a", "aa"] |
0 commit comments