adding db test

Author: matusdrobuliak66

packages/models-library/src/models_library/license_goods.py

@@ -40,4 +40,4 @@ class LicenseGoodDB(BaseModel):
 
 class LicenseGoodUpdateDB(BaseModel):
     name: str | None = None
-    pricing_plan_id: PricingPlanId
+    pricing_plan_id: PricingPlanId | None = None

services/web/server/src/simcore_service_webserver/licenses/_license_goods_db.py

@@ -163,7 +163,7 @@ async def update(
         return LicenseGoodDB.model_validate(row)
 
 
-async def delete_workspace(
+async def delete(
     app: web.Application,
     connection: AsyncConnection | None = None,
     *,

services/web/server/tests/unit/with_dbs/04/licenses/conftest.py

@@ -0,0 +1,17 @@
+# pylint:disable=unused-variable
+# pylint:disable=unused-argument
+# pylint:disable=redefined-outer-name
+from collections.abc import Iterator
+
+import pytest
+import sqlalchemy as sa
+from simcore_postgres_database.models.projects import projects
+from simcore_postgres_database.models.workspaces import workspaces
+
+
+@pytest.fixture
+def workspaces_clean_db(postgres_db: sa.engine.Engine) -> Iterator[None]:
+    with postgres_db.connect() as con:
+        yield
+        con.execute(workspaces.delete())
+        con.execute(projects.delete())

services/web/server/tests/unit/with_dbs/04/licenses/test_license_goods_db.py

@@ -0,0 +1,96 @@
+from collections.abc import AsyncIterator
+
+# pylint: disable=redefined-outer-name
+# pylint: disable=unused-argument
+# pylint: disable=unused-variable
+# pylint: disable=too-many-arguments
+# pylint: disable=too-many-statements
+from http import HTTPStatus
+
+import pytest
+from aiohttp.test_utils import TestClient
+from models_library.license_goods import (
+    LicenseGoodDB,
+    LicenseGoodUpdateDB,
+    LicenseResourceType,
+)
+from models_library.rest_ordering import OrderBy
+from pytest_simcore.helpers.webserver_login import UserInfoDict
+from servicelib.aiohttp import status
+from simcore_service_webserver.db.models import UserRole
+from simcore_service_webserver.licenses import _license_goods_db
+from simcore_service_webserver.licenses.errors import LicenseGoodNotFoundError
+from simcore_service_webserver.projects.models import ProjectDict
+
+
+@pytest.mark.parametrize("user_role,expected", [(UserRole.USER, status.HTTP_200_OK)])
+async def test_license_goods_db_crud(
+    client: TestClient,
+    logged_user: UserInfoDict,
+    user_project: ProjectDict,
+    osparc_product_name: str,
+    expected: HTTPStatus,
+    workspaces_clean_db: AsyncIterator[None],
+):
+    assert client.app
+
+    output: tuple[int, list[LicenseGoodDB]] = await _license_goods_db.list_(
+        client.app,
+        product_name=osparc_product_name,
+        offset=0,
+        limit=10,
+        order_by=OrderBy(field="modified"),
+    )
+    assert output[0] == 0
+
+    license_good_db = await _license_goods_db.create(
+        client.app,
+        product_name=osparc_product_name,
+        name="Model A",
+        license_resource_type=LicenseResourceType.VIP_MODEL,
+        pricing_plan_id=1,
+    )
+    _license_good_id = license_good_db.license_good_id
+
+    output: tuple[int, list[LicenseGoodDB]] = await _license_goods_db.list_(
+        client.app,
+        product_name=osparc_product_name,
+        offset=0,
+        limit=10,
+        order_by=OrderBy(field="modified"),
+    )
+    assert output[0] == 1
+
+    license_good_db = await _license_goods_db.get(
+        client.app,
+        license_good_id=_license_good_id,
+        product_name=osparc_product_name,
+    )
+    assert license_good_db.name == "Model A"
+
+    await _license_goods_db.update(
+        client.app,
+        license_good_id=_license_good_id,
+        product_name=osparc_product_name,
+        updates=LicenseGoodUpdateDB(name="Model B"),
+    )
+
+    license_good_db = await _license_goods_db.get(
+        client.app,
+        license_good_id=_license_good_id,
+        product_name=osparc_product_name,
+    )
+    assert license_good_db.name == "Model B"
+
+    license_good_db = await _license_goods_db.delete(
+        client.app,
+        license_good_id=_license_good_id,
+        product_name=osparc_product_name,
+    )
+
+    with pytest.raises(LicenseGoodNotFoundError):
+        await _license_goods_db.get(
+            client.app,
+            license_good_id=_license_good_id,
+            product_name=osparc_product_name,
+        )