e715a150b9
Allow opting in for backup to move the files instead of copying them. Fixes: https://github.com/scylladb/scylladb/issues/24372 Closes scylladb/scylladb#24503
110 lines
3.5 KiB
Python
110 lines
3.5 KiB
Python
#
|
|
# Copyright 2023-present ScyllaDB
|
|
#
|
|
# SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
#
|
|
|
|
import pytest
|
|
|
|
from test.nodetool.rest_api_mock import expected_request
|
|
|
|
|
|
def test_disablebackup(nodetool):
|
|
nodetool("disablebackup", expected_requests=[
|
|
expected_request("POST", "/storage_service/incremental_backups", params={"value": "false"})])
|
|
|
|
|
|
def test_enablebackup(nodetool):
|
|
nodetool("enablebackup", expected_requests=[
|
|
expected_request("POST", "/storage_service/incremental_backups", params={"value": "true"})])
|
|
|
|
|
|
def test_statusbackup(nodetool):
|
|
res = nodetool("statusbackup", expected_requests=[
|
|
expected_request("GET", "/storage_service/incremental_backups", response=False)])
|
|
assert res.stdout == "not running\n"
|
|
|
|
res = nodetool("statusbackup", expected_requests=[
|
|
expected_request("GET", "/storage_service/incremental_backups", response=True)])
|
|
assert res.stdout == "running\n"
|
|
|
|
|
|
@pytest.mark.parametrize("nowait,task_state,task_error,move_files", [(False, "failed", "error", False),
|
|
(False, "done", "", True),
|
|
(True, "", "", True)])
|
|
def test_backup(nodetool, scylla_only, nowait, task_state, task_error, move_files):
|
|
endpoint = "s3.us-east-2.amazonaws.com"
|
|
bucket = "bucket-foo"
|
|
prefix = "foo/bar/baz"
|
|
keyspace = "ks"
|
|
table = "cf"
|
|
snapshot = "ss"
|
|
params = {"endpoint": endpoint,
|
|
"bucket": bucket,
|
|
"prefix": prefix,
|
|
"keyspace": keyspace,
|
|
"table": table,
|
|
"snapshot": snapshot,
|
|
"move_files": "true" if move_files else "false"}
|
|
task_id = "2c4a3e5f"
|
|
start_time = "2024-08-08T14:29:25Z"
|
|
end_time = "2024-08-08T14:30:42Z"
|
|
task_status = {
|
|
"id": task_id,
|
|
"type": "backup",
|
|
"kind": "node",
|
|
"scope": "node",
|
|
"state": task_state,
|
|
"is_abortable": False,
|
|
"start_time": start_time,
|
|
"end_time": end_time,
|
|
"error": task_error,
|
|
"sequence_number": 0,
|
|
"shard": 0,
|
|
"progress_total": 1.0,
|
|
"progress_completed": 1.0,
|
|
"children_ids": []
|
|
}
|
|
expected_requests = [
|
|
expected_request(
|
|
"POST",
|
|
"/storage_service/backup",
|
|
params,
|
|
response=task_id)
|
|
]
|
|
args = ["backup",
|
|
"--endpoint", endpoint,
|
|
"--bucket", bucket,
|
|
"--prefix", prefix,
|
|
"--keyspace", keyspace,
|
|
"--table", table,
|
|
"--snapshot", snapshot]
|
|
if move_files:
|
|
args.append("--move-files")
|
|
if nowait:
|
|
args.append("--nowait")
|
|
res = nodetool(*args, expected_requests=expected_requests)
|
|
assert task_id in res.stdout
|
|
else:
|
|
# wait for the completion of backup task
|
|
expected_requests.append(
|
|
expected_request(
|
|
"GET",
|
|
f"/task_manager/wait_task/{task_id}",
|
|
response=task_status))
|
|
res = nodetool(*args, expected_requests=expected_requests, check_return_code=False)
|
|
if task_state == "done":
|
|
expected_returncode = 0
|
|
expected_output = f"""{task_state}
|
|
start: {start_time}
|
|
end: {end_time}
|
|
"""
|
|
else:
|
|
expected_returncode = 1
|
|
expected_output = f"""{task_state}: {task_error}
|
|
start: {start_time}
|
|
end: {end_time}
|
|
"""
|
|
assert res.returncode == expected_returncode
|
|
assert res.stdout == expected_output
|