async def evaluate_handler(
item: EvaluateRequest,
request: Request = Depends(AuthorizationRequired()),
Authorization: str = Header(..., description="Bearer your_key"),
):
result = evaluate.evaluate_handler(Authorization, item.language)
if result.get("status") and result.get("error"):
raise HTTPException(status_code=400, detail=result)
return result
async def evaluate_handler(
language: str = Form(...),
repository_version: Optional[int] = Form(default=None),
request: Request = Depends(AuthorizationRequired()),
Authorization: str = Header(..., description="Bearer your_key"),
):
result = evaluate.evaluate_handler(Authorization, language,
repository_version)
if result.get("status") and result.get("error"):
raise HTTPException(status_code=400, detail=result)
return result
def test_invalid_auth(self):
with self.assertRaises(AuthorizationIsRequired):
evaluate_handler("", "en", self.repository_version)
with self.assertRaises(AuthorizationIsRequired):
evaluate_handler(None, "en", self.repository_version)
with self.assertRaises(AuthorizationIsRequired):
evaluate_handler(2, "en", self.repository_version)
with self.assertRaises(AuthorizationIsRequired):
crossvalidation_evaluate_handler("", "en", self.repository_version)
with self.assertRaises(AuthorizationIsRequired):
crossvalidation_evaluate_handler(None, "en", self.repository_version)
with self.assertRaises(AuthorizationIsRequired):
crossvalidation_evaluate_handler(2, "en", self.repository_version)
def test_invalid_lang(self):
with self.assertRaises(ValidationError):
evaluate_handler(
self.authorization, "unknown_lang", self.repository_version
)
with self.assertRaises(ValidationError):
evaluate_handler(self.authorization, None, self.repository_version)
with self.assertRaises(ValidationError):
evaluate_handler(self.authorization, 2, self.repository_version)
with self.assertRaises(ValidationError):
crossvalidation_evaluate_handler(
self.authorization, "unknown_lang", self.repository_version
)
with self.assertRaises(ValidationError):
crossvalidation_evaluate_handler(
self.authorization, None, self.repository_version
)
with self.assertRaises(ValidationError):
crossvalidation_evaluate_handler(
self.authorization, 2, self.repository_version
)
def test_evaluate_untrained_model(self, *args):
with self.assertRaises(ValidationError):
evaluate_handler(self.authorization, "en", self.repository_version)
def test_evaluate_mocked_celery(self, *args):
evaluate_handler(self.authorization, "en", self.repository_version)
def test_evaluate_mocked_celery_timeout(self, *args):
with self.assertRaises(CeleryTimeoutException):
evaluate_handler(self.authorization, "en", self.repository_version)
