drafts test

Author: pcrespov

services/web/server/tests/unit/with_dbs/04/products/test_products_handlers.py

@@ -10,12 +10,13 @@
 
 import pytest
 from aiohttp.test_utils import TestClient
-from models_library.api_schemas_webserver.product import GetProduct
+from models_library.api_schemas_webserver.product import ProductGet
 from models_library.products import ProductName
 from pytest_simcore.helpers.assert_checks import assert_status
 from pytest_simcore.helpers.webserver_login import UserInfoDict
 from servicelib.aiohttp import status
 from servicelib.rest_constants import X_PRODUCT_NAME_HEADER
+from servicelib.status_codes_utils import is_2xx_success
 from simcore_postgres_database.constants import QUANTIZE_EXP_ARG
 from simcore_service_webserver.db.models import UserRole
 from simcore_service_webserver.groups.api import auto_add_user_to_product_group
@@ -99,17 +100,62 @@ async def test_get_product(
     response = await client.get("/v0/products/current", headers=current_project_headers)
     data, error = await assert_status(response, status.HTTP_200_OK)
 
-    got_product = GetProduct(**data)
+    got_product = ProductGet(**data)
     assert got_product.name == product_name
     assert got_product.credits_per_usd == expected_credits_per_usd
     assert not error
 
     response = await client.get(f"/v0/products/{product_name}")
     data, error = await assert_status(response, status.HTTP_200_OK)
-    assert got_product == GetProduct(**data)
+    assert got_product == ProductGet(**data)
     assert not error
 
-    response = await client.get("/v0/product/invalid")
+    response = await client.get("/v0/products/invalid")
     data, error = await assert_status(response, status.HTTP_404_NOT_FOUND)
     assert not data
     assert error
+
+
+@pytest.mark.parametrize(
+    "user_role, expected_status_code",
+    [
+        (UserRole.ANONYMOUS, status.HTTP_401_UNAUTHORIZED),
+        (UserRole.GUEST, status.HTTP_403_FORBIDDEN),
+        (UserRole.USER, status.HTTP_200_OK),
+        (UserRole.TESTER, status.HTTP_200_OK),
+        (UserRole.PRODUCT_OWNER, status.HTTP_200_OK),
+        (UserRole.ADMIN, status.HTTP_200_OK),
+    ],
+)
+async def test_get_current_product_ui(
+    product_name: ProductName,
+    logged_user: UserInfoDict,
+    client: TestClient,
+    user_role: UserRole,
+    expected_status_code: int,
+):
+    assert logged_user["role"] == f"{user_role}"
+
+    # give access to user to this product
+    assert client.app
+    await auto_add_user_to_product_group(
+        client.app, user_id=logged_user["id"], product_name=product_name
+    )
+
+    current_project_headers = {X_PRODUCT_NAME_HEADER: product_name}
+    response = await client.get(
+        "/v0/products/current/ui", headers=current_project_headers
+    )
+
+    data, error = await assert_status(response, expected_status_code)
+
+    if is_2xx_success(expected_status_code):
+        # ui is something owned and fully controlled by the front-end
+        # Will be something like the data stored in this file
+        # https://github.com/itisfoundation/osparc-simcore/blob/1dcd369717959348099cc6241822a1f0aff0382c/services/static-webserver/client/source/resource/osparc/new_studies.json
+        assert not error
+        assert data is not None
+        assert isinstance(data, dict)
+    else:
+        assert error
+        assert not data