mealie/tests/integration_tests/test_migration_routes.py

import json
import shutil
from pathlib import Path

import pytest
from fastapi.testclient import TestClient

from mealie.core.config import app_dirs
from tests.app_routes import AppRoutes
from tests.test_config import TEST_CHOWDOWN_DIR, TEST_NEXTCLOUD_DIR


@pytest.fixture(scope="session")
def chowdown_zip():
    zip = TEST_CHOWDOWN_DIR.joinpath("test_chowdown-gh-pages.zip")

    zip_copy = TEST_CHOWDOWN_DIR.joinpath("chowdown-gh-pages.zip")

    shutil.copy(zip, zip_copy)

    yield zip_copy

    zip_copy.unlink()


def test_upload_chowdown_zip(api_client: TestClient, api_routes: AppRoutes, chowdown_zip: Path, admin_token):
    upload_url = api_routes.migrations_import_type_upload("chowdown")
    response = api_client.post(upload_url, files={"archive": chowdown_zip.open("rb")}, headers=admin_token)

    assert response.status_code == 200

    assert app_dirs.MIGRATION_DIR.joinpath("chowdown", chowdown_zip.name).is_file()


def test_import_chowdown_directory(api_client: TestClient, api_routes: AppRoutes, chowdown_zip: Path, admin_token):
    delete_url = api_routes.recipes_recipe_slug("roasted-okra")
    api_client.delete(delete_url, headers=admin_token)  # TODO: Manage Test Data better
    selection = chowdown_zip.name

    import_url = api_routes.migrations_import_type_file_name_import("chowdown", selection)
    response = api_client.post(import_url, headers=admin_token)

    assert response.status_code == 200

    reports = json.loads(response.content)

    for report in reports:
        assert report.get("status") is True


def test_delete_chowdown_migration_data(api_client: TestClient, api_routes: AppRoutes, chowdown_zip: Path, admin_token):
    selection = chowdown_zip.name
    delete_url = api_routes.migrations_import_type_file_name_delete("chowdown", selection)
    response = api_client.delete(delete_url, headers=admin_token)

    assert response.status_code == 200
    assert not app_dirs.MIGRATION_DIR.joinpath(chowdown_zip.name).is_file()


# Nextcloud
@pytest.fixture(scope="session")
def nextcloud_zip():
    zip = TEST_NEXTCLOUD_DIR.joinpath("nextcloud.zip")

    zip_copy = TEST_NEXTCLOUD_DIR.joinpath("new_nextcloud.zip")

    shutil.copy(zip, zip_copy)

    yield zip_copy

    zip_copy.unlink()


def test_upload_nextcloud_zip(api_client: TestClient, api_routes: AppRoutes, nextcloud_zip, admin_token):
    upload_url = api_routes.migrations_import_type_upload("nextcloud")
    response = api_client.post(upload_url, files={"archive": nextcloud_zip.open("rb")}, headers=admin_token)

    assert response.status_code == 200

    assert app_dirs.MIGRATION_DIR.joinpath("nextcloud", nextcloud_zip.name).is_file()


def test_import_nextcloud_directory(api_client: TestClient, api_routes: AppRoutes, nextcloud_zip, admin_token):
    selection = nextcloud_zip.name
    import_url = api_routes.migrations_import_type_file_name_import("nextcloud", selection)
    response = api_client.post(import_url, headers=admin_token)

    assert response.status_code == 200

    reports = json.loads(response.content)
    for report in reports:
        assert report.get("status") is True


def test_delete__nextcloud_migration_data(
    api_client: TestClient, api_routes: AppRoutes, nextcloud_zip: Path, admin_token
):
    selection = nextcloud_zip.name
    delete_url = api_routes.migrations_import_type_file_name_delete("nextcloud", selection)
    response = api_client.delete(delete_url, headers=admin_token)

    assert response.status_code == 200
    assert not app_dirs.MIGRATION_DIR.joinpath(nextcloud_zip.name).is_file()