selfprivacy-rest-api/tests/test_graphql/test_api_backup.py

from tests.test_graphql.test_backup import dummy_service, backups, raw_dummy_service
from tests.common import generate_backup_query

from selfprivacy_api.jobs import Jobs, JobStatus

API_SNAPSHOTS_QUERY = """
allSnapshots {
    id
    service {
        id
    }
    createdAt
}
"""

API_BACK_UP_MUTATION = """
mutation TestBackupService($service_id: String) {
        startBackup(serviceId: $service_id) {
            success
            message
            code
            job { 
                uid
                status
            }
        }
}
"""


def api_backup(authorized_client, service):
    response = authorized_client.post(
        "/graphql",
        json={
            "query": API_BACK_UP_MUTATION,
            "variables": {"service_id": service.get_id()},
        },
    ).json()
    return response


def get_data(response):
    assert response.status_code == 200
    response = response.json()
    assert response["data"] is not None
    data = response["data"]
    return data


def api_snapshots(authorized_client, service):
    response = authorized_client.post(
        "/graphql",
        json={"query": generate_backup_query([API_SNAPSHOTS_QUERY])},
    )
    data = get_data(response)
    result = data["backup"]["allSnapshots"]
    assert result is not None
    return result


def test_snapshots_empty(authorized_client, dummy_service):
    snaps = api_snapshots(authorized_client, dummy_service)
    assert snaps == []


def test_start_backup(authorized_client, dummy_service):
    response = api_backup(authorized_client, dummy_service)
    assert response["data"]["startBackup"]["success"] is True
    job = response["data"]["startBackup"]["job"]
    assert Jobs.get_job(job["uid"]).status == JobStatus.FINISHED