110 lines
3.6 KiB
Python
110 lines
3.6 KiB
Python
from pathlib import Path
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
|
|
from lnbits.core.crud.extensions import create_user_extension, get_user_extension
|
|
from lnbits.core.crud.users import get_account
|
|
from lnbits.core.models.extensions import (
|
|
Extension,
|
|
UserExtension,
|
|
)
|
|
from lnbits.core.models.users import AccountId
|
|
from lnbits.core.views.extensions_builder_api import (
|
|
api_build_extension,
|
|
api_delete_extension_builder_data,
|
|
api_deploy_extension,
|
|
api_preview_extension,
|
|
)
|
|
from lnbits.settings import Settings
|
|
from tests.helpers import make_extension_data, make_extension_release
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_extensions_builder_api_build_preview_and_cleanup(
|
|
tmp_path, settings: Settings, mocker, from_user
|
|
):
|
|
ext_id = f"builder_{uuid4().hex[:8]}"
|
|
data = make_extension_data(ext_id)
|
|
build_dir = tmp_path / "build"
|
|
build_dir.mkdir(parents=True, exist_ok=True)
|
|
(build_dir / "index.txt").write_text("hello")
|
|
|
|
original_data_folder = settings.lnbits_data_folder
|
|
build_mock = mocker.patch(
|
|
"lnbits.core.views.extensions_builder_api.build_extension_from_data",
|
|
mocker.AsyncMock(
|
|
return_value=(make_extension_release(ext_id, "0.1.0"), build_dir)
|
|
),
|
|
)
|
|
clean_mock = mocker.patch(
|
|
"lnbits.core.views.extensions_builder_api.clean_extension_builder_data"
|
|
)
|
|
|
|
try:
|
|
settings.lnbits_data_folder = str(tmp_path)
|
|
|
|
build_response = await api_build_extension(data)
|
|
assert Path(build_response.path).is_file()
|
|
assert build_response.filename == f"{ext_id}.zip"
|
|
|
|
preview = await api_preview_extension(data, AccountId(id=from_user.id))
|
|
assert preview.success is True
|
|
assert ext_id in preview.message
|
|
|
|
cleaned = await api_delete_extension_builder_data()
|
|
assert cleaned.success is True
|
|
clean_mock.assert_called_once()
|
|
assert build_mock.await_count == 2
|
|
finally:
|
|
settings.lnbits_data_folder = original_data_folder
|
|
|
|
|
|
@pytest.mark.anyio
|
|
async def test_extensions_builder_api_deploy_updates_user_extension(
|
|
tmp_path, settings: Settings, mocker, admin_user
|
|
):
|
|
ext_id = f"deploy_{uuid4().hex[:8]}"
|
|
data = make_extension_data(ext_id)
|
|
account = await get_account(admin_user.id)
|
|
assert account is not None
|
|
|
|
build_root = tmp_path / "deploy-root" / ext_id
|
|
build_root.mkdir(parents=True, exist_ok=True)
|
|
(build_root / "manifest.json").write_text("{}")
|
|
|
|
original_data_folder = settings.lnbits_data_folder
|
|
await create_user_extension(
|
|
UserExtension(user=account.id, extension=ext_id, active=False)
|
|
)
|
|
|
|
mocker.patch(
|
|
"lnbits.core.views.extensions_builder_api.build_extension_from_data",
|
|
mocker.AsyncMock(
|
|
return_value=(make_extension_release(ext_id, "0.1.0"), build_root)
|
|
),
|
|
)
|
|
install_mock = mocker.patch(
|
|
"lnbits.core.views.extensions_builder_api.install_extension",
|
|
mocker.AsyncMock(return_value=Extension(code=ext_id, is_valid=True)),
|
|
)
|
|
activate_mock = mocker.patch(
|
|
"lnbits.core.views.extensions_builder_api.activate_extension",
|
|
mocker.AsyncMock(),
|
|
)
|
|
|
|
try:
|
|
settings.lnbits_data_folder = str(tmp_path)
|
|
deployed = await api_deploy_extension(data, account=account)
|
|
finally:
|
|
settings.lnbits_data_folder = original_data_folder
|
|
|
|
assert deployed.success is True
|
|
assert ext_id in deployed.message
|
|
install_mock.assert_awaited_once()
|
|
activate_mock.assert_awaited_once()
|
|
|
|
user_ext = await get_user_extension(account.id, ext_id)
|
|
assert user_ext is not None
|
|
assert user_ext.active is True
|