app->libretranslate; mv tests/ inside libretranslate/

libretranslate/tests/test_api/conftest.py

@@ -0,0 +1,20 @@
+import sys
+import pytest
+
+from libretranslate.app import create_app
+from libretranslate.default_values import DEFAULT_ARGUMENTS
+from libretranslate.main import get_args
+
+
+@pytest.fixture()
+def app():
+    sys.argv = ['']
+    DEFAULT_ARGUMENTS['LOAD_ONLY'] = "en,es"
+    app = create_app(get_args())
+
+    yield app
+
+
+@pytest.fixture()
+def client(app):
+    return app.test_client()

libretranslate/tests/test_api/test_api_detect_language.py

@@ -0,0 +1,26 @@
+import json
+
+
+def test_api_detect_language(client):
+    response = client.post("/detect", data={
+        "q": "Hello"
+    })
+    response_json = json.loads(response.data)
+
+    assert "confidence" in response_json[0] and "language" in response_json[0]
+    assert len(response_json) >= 1
+    assert response.status_code == 200
+
+
+def test_api_detect_language_must_fail_without_parameters(client):
+    response = client.post("/detect")
+    response_json = json.loads(response.data)
+
+    assert "error" in response_json
+    assert response.status_code == 400
+
+
+def test_api_detect_language_must_fail_bad_request_type(client):
+    response = client.get("/detect")
+
+    assert response.status_code == 405

libretranslate/tests/test_api/test_api_frontend_settings.py

@@ -0,0 +1,4 @@
+def test_api_get_frontend_settings(client):
+    response = client.get("/frontend/settings")
+
+    assert response.status_code == 200

libretranslate/tests/test_api/test_api_get_languages.py

@@ -0,0 +1,16 @@
+import json
+
+
+def test_api_get_languages(client):
+    response = client.get("/languages")
+    response_json = json.loads(response.data)
+
+    assert "code" in response_json[0] and "name" in response_json[0]
+    assert len(response_json) >= 1
+    assert response.status_code == 200
+
+
+def test_api_get_languages_must_fail_bad_request_type(client):
+    response = client.post("/languages")
+
+    assert response.status_code == 405

libretranslate/tests/test_api/test_api_spec.py

@@ -0,0 +1,10 @@
+def test_api_get_spec(client):
+    response = client.get("/spec")
+
+    assert response.status_code == 200
+
+
+def test_api_get_spec_must_fail_bad_request_type(client):
+    response = client.post("/spec")
+
+    assert response.status_code == 405

libretranslate/tests/test_api/test_api_translate.py

@@ -0,0 +1,61 @@
+import json
+
+
+def test_api_translate(client):
+    response = client.post("/translate", data={
+        "q": "Hello",
+        "source": "en",
+        "target": "es",
+        "format": "text"
+    })
+
+    response_json = json.loads(response.data)
+
+    assert "translatedText" in response_json
+    assert response.status_code == 200
+
+
+def test_api_translate_batch(client):
+
+    response = client.post("/translate", json={
+        "q": ["Hello", "World"],
+        "source": "en",
+        "target": "es",
+        "format": "text"
+    })
+
+    response_json = json.loads(response.data)
+
+    assert "translatedText" in response_json
+    assert isinstance(response_json["translatedText"], list)
+    assert len(response_json["translatedText"]) == 2
+    assert response.status_code == 200
+
+
+def test_api_translate_unsupported_language(client):
+    response = client.post("/translate", data={
+        "q": "Hello",
+        "source": "en",
+        "target": "zz",
+        "format": "text"
+    })
+
+    response_json = json.loads(response.data)
+
+    assert "error" in response_json
+    assert "zz is not supported" == response_json["error"]
+    assert response.status_code == 400
+
+
+def test_api_translate_missing_parameter(client):
+    response = client.post("/translate", data={
+        "source": "en",
+        "target": "es",
+        "format": "text"
+    })
+
+    response_json = json.loads(response.data)
+
+    assert "error" in response_json
+    assert "Invalid request: missing q parameter" == response_json["error"]
+    assert response.status_code == 400