feat: Mover-Engine mit atomarem Schreiben, Stabilität und Quarantäne
Lokale/Kernel-Mount- und SFTP-Ziele schreiben in .part, fsyncen, prüfen Größe und Quelle und benennen ohne Überschreiben um; erst dann wird die Quelle gelöscht. Stabilitätsfenster, Platzprüfung, Metadaten, Berechtigungen, Kopiermodus mit Gedächtnis, Quarantäne. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 5.5
parent
c3a42ce6f5
commit
5633d3057a
@@ -0,0 +1,489 @@
|
||||
"""Ziele: lokales Verzeichnis bzw. Kernel-Mount, oder SFTP.
|
||||
|
||||
Fallstrick 3 – nie eine wachsende Datei unter ihrem Endnamen, nie beide Kopien
|
||||
verlieren. Ablauf für jede Datei:
|
||||
|
||||
1. in ``<ziel>/.netmover-<uuid>.part`` schreiben, ``flush`` + ``fsync``,
|
||||
Größe prüfen,
|
||||
2. Metadaten setzen (mtime, Modus, Eigentümer – Fehler sind nur Hinweise),
|
||||
3. prüfen, dass sich die Quelle seit der Stabilitätsmessung nicht verändert hat,
|
||||
4. atomar auf den Endnamen umbenennen – ohne zu überschreiben (``link`` bzw.
|
||||
SFTP-``rename``), außer „überschreiben“ ist ausdrücklich eingestellt,
|
||||
5. erst danach löscht der Aufrufer die Quelle.
|
||||
|
||||
Scheitert irgendein Schritt vor 4, wird nur die ``.part``-Datei entfernt; die
|
||||
Quelle ist unangetastet. Bricht der Prozess hart ab, bleibt höchstens eine
|
||||
``.part``-Datei zurück, die :meth:`Destination.prepare` später aufräumt.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import contextlib
|
||||
import errno
|
||||
import os
|
||||
import posixpath
|
||||
import stat
|
||||
import time
|
||||
import uuid
|
||||
from abc import ABC, abstractmethod
|
||||
from collections.abc import Callable
|
||||
from dataclasses import dataclass, field
|
||||
from pathlib import Path
|
||||
from typing import Literal
|
||||
|
||||
from netmover.models import CollisionPolicy
|
||||
from netmover.mounting.sftp import SftpSession
|
||||
from netmover.mover.naming import collision_candidates
|
||||
from netmover.timeouts import CancelToken
|
||||
|
||||
CHUNK = 1024 * 1024
|
||||
STALE_PART_SECONDS = 3600
|
||||
PART_PREFIX = ".netmover-"
|
||||
PART_SUFFIX = ".part"
|
||||
|
||||
FsKind = Literal["local", "cifs", "nfs", "sftp"]
|
||||
|
||||
# Fehler, bei denen ``link`` nicht geht und wir auf „prüfen + rename“ ausweichen.
|
||||
_NO_LINK_ERRNOS = frozenset(
|
||||
{errno.EPERM, errno.ENOTSUP, errno.EOPNOTSUPP, errno.EMLINK, errno.EXDEV, errno.ENOSYS}
|
||||
)
|
||||
|
||||
|
||||
class TransferError(Exception):
|
||||
"""Übertragung dieser Datei gescheitert (Meldung im Klartext)."""
|
||||
|
||||
|
||||
class SourceChanged(TransferError):
|
||||
"""Quelle hat sich seit der Stabilitätsmessung verändert – später erneut versuchen."""
|
||||
|
||||
|
||||
class CollisionSkip(Exception):
|
||||
"""Ziel existiert und die Regel sagt „überspringen“."""
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class MetaOptions:
|
||||
atime_ns: int
|
||||
mtime_ns: int
|
||||
mode: int | None # Rechte-Bits, None = nicht setzen
|
||||
owner: tuple[int, int] | None # None = nicht setzen
|
||||
|
||||
|
||||
@dataclass
|
||||
class CommitResult:
|
||||
final_rel: str
|
||||
notes: list[str] = field(default_factory=list)
|
||||
source_consumed: bool = False # True: Quelle wurde per rename bereits „verschoben“
|
||||
|
||||
|
||||
Verify = Callable[[], None]
|
||||
|
||||
|
||||
class Destination(ABC):
|
||||
kind: FsKind
|
||||
|
||||
@abstractmethod
|
||||
def describe(self, rel: str) -> str:
|
||||
"""Lesbarer Zielpfad für Protokoll und Probelauf."""
|
||||
|
||||
@abstractmethod
|
||||
def prepare(self) -> None:
|
||||
"""Vor dem Lauf: Zielordner prüfen, alte ``.part``-Reste entfernen."""
|
||||
|
||||
@abstractmethod
|
||||
def exists(self, rel: str) -> bool: ...
|
||||
|
||||
@abstractmethod
|
||||
def free_bytes(self) -> int | None: ...
|
||||
|
||||
def close(self) -> None: # noqa: B027 – absichtlich optional
|
||||
pass
|
||||
|
||||
# -- Primitive der Unterklassen -------------------------------------------------
|
||||
|
||||
@abstractmethod
|
||||
def _ensure_parent(self, rel: str) -> None: ...
|
||||
|
||||
@abstractmethod
|
||||
def _upload(self, src: Path, tmp: str, size: int, token: CancelToken) -> list[str]: ...
|
||||
|
||||
@abstractmethod
|
||||
def _apply_meta(self, rel: str, meta: MetaOptions) -> list[str]: ...
|
||||
|
||||
@abstractmethod
|
||||
def _link_noreplace(self, tmp: str, final: str) -> None:
|
||||
"""Atomar umbenennen; ``FileExistsError``, wenn ``final`` existiert."""
|
||||
|
||||
@abstractmethod
|
||||
def _replace(self, tmp: str, final: str) -> list[str]: ...
|
||||
|
||||
@abstractmethod
|
||||
def _remove(self, rel: str) -> None: ...
|
||||
|
||||
def _sync_dir(self, rel_dir: str) -> None: # noqa: B027
|
||||
pass
|
||||
|
||||
# -- gemeinsamer Ablauf ---------------------------------------------------------
|
||||
|
||||
def plan(self, rel: str, policy: CollisionPolicy) -> str | None:
|
||||
"""Voraussichtlicher Endname (Probelauf / Vorabprüfung); None = überspringen."""
|
||||
if policy is CollisionPolicy.OVERWRITE:
|
||||
return rel
|
||||
if policy is CollisionPolicy.SKIP:
|
||||
return None if self.exists(rel) else rel
|
||||
for cand in collision_candidates(rel):
|
||||
if not self.exists(cand):
|
||||
return cand
|
||||
raise TransferError(f"Kein freier Dateiname für {rel}")
|
||||
|
||||
def commit(
|
||||
self,
|
||||
src: Path,
|
||||
size: int,
|
||||
rel: str,
|
||||
policy: CollisionPolicy,
|
||||
meta: MetaOptions,
|
||||
verify: Verify,
|
||||
token: CancelToken,
|
||||
) -> CommitResult:
|
||||
self._ensure_parent(rel)
|
||||
tmp = posixpath.join(
|
||||
posixpath.dirname(rel), f"{PART_PREFIX}{uuid.uuid4().hex}{PART_SUFFIX}"
|
||||
)
|
||||
try:
|
||||
notes = self._upload(src, tmp, size, token)
|
||||
notes += self._apply_meta(tmp, meta)
|
||||
verify()
|
||||
token.check()
|
||||
final, more = self._finalize(tmp, rel, policy)
|
||||
notes += more
|
||||
except BaseException:
|
||||
with contextlib.suppress(Exception):
|
||||
self._remove(tmp)
|
||||
raise
|
||||
self._sync_dir(posixpath.dirname(final))
|
||||
return CommitResult(final, notes)
|
||||
|
||||
def move_same_device(
|
||||
self, src: Path, rel: str, policy: CollisionPolicy, meta: MetaOptions
|
||||
) -> CommitResult | None:
|
||||
"""Schneller Weg ohne Kopie, wenn Quelle und Ziel auf demselben Gerät liegen."""
|
||||
return None
|
||||
|
||||
def _finalize(self, tmp: str, rel: str, policy: CollisionPolicy) -> tuple[str, list[str]]:
|
||||
if policy is CollisionPolicy.OVERWRITE:
|
||||
return rel, self._replace(tmp, rel)
|
||||
candidates = collision_candidates(rel) if policy is CollisionPolicy.RENAME else iter([rel])
|
||||
for cand in candidates:
|
||||
if self.exists(cand):
|
||||
if policy is CollisionPolicy.SKIP:
|
||||
raise CollisionSkip
|
||||
continue
|
||||
try:
|
||||
self._link_noreplace(tmp, cand)
|
||||
except FileExistsError:
|
||||
# Wettlauf: jemand anderes hat den Namen gerade belegt
|
||||
if policy is CollisionPolicy.SKIP:
|
||||
raise CollisionSkip from None
|
||||
continue
|
||||
return cand, []
|
||||
raise TransferError(f"Kein freier Dateiname für {rel}")
|
||||
|
||||
|
||||
class LocalDestination(Destination):
|
||||
"""Lokales Verzeichnis oder Verzeichnis auf einem (geprüften) Kernel-Mount."""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
root: Path,
|
||||
kind: Literal["local", "cifs", "nfs"] = "local",
|
||||
*,
|
||||
create_root: bool = False,
|
||||
) -> None:
|
||||
self.root = root
|
||||
self.kind = kind
|
||||
self.create_root = create_root
|
||||
|
||||
def _p(self, rel: str) -> Path:
|
||||
return self.root / rel
|
||||
|
||||
def describe(self, rel: str) -> str:
|
||||
return str(self._p(rel))
|
||||
|
||||
def prepare(self) -> None:
|
||||
if not self.root.is_dir():
|
||||
if not self.create_root:
|
||||
raise TransferError(f"Zielordner {self.root} existiert nicht")
|
||||
self.root.mkdir(mode=0o755, parents=True, exist_ok=True)
|
||||
cutoff = time.time() - STALE_PART_SECONDS
|
||||
for entry in os.scandir(self.root):
|
||||
if entry.name.startswith(PART_PREFIX) and entry.name.endswith(PART_SUFFIX):
|
||||
with contextlib.suppress(OSError):
|
||||
if entry.stat(follow_symlinks=False).st_mtime < cutoff:
|
||||
os.unlink(entry.path)
|
||||
|
||||
def exists(self, rel: str) -> bool:
|
||||
return os.path.lexists(self._p(rel))
|
||||
|
||||
def free_bytes(self) -> int | None:
|
||||
st = os.statvfs(self.root)
|
||||
return st.f_bavail * st.f_frsize
|
||||
|
||||
def _ensure_parent(self, rel: str) -> None:
|
||||
self._p(rel).parent.mkdir(mode=0o755, parents=True, exist_ok=True)
|
||||
|
||||
def _upload(self, src: Path, tmp: str, size: int, token: CancelToken) -> list[str]:
|
||||
written = 0
|
||||
fd = os.open(self._p(tmp), os.O_WRONLY | os.O_CREAT | os.O_EXCL, 0o600)
|
||||
try:
|
||||
with src.open("rb") as fin:
|
||||
while chunk := fin.read(CHUNK):
|
||||
token.check()
|
||||
view = memoryview(chunk)
|
||||
while view:
|
||||
n = os.write(fd, view)
|
||||
view = view[n:]
|
||||
written += len(chunk)
|
||||
if written != size:
|
||||
raise SourceChanged(
|
||||
"Quelle hat sich während der Übertragung geändert "
|
||||
f"({written} statt {size} Bytes)"
|
||||
)
|
||||
os.fsync(fd)
|
||||
if os.fstat(fd).st_size != size:
|
||||
raise TransferError("Größe der Zieldatei stimmt nach dem Schreiben nicht")
|
||||
finally:
|
||||
os.close(fd)
|
||||
return []
|
||||
|
||||
def _apply_meta(self, rel: str, meta: MetaOptions) -> list[str]:
|
||||
notes: list[str] = []
|
||||
path = self._p(rel)
|
||||
if self.kind != "cifs": # auf CIFS bestimmen die Mount-Optionen Eigentümer/Rechte
|
||||
if meta.owner is not None:
|
||||
try:
|
||||
os.chown(path, *meta.owner)
|
||||
except PermissionError:
|
||||
notes.append(
|
||||
f"Eigentümer {meta.owner[0]}:{meta.owner[1]} nicht gesetzt – keine "
|
||||
"Berechtigung (läuft NetMover als root? Root-Squash?)"
|
||||
)
|
||||
except OSError as exc:
|
||||
notes.append(f"Eigentümer nicht gesetzt: {exc.strerror}")
|
||||
if meta.mode is not None:
|
||||
try:
|
||||
os.chmod(path, meta.mode)
|
||||
except OSError as exc:
|
||||
notes.append(f"Modus {meta.mode:04o} nicht gesetzt: {exc.strerror}")
|
||||
try:
|
||||
os.utime(path, ns=(meta.atime_ns, meta.mtime_ns))
|
||||
except OSError as exc:
|
||||
notes.append(f"Änderungszeit nicht übernommen: {exc.strerror}")
|
||||
return notes
|
||||
|
||||
def _link_noreplace(self, tmp: str, final: str) -> None:
|
||||
src, dst = self._p(tmp), self._p(final)
|
||||
try:
|
||||
os.link(src, dst)
|
||||
except FileExistsError:
|
||||
raise
|
||||
except OSError as exc:
|
||||
if exc.errno not in _NO_LINK_ERRNOS:
|
||||
raise
|
||||
# Dateisystem ohne Hardlinks (häufig CIFS): prüfen + rename.
|
||||
# Kleines Wettlauf-Fenster, aber nie eine halbe Datei unter dem Endnamen.
|
||||
if os.path.lexists(dst):
|
||||
raise FileExistsError(errno.EEXIST, "exists", str(dst)) from None
|
||||
os.rename(src, dst)
|
||||
return
|
||||
with contextlib.suppress(OSError):
|
||||
os.unlink(src) # Rest wird sonst später als alte .part-Datei aufgeräumt
|
||||
|
||||
def _replace(self, tmp: str, final: str) -> list[str]:
|
||||
os.replace(self._p(tmp), self._p(final))
|
||||
return []
|
||||
|
||||
def _remove(self, rel: str) -> None:
|
||||
with contextlib.suppress(FileNotFoundError):
|
||||
os.unlink(self._p(rel))
|
||||
|
||||
def _sync_dir(self, rel_dir: str) -> None:
|
||||
with contextlib.suppress(OSError):
|
||||
fd = os.open(self._p(rel_dir) if rel_dir else self.root, os.O_RDONLY | os.O_DIRECTORY)
|
||||
try:
|
||||
os.fsync(fd)
|
||||
finally:
|
||||
os.close(fd)
|
||||
|
||||
def move_same_device(
|
||||
self, src: Path, rel: str, policy: CollisionPolicy, meta: MetaOptions
|
||||
) -> CommitResult | None:
|
||||
try:
|
||||
same = os.stat(src).st_dev == os.stat(self.root).st_dev
|
||||
except OSError:
|
||||
return None
|
||||
if not same:
|
||||
return None
|
||||
self._ensure_parent(rel)
|
||||
consumed = True
|
||||
if policy is CollisionPolicy.OVERWRITE:
|
||||
os.replace(src, self._p(rel))
|
||||
final = rel
|
||||
else:
|
||||
final = ""
|
||||
candidates = (
|
||||
collision_candidates(rel) if policy is CollisionPolicy.RENAME else iter([rel])
|
||||
)
|
||||
for cand in candidates:
|
||||
if self.exists(cand):
|
||||
if policy is CollisionPolicy.SKIP:
|
||||
raise CollisionSkip
|
||||
continue
|
||||
try:
|
||||
os.link(src, self._p(cand))
|
||||
except FileExistsError:
|
||||
if policy is CollisionPolicy.SKIP:
|
||||
raise CollisionSkip from None
|
||||
continue
|
||||
except OSError as exc:
|
||||
if exc.errno not in _NO_LINK_ERRNOS:
|
||||
raise
|
||||
if self.exists(cand):
|
||||
continue
|
||||
os.rename(src, self._p(cand))
|
||||
final = cand
|
||||
break
|
||||
# Ziel ist vollständig da; den alten Namen entfernt der Aufrufer –
|
||||
# mit derselben Fehlerbehandlung wie nach einer Kopie.
|
||||
consumed = False
|
||||
final = cand
|
||||
break
|
||||
if not final:
|
||||
raise TransferError(f"Kein freier Dateiname für {rel}")
|
||||
notes = self._apply_meta(final, meta)
|
||||
self._sync_dir(posixpath.dirname(final))
|
||||
return CommitResult(final, notes, source_consumed=consumed)
|
||||
|
||||
|
||||
class SftpDestination(Destination):
|
||||
kind: FsKind = "sftp"
|
||||
|
||||
def __init__(self, session: SftpSession, base: str, label: str = "") -> None:
|
||||
self.session = session
|
||||
self.client = session.client
|
||||
self.base = base
|
||||
self.label = label
|
||||
|
||||
def _p(self, rel: str) -> str:
|
||||
return posixpath.join(self.base, rel) if rel else self.base
|
||||
|
||||
def describe(self, rel: str) -> str:
|
||||
prefix = f"{self.label}:" if self.label else "sftp:"
|
||||
return prefix + self._p(rel)
|
||||
|
||||
def close(self) -> None:
|
||||
self.session.close()
|
||||
|
||||
def prepare(self) -> None:
|
||||
try:
|
||||
attrs = self.client.stat(self.base)
|
||||
except OSError as exc:
|
||||
raise TransferError(f"SFTP-Zielordner {self.base}: {exc}") from None
|
||||
if attrs.st_mode is None or not stat.S_ISDIR(attrs.st_mode):
|
||||
raise TransferError(f"SFTP-Ziel {self.base} ist kein Verzeichnis")
|
||||
cutoff = time.time() - STALE_PART_SECONDS
|
||||
for a in self.client.listdir_attr(self.base):
|
||||
is_part = a.filename.startswith(PART_PREFIX) and a.filename.endswith(PART_SUFFIX)
|
||||
if is_part and a.st_mtime is not None and a.st_mtime < cutoff:
|
||||
with contextlib.suppress(OSError):
|
||||
self.client.remove(self._p(a.filename))
|
||||
|
||||
def exists(self, rel: str) -> bool:
|
||||
try:
|
||||
self.client.lstat(self._p(rel))
|
||||
except FileNotFoundError:
|
||||
return False
|
||||
return True
|
||||
|
||||
def free_bytes(self) -> int | None:
|
||||
usage = self.session.statvfs(self.base)
|
||||
return usage[1] if usage else None
|
||||
|
||||
def _ensure_parent(self, rel: str) -> None:
|
||||
parent = posixpath.dirname(rel)
|
||||
if not parent:
|
||||
return
|
||||
current = self.base
|
||||
for part in parent.split("/"):
|
||||
current = posixpath.join(current, part)
|
||||
try:
|
||||
self.client.stat(current)
|
||||
except FileNotFoundError:
|
||||
self.client.mkdir(current, mode=0o755)
|
||||
|
||||
def _upload(self, src: Path, tmp: str, size: int, token: CancelToken) -> list[str]:
|
||||
notes: list[str] = []
|
||||
written = 0
|
||||
with self.client.open(self._p(tmp), "wb") as fh, src.open("rb") as fin:
|
||||
fh.set_pipelined(True)
|
||||
while chunk := fin.read(CHUNK // 4):
|
||||
token.check()
|
||||
fh.write(chunk)
|
||||
written += len(chunk)
|
||||
fh.flush()
|
||||
if written != size:
|
||||
raise SourceChanged(
|
||||
"Quelle hat sich während der Übertragung geändert "
|
||||
f"({written} statt {size} Bytes)"
|
||||
)
|
||||
if not self.session.fsync(fh):
|
||||
notes.append("SFTP-Server unterstützt kein fsync")
|
||||
remote = self.client.stat(self._p(tmp)).st_size
|
||||
if remote != size:
|
||||
raise TransferError(f"Zieldatei hat {remote} statt {size} Bytes")
|
||||
return notes
|
||||
|
||||
def _apply_meta(self, rel: str, meta: MetaOptions) -> list[str]:
|
||||
notes: list[str] = []
|
||||
path = self._p(rel)
|
||||
if meta.owner is not None:
|
||||
try:
|
||||
self.client.chown(path, *meta.owner)
|
||||
except OSError as exc:
|
||||
notes.append(f"Eigentümer per SFTP nicht gesetzt: {exc}")
|
||||
if meta.mode is not None:
|
||||
try:
|
||||
self.client.chmod(path, meta.mode)
|
||||
except OSError as exc:
|
||||
notes.append(f"Modus per SFTP nicht gesetzt: {exc}")
|
||||
try:
|
||||
self.client.utime(path, (meta.atime_ns / 1e9, meta.mtime_ns / 1e9))
|
||||
except OSError as exc:
|
||||
notes.append(f"Änderungszeit nicht übernommen: {exc}")
|
||||
return notes
|
||||
|
||||
def _link_noreplace(self, tmp: str, final: str) -> None:
|
||||
if self.exists(final):
|
||||
raise FileExistsError(errno.EEXIST, "exists", final)
|
||||
try:
|
||||
# SFTPv3-rename überschreibt nicht (OpenSSH: link + unlink).
|
||||
self.client.rename(self._p(tmp), self._p(final))
|
||||
except OSError:
|
||||
if self.exists(final):
|
||||
raise FileExistsError(errno.EEXIST, "exists", final) from None
|
||||
raise
|
||||
|
||||
def _replace(self, tmp: str, final: str) -> list[str]:
|
||||
try:
|
||||
self.client.posix_rename(self._p(tmp), self._p(final))
|
||||
except OSError:
|
||||
# Server ohne posix-rename: löschen + umbenennen (kurz kein Ziel, nie halb)
|
||||
with contextlib.suppress(FileNotFoundError):
|
||||
self.client.remove(self._p(final))
|
||||
self.client.rename(self._p(tmp), self._p(final))
|
||||
return ["Server kann nicht atomar überschreiben – Ziel kurz nicht vorhanden"]
|
||||
return []
|
||||
|
||||
def _remove(self, rel: str) -> None:
|
||||
with contextlib.suppress(FileNotFoundError):
|
||||
self.client.remove(self._p(rel))
|
||||
@@ -0,0 +1,434 @@
|
||||
"""Ablauf eines Regellaufs – ohne Scheduler, ohne Datenbank-Protokoll.
|
||||
|
||||
Reihenfolge pro Lauf:
|
||||
|
||||
1. Ziel vorbereiten, Quelle scannen (Filter, versteckte/unfertige Dateien raus),
|
||||
2. Kopier-Buchführung: bereits übertragene Dateiversionen still auslassen,
|
||||
3. Größe und Mindestalter prüfen,
|
||||
4. Stabilität: nach dem Stabilitätsfenster erneut messen – nur Dateien mit
|
||||
unveränderter Größe **und** mtime gehen weiter (Fallstrick 1),
|
||||
5. pro Datei: ggf. Quarantäne, Kollisionsregel, Platzprüfung (Größe + 5 %,
|
||||
Fallstrick 14), atomare Übertragung, erst dann Quelle löschen.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import errno
|
||||
import os
|
||||
import stat
|
||||
import time
|
||||
from collections.abc import Callable
|
||||
from dataclasses import dataclass, field
|
||||
from pathlib import Path
|
||||
|
||||
from netmover.logging_setup import event, get_logger
|
||||
from netmover.models import (
|
||||
CollisionPolicy,
|
||||
ItemStatus,
|
||||
PatternType,
|
||||
RuleAction,
|
||||
RunStatus,
|
||||
)
|
||||
from netmover.mover.destinations import (
|
||||
CollisionSkip,
|
||||
Destination,
|
||||
LocalDestination,
|
||||
MetaOptions,
|
||||
SourceChanged,
|
||||
TransferError,
|
||||
)
|
||||
from netmover.mover.ledger import FailureInfo, Ledger
|
||||
from netmover.mover.naming import collision_candidates
|
||||
from netmover.mover.scan import Candidate, FileState, Matcher, current_state, scan
|
||||
from netmover.timeouts import Cancelled, CancelToken
|
||||
|
||||
log = get_logger(__name__)
|
||||
|
||||
SPACE_RESERVE = 1.05
|
||||
# Fehler, die auf eine gestörte Anbindung hindeuten: Lauf abbrechen statt jede
|
||||
# weitere Datei scheitern zu lassen – der nächste Lauf prüft und hängt neu ein.
|
||||
MOUNT_ERRNOS = frozenset({errno.ESTALE, errno.EIO, errno.ENOTCONN, errno.EHOSTDOWN})
|
||||
|
||||
|
||||
class RunAborted(Exception):
|
||||
pass
|
||||
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class RuleSnapshot:
|
||||
id: int
|
||||
name: str
|
||||
action: RuleAction = RuleAction.MOVE
|
||||
on_collision: CollisionPolicy = CollisionPolicy.RENAME
|
||||
pattern: str = "*"
|
||||
pattern_type: PatternType = PatternType.GLOB
|
||||
recursive: bool = False
|
||||
stability_seconds: int = 5
|
||||
min_age_seconds: int = 0
|
||||
max_size_mb: int | None = None
|
||||
set_owner: tuple[int, int] | None = None
|
||||
set_mode: int | None = None
|
||||
quarantine_after: int = 3
|
||||
dst_root_squash: bool | None = None # nur NFS-Ziele: Ergebnis des Schreibtests
|
||||
|
||||
|
||||
@dataclass
|
||||
class ItemResult:
|
||||
src: str
|
||||
status: ItemStatus
|
||||
dst: str | None = None
|
||||
size: int | None = None
|
||||
message: str | None = None
|
||||
|
||||
|
||||
@dataclass
|
||||
class RunResult:
|
||||
status: RunStatus = RunStatus.OK
|
||||
dry_run: bool = False
|
||||
seen: int = 0
|
||||
moved: int = 0
|
||||
skipped: int = 0
|
||||
failed: int = 0
|
||||
message: str | None = None
|
||||
items: list[ItemResult] = field(default_factory=list)
|
||||
|
||||
def add(self, item: ItemResult) -> None:
|
||||
self.items.append(item)
|
||||
if item.status in (ItemStatus.MOVED, ItemStatus.COPIED, ItemStatus.PLANNED):
|
||||
self.moved += 1
|
||||
elif item.status in (ItemStatus.SKIPPED, ItemStatus.QUARANTINED):
|
||||
self.skipped += 1
|
||||
else:
|
||||
self.failed += 1
|
||||
|
||||
|
||||
def _describe_os_error(exc: OSError) -> str:
|
||||
where = f" ({exc.filename})" if exc.filename else ""
|
||||
return f"{exc.strerror or exc}{where}"
|
||||
|
||||
|
||||
def _mb(n: int) -> str:
|
||||
return f"{n / 1024 / 1024:.1f} MB"
|
||||
|
||||
|
||||
class Engine:
|
||||
def __init__(
|
||||
self,
|
||||
ledger: Ledger,
|
||||
*,
|
||||
sleep: Callable[[float], None] = time.sleep,
|
||||
clock: Callable[[], float] = time.time,
|
||||
warn_once: Callable[[str], bool] | None = None,
|
||||
) -> None:
|
||||
self.ledger = ledger
|
||||
self._sleep = sleep
|
||||
self._clock = clock
|
||||
self._warned: set[str] = set()
|
||||
self._warn_once = warn_once or self._default_warn_once
|
||||
|
||||
def _default_warn_once(self, key: str) -> bool:
|
||||
if key in self._warned:
|
||||
return False
|
||||
self._warned.add(key)
|
||||
return True
|
||||
|
||||
# -- Lauf -------------------------------------------------------------------------
|
||||
|
||||
def execute(
|
||||
self,
|
||||
rule: RuleSnapshot,
|
||||
src_root: Path,
|
||||
dest: Destination,
|
||||
token: CancelToken,
|
||||
*,
|
||||
dry_run: bool = False,
|
||||
quarantine_root: Path | None = None,
|
||||
) -> RunResult:
|
||||
result = RunResult(dry_run=dry_run)
|
||||
try:
|
||||
self._execute(rule, src_root, dest, token, result, dry_run, quarantine_root)
|
||||
except Cancelled:
|
||||
result.status = RunStatus.ERROR
|
||||
result.message = "Lauf abgebrochen"
|
||||
except RunAborted as exc:
|
||||
result.status = RunStatus.ERROR
|
||||
result.message = str(exc)
|
||||
except TransferError as exc:
|
||||
result.status = RunStatus.ERROR
|
||||
result.message = str(exc)
|
||||
except OSError as exc:
|
||||
result.status = RunStatus.ERROR
|
||||
result.message = _describe_os_error(exc)
|
||||
else:
|
||||
if result.failed:
|
||||
result.status = RunStatus.PARTIAL
|
||||
return result
|
||||
|
||||
def _execute(
|
||||
self,
|
||||
rule: RuleSnapshot,
|
||||
src_root: Path,
|
||||
dest: Destination,
|
||||
token: CancelToken,
|
||||
result: RunResult,
|
||||
dry_run: bool,
|
||||
quarantine_root: Path | None,
|
||||
) -> None:
|
||||
if not src_root.is_dir():
|
||||
raise RunAborted(f"Quellordner {src_root} existiert nicht")
|
||||
if not dry_run:
|
||||
dest.prepare()
|
||||
token.check()
|
||||
|
||||
exclude: set[Path] = set()
|
||||
if quarantine_root is not None:
|
||||
exclude.add(quarantine_root.resolve())
|
||||
if isinstance(dest, LocalDestination):
|
||||
exclude.add(dest.root.resolve()) # Ziel innerhalb der Quelle nie erneut scannen
|
||||
matcher = Matcher(rule.pattern, rule.pattern_type)
|
||||
candidates = list(scan(src_root, matcher, rule.recursive, frozenset(exclude)))
|
||||
result.seen = len(candidates)
|
||||
token.check()
|
||||
|
||||
# Kopier-Buchführung (Fallstrick 11). Gilt auch für „verschieben“, falls
|
||||
# dort einmal die Quelle nicht gelöscht werden konnte.
|
||||
done = self.ledger.copied_keys(rule.id)
|
||||
if not dry_run:
|
||||
self.ledger.prune_copied(rule.id, {c.relpath for c in candidates})
|
||||
fresh = [c for c in candidates if (c.relpath, c.state.size, c.state.mtime_ns) not in done]
|
||||
unchanged = len(candidates) - len(fresh)
|
||||
|
||||
pending: list[Candidate] = []
|
||||
now = self._clock()
|
||||
for c in fresh:
|
||||
if rule.max_size_mb is not None and c.state.size > rule.max_size_mb * 1024 * 1024:
|
||||
result.add(
|
||||
ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.SKIPPED,
|
||||
size=c.state.size,
|
||||
message=f"größer als {rule.max_size_mb} MB",
|
||||
)
|
||||
)
|
||||
elif rule.min_age_seconds and now - c.state.mtime_ns / 1e9 < rule.min_age_seconds:
|
||||
result.add(
|
||||
ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.SKIPPED,
|
||||
size=c.state.size,
|
||||
message=f"jünger als {rule.min_age_seconds} s",
|
||||
)
|
||||
)
|
||||
else:
|
||||
pending.append(c)
|
||||
|
||||
# Stabilität (Fallstrick 1): zweite Messung nach dem Stabilitätsfenster.
|
||||
if pending and rule.stability_seconds > 0:
|
||||
self._sleep(rule.stability_seconds)
|
||||
token.check()
|
||||
stable: list[Candidate] = []
|
||||
for c in pending:
|
||||
st = current_state(c.path)
|
||||
if st is None:
|
||||
continue # inzwischen verschwunden – nichts zu tun
|
||||
if st != c.state:
|
||||
result.add(
|
||||
ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.SKIPPED,
|
||||
size=st.size,
|
||||
message="wird noch geschrieben (Größe/Zeit geändert)",
|
||||
)
|
||||
)
|
||||
continue
|
||||
stable.append(c)
|
||||
|
||||
failures = self.ledger.failures(rule.id)
|
||||
for c in stable:
|
||||
token.check()
|
||||
result.add(self._process(rule, c, dest, token, dry_run, quarantine_root, failures))
|
||||
|
||||
if unchanged:
|
||||
result.message = f"{unchanged} bereits übertragen"
|
||||
|
||||
# -- einzelne Datei ------------------------------------------------------------------
|
||||
|
||||
def _meta(
|
||||
self, rule: RuleSnapshot, path: Path, dest: Destination
|
||||
) -> tuple[MetaOptions, list[str]]:
|
||||
st = os.stat(path)
|
||||
notes: list[str] = []
|
||||
owner = rule.set_owner
|
||||
if owner is not None:
|
||||
if dest.kind == "cifs":
|
||||
owner = None
|
||||
if self._warn_once(f"{rule.id}:cifs-chown"):
|
||||
notes.append(
|
||||
"CIFS-Ziel: Eigentümer bestimmen die Mount-Optionen, chown übersprungen"
|
||||
)
|
||||
elif dest.kind == "nfs" and rule.dst_root_squash is not False:
|
||||
owner = None
|
||||
if self._warn_once(f"{rule.id}:nfs-chown"):
|
||||
reason = "Root-Squash aktiv" if rule.dst_root_squash else "Schreibtest fehlt"
|
||||
notes.append(f"NFS-Ziel: chown übersprungen ({reason})")
|
||||
mode = rule.set_mode if rule.set_mode is not None else stat.S_IMODE(st.st_mode)
|
||||
return MetaOptions(st.st_atime_ns, st.st_mtime_ns, mode, owner), notes
|
||||
|
||||
def _process(
|
||||
self,
|
||||
rule: RuleSnapshot,
|
||||
c: Candidate,
|
||||
dest: Destination,
|
||||
token: CancelToken,
|
||||
dry_run: bool,
|
||||
quarantine_root: Path | None,
|
||||
failures: dict[str, FailureInfo],
|
||||
) -> ItemResult:
|
||||
size = c.state.size
|
||||
failure = failures.get(c.relpath)
|
||||
|
||||
# Quarantäne für wiederholt scheiternde Dateien
|
||||
if (
|
||||
quarantine_root is not None
|
||||
and failure is not None
|
||||
and failure.attempts >= rule.quarantine_after
|
||||
and (failure.size, failure.mtime_ns) == (c.state.size, c.state.mtime_ns)
|
||||
):
|
||||
if dry_run:
|
||||
return ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.PLANNED,
|
||||
str(quarantine_root / c.relpath),
|
||||
size,
|
||||
"würde in Quarantäne verschoben",
|
||||
)
|
||||
return self._quarantine(rule, c, quarantine_root, failure.attempts)
|
||||
|
||||
try:
|
||||
planned = dest.plan(c.relpath, rule.on_collision)
|
||||
except OSError as exc:
|
||||
return self._fail(rule, c, exc, dry_run)
|
||||
if planned is None:
|
||||
return ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.SKIPPED,
|
||||
dest.describe(c.relpath),
|
||||
size,
|
||||
"existiert bereits im Ziel",
|
||||
)
|
||||
if dry_run:
|
||||
msg = (
|
||||
"würde überschreiben"
|
||||
if (rule.on_collision is CollisionPolicy.OVERWRITE and dest.exists(planned))
|
||||
else None
|
||||
)
|
||||
verb = "kopieren" if rule.action is RuleAction.COPY else "verschieben"
|
||||
return ItemResult(
|
||||
c.relpath, ItemStatus.PLANNED, dest.describe(planned), size, msg or f"würde {verb}"
|
||||
)
|
||||
|
||||
try:
|
||||
free = dest.free_bytes()
|
||||
except OSError:
|
||||
free = None
|
||||
if free is not None and free < size * SPACE_RESERVE:
|
||||
return ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.FAILED,
|
||||
dest.describe(planned),
|
||||
size,
|
||||
f"zu wenig Platz am Ziel (frei {_mb(free)}, "
|
||||
f"benötigt {_mb(int(size * SPACE_RESERVE))})",
|
||||
)
|
||||
|
||||
def verify() -> None:
|
||||
if current_state(c.path) != c.state:
|
||||
raise SourceChanged("Quelle hat sich während der Übertragung geändert")
|
||||
|
||||
try:
|
||||
meta, notes = self._meta(rule, c.path, dest)
|
||||
res = None
|
||||
if rule.action is RuleAction.MOVE:
|
||||
res = dest.move_same_device(c.path, c.relpath, rule.on_collision, meta)
|
||||
if res is None:
|
||||
res = dest.commit(c.path, size, c.relpath, rule.on_collision, meta, verify, token)
|
||||
except CollisionSkip:
|
||||
return ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.SKIPPED,
|
||||
dest.describe(c.relpath),
|
||||
size,
|
||||
"existiert bereits im Ziel",
|
||||
)
|
||||
except SourceChanged as exc:
|
||||
return ItemResult(c.relpath, ItemStatus.SKIPPED, None, size, str(exc))
|
||||
except (TransferError, OSError) as exc:
|
||||
return self._fail(rule, c, exc, dry_run)
|
||||
|
||||
notes += res.notes
|
||||
target = dest.describe(res.final_rel)
|
||||
if c.relpath in failures:
|
||||
self.ledger.clear_failure(rule.id, c.relpath)
|
||||
|
||||
if rule.action is RuleAction.COPY:
|
||||
self.ledger.record_copy(rule.id, c.relpath, size, c.state.mtime_ns, target)
|
||||
return ItemResult(c.relpath, ItemStatus.COPIED, target, size, "; ".join(notes) or None)
|
||||
|
||||
if not res.source_consumed:
|
||||
try:
|
||||
os.unlink(c.path)
|
||||
except OSError as exc:
|
||||
# Ziel ist vollständig – Buchführung verhindert Duplikate im nächsten Lauf.
|
||||
self.ledger.record_copy(rule.id, c.relpath, size, c.state.mtime_ns, target)
|
||||
return ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.FAILED,
|
||||
target,
|
||||
size,
|
||||
f"Ziel geschrieben, aber Quelle nicht löschbar: {_describe_os_error(exc)} – "
|
||||
"Datei wird nicht erneut übertragen",
|
||||
)
|
||||
return ItemResult(c.relpath, ItemStatus.MOVED, target, size, "; ".join(notes) or None)
|
||||
|
||||
def _fail(self, rule: RuleSnapshot, c: Candidate, exc: Exception, dry_run: bool) -> ItemResult:
|
||||
if isinstance(exc, OSError) and exc.errno in MOUNT_ERRNOS:
|
||||
raise RunAborted(f"Anbindung gestört: {_describe_os_error(exc)}") from exc
|
||||
msg = _describe_os_error(exc) if isinstance(exc, OSError) else str(exc)
|
||||
if not dry_run:
|
||||
attempts = self.ledger.record_failure(
|
||||
rule.id, c.relpath, c.state.size, c.state.mtime_ns, msg
|
||||
)
|
||||
msg = f"{msg} (Versuch {attempts})"
|
||||
event(log, "file.failed", level=30, rule=rule.name, file=c.relpath, error=msg)
|
||||
return ItemResult(c.relpath, ItemStatus.FAILED, None, c.state.size, msg)
|
||||
|
||||
def _quarantine(
|
||||
self, rule: RuleSnapshot, c: Candidate, root: Path, attempts: int
|
||||
) -> ItemResult:
|
||||
try:
|
||||
for cand in collision_candidates(c.relpath):
|
||||
target = root / cand
|
||||
if os.path.lexists(target):
|
||||
continue
|
||||
target.parent.mkdir(mode=0o755, parents=True, exist_ok=True)
|
||||
os.rename(c.path, target)
|
||||
break
|
||||
else:
|
||||
raise TransferError("kein freier Name im Quarantäneordner")
|
||||
except (OSError, TransferError) as exc:
|
||||
msg = _describe_os_error(exc) if isinstance(exc, OSError) else str(exc)
|
||||
return ItemResult(
|
||||
c.relpath, ItemStatus.FAILED, None, c.state.size, f"Quarantäne gescheitert: {msg}"
|
||||
)
|
||||
self.ledger.clear_failure(rule.id, c.relpath)
|
||||
event(log, "file.quarantined", level=30, rule=rule.name, file=c.relpath)
|
||||
return ItemResult(
|
||||
c.relpath,
|
||||
ItemStatus.QUARANTINED,
|
||||
str(target),
|
||||
c.state.size,
|
||||
f"nach {attempts} Fehlversuchen in Quarantäne verschoben",
|
||||
)
|
||||
|
||||
|
||||
__all__ = ["Engine", "FileState", "ItemResult", "RuleSnapshot", "RunAborted", "RunResult"]
|
||||
@@ -0,0 +1,680 @@
|
||||
"""Mover-Engine mit lokalen Zielen – Fallstricke 1, 3, 4, 10, 11, 14 und Quarantäne."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import errno
|
||||
import os
|
||||
import time
|
||||
from collections.abc import Callable
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.orm import Session, sessionmaker
|
||||
|
||||
from netmover.db import session_scope
|
||||
from netmover.models import (
|
||||
CollisionPolicy,
|
||||
CopiedFile,
|
||||
ItemStatus,
|
||||
PatternType,
|
||||
Rule,
|
||||
RuleAction,
|
||||
RunStatus,
|
||||
)
|
||||
from netmover.mover import destinations as dest_mod
|
||||
from netmover.mover.destinations import LocalDestination, TransferError
|
||||
from netmover.mover.engine import Engine, RuleSnapshot, RunResult
|
||||
from netmover.mover.ledger import Ledger
|
||||
from netmover.mover.naming import collision_candidates
|
||||
from netmover.mover.scan import Matcher
|
||||
from netmover.timeouts import CancelToken
|
||||
from tests.conftest import MountFactory
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def rule_id(make_mount: MountFactory, session_factory: sessionmaker[Session]) -> int:
|
||||
mid = make_mount()
|
||||
with session_scope(session_factory) as s:
|
||||
r = Rule(name="r", src_mount_id=mid, dst_path="/x")
|
||||
s.add(r)
|
||||
s.flush()
|
||||
return r.id
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def ledger(session_factory: sessionmaker[Session]) -> Ledger:
|
||||
return Ledger(session_factory)
|
||||
|
||||
|
||||
Sleeper = Callable[[float], None]
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def hooks() -> list[Callable[[], None]]:
|
||||
"""Aktionen, die während des Stabilitätsfensters ausgeführt werden."""
|
||||
return []
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def engine(ledger: Ledger, hooks: list[Callable[[], None]]) -> Engine:
|
||||
def sleep(_: float) -> None:
|
||||
for h in hooks:
|
||||
h()
|
||||
|
||||
return Engine(ledger, sleep=sleep)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def cross_device(monkeypatch: pytest.MonkeyPatch) -> None:
|
||||
"""tmp_path liegt auf einem Gerät – für Kopier-Tests den Schnellweg abschalten."""
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def src(tmp_path: Path) -> Path:
|
||||
p = tmp_path / "src"
|
||||
p.mkdir()
|
||||
return p
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def dst(tmp_path: Path) -> Path:
|
||||
p = tmp_path / "dst"
|
||||
p.mkdir()
|
||||
return p
|
||||
|
||||
|
||||
def snap(rule_id: int, **kw: object) -> RuleSnapshot:
|
||||
return RuleSnapshot(id=rule_id, name="r", **kw) # type: ignore[arg-type]
|
||||
|
||||
|
||||
def run(
|
||||
engine: Engine, rule: RuleSnapshot, src: Path, dst: Path | LocalDestination, **kw: object
|
||||
) -> RunResult:
|
||||
d = dst if isinstance(dst, LocalDestination) else LocalDestination(dst)
|
||||
return engine.execute(rule, src, d, CancelToken(), **kw) # type: ignore[arg-type]
|
||||
|
||||
|
||||
def write(path: Path, data: str = "inhalt") -> Path:
|
||||
path.parent.mkdir(parents=True, exist_ok=True)
|
||||
path.write_text(data)
|
||||
return path
|
||||
|
||||
|
||||
def statuses(result: RunResult) -> dict[str, ItemStatus]:
|
||||
return {i.src: i.status for i in result.items}
|
||||
|
||||
|
||||
# -- Fallstrick 1: halb geschriebene Dateien --------------------------------------------
|
||||
|
||||
|
||||
def test_growing_file_is_not_touched(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, hooks: list[Callable[[], None]]
|
||||
) -> None:
|
||||
write(src / "fertig.pdf", "a" * 100)
|
||||
growing = write(src / "wachsend.pdf", "b" * 10)
|
||||
hooks.append(lambda: growing.write_text("b" * 20)) # wächst während des Fensters
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert statuses(result) == {"fertig.pdf": ItemStatus.MOVED, "wachsend.pdf": ItemStatus.SKIPPED}
|
||||
assert growing.read_text() == "b" * 20
|
||||
assert not (dst / "wachsend.pdf").exists()
|
||||
assert (dst / "fertig.pdf").read_text() == "a" * 100
|
||||
|
||||
|
||||
def test_mtime_change_alone_counts_as_unstable(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, hooks: list[Callable[[], None]]
|
||||
) -> None:
|
||||
f = write(src / "scan.pdf")
|
||||
st = f.stat()
|
||||
hooks.append(lambda: os.utime(f, ns=(st.st_atime_ns, st.st_mtime_ns + 1_000_000)))
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert statuses(result) == {"scan.pdf": ItemStatus.SKIPPED}
|
||||
assert f.exists()
|
||||
|
||||
|
||||
def test_stability_window_is_waited(ledger: Ledger, rule_id: int, src: Path, dst: Path) -> None:
|
||||
waited: list[float] = []
|
||||
eng = Engine(ledger, sleep=waited.append)
|
||||
write(src / "a.pdf")
|
||||
run(eng, snap(rule_id, stability_seconds=7), src, dst)
|
||||
assert waited == [7]
|
||||
waited.clear()
|
||||
run(eng, snap(rule_id, stability_seconds=7), src, dst) # nichts mehr da → kein Warten
|
||||
assert waited == []
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"name",
|
||||
[".versteckt.pdf", "scan.pdf.part", "x.tmp", "dl.crdownload", "f.filepart", "t.!ut", "A.PART"],
|
||||
)
|
||||
def test_hidden_and_partial_files_are_ignored(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, name: str
|
||||
) -> None:
|
||||
write(src / name)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.seen == 0
|
||||
assert (src / name).exists()
|
||||
|
||||
|
||||
def test_min_age_and_max_size(engine: Engine, rule_id: int, src: Path, dst: Path) -> None:
|
||||
write(src / "neu.pdf")
|
||||
old = write(src / "alt.pdf")
|
||||
os.utime(old, (time.time() - 3600, time.time() - 3600))
|
||||
write(src / "gross.pdf", "x" * (2 * 1024 * 1024))
|
||||
os.utime(src / "gross.pdf", (time.time() - 3600, time.time() - 3600))
|
||||
result = run(engine, snap(rule_id, min_age_seconds=600, max_size_mb=1), src, dst)
|
||||
assert statuses(result) == {
|
||||
"alt.pdf": ItemStatus.MOVED,
|
||||
"gross.pdf": ItemStatus.SKIPPED,
|
||||
"neu.pdf": ItemStatus.SKIPPED,
|
||||
}
|
||||
|
||||
|
||||
# -- Fallstrick 3: Atomarität ---------------------------------------------------------------
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("cross_device")
|
||||
def test_final_name_never_shows_partial_content(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
data = "x" * (3 * dest_mod.CHUNK + 17)
|
||||
write(src / "gross.pdf", data)
|
||||
seen_during_write: list[list[str]] = []
|
||||
original = LocalDestination._upload
|
||||
|
||||
def spy(self: LocalDestination, *a: object, **kw: object) -> list[str]:
|
||||
notes = original(self, *a, **kw) # type: ignore[arg-type]
|
||||
seen_during_write.append(sorted(p.name for p in dst.iterdir()))
|
||||
return notes
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "_upload", spy)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert statuses(result) == {"gross.pdf": ItemStatus.MOVED}
|
||||
# während/nach dem Schreiben existiert nur die .part-Datei, nicht der Endname
|
||||
(names,) = seen_during_write
|
||||
assert len(names) == 1 and names[0].startswith(".netmover-") and names[0].endswith(".part")
|
||||
assert (dst / "gross.pdf").read_text() == data
|
||||
assert not (src / "gross.pdf").exists()
|
||||
|
||||
|
||||
STAGES = ["_upload", "_apply_meta", "_link_noreplace", "verify"]
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("cross_device")
|
||||
@pytest.mark.parametrize("stage", STAGES)
|
||||
def test_abort_at_any_stage_keeps_source_and_leaves_no_part(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch, stage: str
|
||||
) -> None:
|
||||
write(src / "scan.pdf", "wichtig")
|
||||
if stage == "verify":
|
||||
# Quelle wird zwischen Kopie und Umbenennen verändert
|
||||
original = LocalDestination._apply_meta
|
||||
|
||||
def touch(self: LocalDestination, rel: str, meta: object) -> list[str]:
|
||||
(src / "scan.pdf").write_text("wichtig!!")
|
||||
return original(self, rel, meta) # type: ignore[arg-type]
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "_apply_meta", touch)
|
||||
else:
|
||||
|
||||
def boom(self: object, *a: object, **kw: object) -> None:
|
||||
raise OSError(errno.EIO if stage == "_upload" else errno.ENOSPC, "Fehler")
|
||||
|
||||
monkeypatch.setattr(LocalDestination, stage, boom)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert (src / "scan.pdf").exists()
|
||||
assert list(dst.iterdir()) == []
|
||||
item = result.items[0] if result.items else None
|
||||
if stage == "_upload": # EIO → Anbindung gestört, ganzer Lauf abgebrochen
|
||||
assert result.status is RunStatus.ERROR
|
||||
elif stage == "verify":
|
||||
assert item is not None and item.status is ItemStatus.SKIPPED
|
||||
else:
|
||||
assert item is not None and item.status is ItemStatus.FAILED
|
||||
|
||||
|
||||
@pytest.mark.parametrize("path", ["same-device", "copy"])
|
||||
def test_source_not_deletable_keeps_target_and_prevents_duplicate(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch, path: str
|
||||
) -> None:
|
||||
if path == "copy":
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
write(src / "scan.pdf")
|
||||
real_unlink = os.unlink
|
||||
|
||||
def deny(path: object, *a: object, **kw: object) -> None:
|
||||
if str(path).startswith(str(src)):
|
||||
raise PermissionError(errno.EACCES, "Permission denied", str(path))
|
||||
real_unlink(path) # type: ignore[arg-type]
|
||||
|
||||
monkeypatch.setattr("netmover.mover.engine.os.unlink", deny)
|
||||
first = run(engine, snap(rule_id), src, dst)
|
||||
assert first.items[0].status is ItemStatus.FAILED
|
||||
assert "nicht löschbar" in (first.items[0].message or "")
|
||||
assert (dst / "scan.pdf").exists() and (src / "scan.pdf").exists()
|
||||
second = run(engine, snap(rule_id), src, dst)
|
||||
assert second.items == [] and second.message == "1 bereits übertragen"
|
||||
assert sorted(p.name for p in dst.iterdir()) == ["scan.pdf"]
|
||||
|
||||
|
||||
def test_stale_part_files_are_cleaned(engine: Engine, rule_id: int, src: Path, dst: Path) -> None:
|
||||
old = write(dst / ".netmover-dead.part", "rest")
|
||||
young = write(dst / ".netmover-busy.part", "rest")
|
||||
os.utime(old, (time.time() - 7200, time.time() - 7200))
|
||||
run(engine, snap(rule_id), src, dst)
|
||||
assert not old.exists()
|
||||
assert young.exists()
|
||||
|
||||
|
||||
def test_same_device_move_uses_link_without_copy(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
write(src / "a.pdf", "x")
|
||||
ino = (src / "a.pdf").stat().st_ino
|
||||
|
||||
def no_copy(*a: object, **kw: object) -> None:
|
||||
raise AssertionError("darf auf demselben Gerät nicht kopieren")
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "_upload", no_copy)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert statuses(result) == {"a.pdf": ItemStatus.MOVED}
|
||||
assert (dst / "a.pdf").stat().st_ino == ino
|
||||
assert not (src / "a.pdf").exists()
|
||||
|
||||
|
||||
def test_cross_device_path_is_used_when_devices_differ(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
write(src / "a.pdf", "x")
|
||||
ino = (src / "a.pdf").stat().st_ino
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert statuses(result) == {"a.pdf": ItemStatus.MOVED}
|
||||
assert (dst / "a.pdf").stat().st_ino != ino
|
||||
|
||||
|
||||
def test_filesystem_without_hardlinks_falls_back_to_rename(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
|
||||
def no_link(*a: object, **kw: object) -> None:
|
||||
raise OSError(errno.EPERM, "Operation not permitted")
|
||||
|
||||
monkeypatch.setattr("netmover.mover.destinations.os.link", no_link)
|
||||
write(src / "a.pdf")
|
||||
write(dst / "a.pdf", "alt")
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.items[0].dst == str(dst / "a-1.pdf")
|
||||
assert (dst / "a.pdf").read_text() == "alt"
|
||||
|
||||
|
||||
# -- Fallstrick 4: Namenskollisionen ---------------------------------------------------------
|
||||
|
||||
|
||||
def test_collision_candidates() -> None:
|
||||
gen = collision_candidates("sub/scan001.pdf")
|
||||
assert [next(gen) for _ in range(3)] == [
|
||||
"sub/scan001.pdf",
|
||||
"sub/scan001-1.pdf",
|
||||
"sub/scan001-2.pdf",
|
||||
]
|
||||
gen = collision_candidates("README")
|
||||
assert [next(gen) for _ in range(2)] == ["README", "README-1"]
|
||||
|
||||
|
||||
def test_rename_counts_up_against_filesystem(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path
|
||||
) -> None:
|
||||
write(dst / "scan001.pdf", "alt")
|
||||
for i in range(3):
|
||||
write(src / "scan001.pdf", f"neu{i}")
|
||||
run(engine, snap(rule_id), src, dst)
|
||||
assert sorted(p.name for p in dst.iterdir()) == [
|
||||
"scan001-1.pdf",
|
||||
"scan001-2.pdf",
|
||||
"scan001-3.pdf",
|
||||
"scan001.pdf",
|
||||
]
|
||||
assert (dst / "scan001.pdf").read_text() == "alt"
|
||||
assert (dst / "scan001-3.pdf").read_text() == "neu2"
|
||||
|
||||
|
||||
def test_rename_rechecks_filesystem_each_time(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
"""Zwischen Plan und Umbenennen belegt ein Dritter den Namen – kein Überschreiben."""
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
write(dst / "scan.pdf", "alt")
|
||||
write(src / "scan.pdf", "neu")
|
||||
original = LocalDestination._apply_meta
|
||||
|
||||
def intruder(self: LocalDestination, rel: str, meta: object) -> list[str]:
|
||||
write(dst / "scan-1.pdf", "fremd")
|
||||
return original(self, rel, meta) # type: ignore[arg-type]
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "_apply_meta", intruder)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.items[0].dst == str(dst / "scan-2.pdf")
|
||||
assert (dst / "scan-1.pdf").read_text() == "fremd"
|
||||
assert (dst / "scan-2.pdf").read_text() == "neu"
|
||||
|
||||
|
||||
def test_skip_policy(engine: Engine, rule_id: int, src: Path, dst: Path) -> None:
|
||||
write(dst / "a.pdf", "alt")
|
||||
write(src / "a.pdf", "neu")
|
||||
result = run(engine, snap(rule_id, on_collision=CollisionPolicy.SKIP), src, dst)
|
||||
assert statuses(result) == {"a.pdf": ItemStatus.SKIPPED}
|
||||
assert (dst / "a.pdf").read_text() == "alt"
|
||||
assert (src / "a.pdf").exists()
|
||||
|
||||
|
||||
def test_overwrite_policy(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
write(dst / "a.pdf", "alt")
|
||||
write(src / "a.pdf", "neu")
|
||||
result = run(engine, snap(rule_id, on_collision=CollisionPolicy.OVERWRITE), src, dst)
|
||||
assert statuses(result) == {"a.pdf": ItemStatus.MOVED}
|
||||
assert (dst / "a.pdf").read_text() == "neu"
|
||||
|
||||
|
||||
# -- Fallstrick 11: Kopiermodus mit Gedächtnis ---------------------------------------------
|
||||
|
||||
|
||||
def test_copy_mode_copies_each_version_once(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, session_factory: sessionmaker[Session]
|
||||
) -> None:
|
||||
rule = snap(rule_id, action=RuleAction.COPY)
|
||||
f = write(src / "scan.pdf", "v1")
|
||||
for _ in range(3):
|
||||
run(engine, rule, src, dst)
|
||||
assert sorted(p.name for p in dst.iterdir()) == ["scan.pdf"]
|
||||
assert f.exists()
|
||||
|
||||
# neue Version → läuft durch die Kollisionsregel
|
||||
f.write_text("version 2")
|
||||
os.utime(f, ns=(f.stat().st_atime_ns, f.stat().st_mtime_ns + 5_000_000_000))
|
||||
result = run(engine, rule, src, dst)
|
||||
assert statuses(result) == {"scan.pdf": ItemStatus.COPIED}
|
||||
run(engine, rule, src, dst)
|
||||
assert sorted(p.name for p in dst.iterdir()) == ["scan-1.pdf", "scan.pdf"]
|
||||
assert (dst / "scan-1.pdf").read_text() == "version 2"
|
||||
|
||||
# Quelle verschwindet → Buchführung wird bereinigt
|
||||
f.unlink()
|
||||
run(engine, rule, src, dst)
|
||||
with session_factory() as s:
|
||||
assert s.scalars(select(CopiedFile)).all() == []
|
||||
|
||||
|
||||
def test_copy_mode_preserves_mtime(engine: Engine, rule_id: int, src: Path, dst: Path) -> None:
|
||||
f = write(src / "a.pdf")
|
||||
os.utime(f, ns=(1_600_000_000_000_000_000, 1_600_000_000_123_456_789))
|
||||
run(engine, snap(rule_id, action=RuleAction.COPY), src, dst)
|
||||
assert (dst / "a.pdf").stat().st_mtime_ns == 1_600_000_000_123_456_789
|
||||
|
||||
|
||||
# -- Fallstrick 14: Platz ---------------------------------------------------------------------
|
||||
|
||||
|
||||
def test_insufficient_space_skips_file_without_partial_copy(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
write(src / "a.pdf", "x" * 1000)
|
||||
monkeypatch.setattr(LocalDestination, "free_bytes", lambda self: 1040) # < 1000 * 1.05
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.items[0].status is ItemStatus.FAILED
|
||||
assert "zu wenig Platz" in (result.items[0].message or "")
|
||||
assert list(dst.iterdir()) == []
|
||||
assert (src / "a.pdf").exists()
|
||||
|
||||
|
||||
def test_metadata_failure_is_only_a_hint(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
|
||||
def deny(*a: object, **kw: object) -> None:
|
||||
raise PermissionError(errno.EPERM, "Operation not permitted")
|
||||
|
||||
monkeypatch.setattr("netmover.mover.destinations.os.utime", deny)
|
||||
write(src / "a.pdf")
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.items[0].status is ItemStatus.MOVED
|
||||
assert "Änderungszeit nicht übernommen" in (result.items[0].message or "")
|
||||
|
||||
|
||||
# -- Fallstrick 10: Berechtigungen ---------------------------------------------------------
|
||||
|
||||
|
||||
def test_set_mode_applied(engine: Engine, rule_id: int, src: Path, dst: Path) -> None:
|
||||
write(src / "a.pdf").chmod(0o600)
|
||||
run(engine, snap(rule_id, set_mode=0o640), src, dst)
|
||||
assert (dst / "a.pdf").stat().st_mode & 0o777 == 0o640
|
||||
|
||||
|
||||
def test_unchanged_mode_is_copied_from_source(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
write(src / "a.pdf").chmod(0o604)
|
||||
run(engine, snap(rule_id), src, dst)
|
||||
assert (dst / "a.pdf").stat().st_mode & 0o777 == 0o604
|
||||
|
||||
|
||||
def test_chown_without_permission_is_clear_message(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path
|
||||
) -> None:
|
||||
if os.geteuid() == 0:
|
||||
pytest.skip("als root ist chown erlaubt")
|
||||
write(src / "a.pdf")
|
||||
result = run(engine, snap(rule_id, set_owner=(0, 0)), src, dst)
|
||||
assert result.items[0].status is ItemStatus.MOVED
|
||||
assert "keine Berechtigung" in (result.items[0].message or "")
|
||||
|
||||
|
||||
@pytest.mark.parametrize(("kind", "squash"), [("cifs", None), ("nfs", True), ("nfs", None)])
|
||||
def test_chown_skipped_on_cifs_and_squashed_nfs_warns_once(
|
||||
engine: Engine,
|
||||
rule_id: int,
|
||||
src: Path,
|
||||
dst: Path,
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
kind: str,
|
||||
squash: bool | None,
|
||||
) -> None:
|
||||
called: list[object] = []
|
||||
monkeypatch.setattr("netmover.mover.destinations.os.chown", lambda *a: called.append(a))
|
||||
write(src / "a.pdf")
|
||||
write(src / "b.pdf")
|
||||
d = LocalDestination(dst, kind) # type: ignore[arg-type]
|
||||
result = run(engine, snap(rule_id, set_owner=(1000, 1000), dst_root_squash=squash), src, d)
|
||||
assert called == []
|
||||
msgs = [i.message for i in result.items if i.message and "chown übersprungen" in i.message]
|
||||
assert len(msgs) == 1 # einmalig warnen
|
||||
|
||||
|
||||
def test_chown_on_nfs_without_squash_is_tried(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
called: list[object] = []
|
||||
monkeypatch.setattr("netmover.mover.destinations.os.chown", lambda *a: called.append(a))
|
||||
write(src / "a.pdf")
|
||||
run(
|
||||
engine,
|
||||
snap(rule_id, set_owner=(1000, 100), dst_root_squash=False),
|
||||
src,
|
||||
LocalDestination(dst, "nfs"),
|
||||
)
|
||||
assert called and called[0][1:] == (1000, 100) # type: ignore[index]
|
||||
|
||||
|
||||
# -- Filter, Rekursion, Probelauf, Quarantäne ---------------------------------------------
|
||||
|
||||
|
||||
def test_matcher() -> None:
|
||||
m = Matcher("*.pdf; *.JPG", PatternType.GLOB)
|
||||
assert m("a/Scan.PDF") and m("b.jpg") and not m("c.txt")
|
||||
r = Matcher(r"^rechnungen/.*\.pdf$", PatternType.REGEX)
|
||||
assert r("rechnungen/2026.pdf") and not r("scan/2026.pdf")
|
||||
|
||||
|
||||
def test_recursive_preserves_structure_and_skips_synology_dirs(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path
|
||||
) -> None:
|
||||
write(src / "max" / "a.pdf")
|
||||
write(src / "@eaDir" / "a.pdf" / "SYNOFILE_THUMB_M.jpg")
|
||||
write(src / "#recycle" / "alt.pdf")
|
||||
write(src / "top.pdf")
|
||||
flat = run(engine, snap(rule_id), src, dst)
|
||||
assert statuses(flat) == {"top.pdf": ItemStatus.MOVED}
|
||||
rec = run(engine, snap(rule_id, recursive=True), src, dst)
|
||||
assert statuses(rec) == {"max/a.pdf": ItemStatus.MOVED}
|
||||
assert (dst / "max" / "a.pdf").exists()
|
||||
assert (src / "#recycle" / "alt.pdf").exists()
|
||||
|
||||
|
||||
def test_symlinks_are_ignored(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, tmp_path: Path
|
||||
) -> None:
|
||||
secret = write(tmp_path / "geheim.txt")
|
||||
(src / "link.txt").symlink_to(secret)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.seen == 0 and secret.exists()
|
||||
|
||||
|
||||
def test_dry_run_touches_nothing(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, session_factory: sessionmaker[Session]
|
||||
) -> None:
|
||||
write(src / "a.pdf")
|
||||
write(dst / "a.pdf", "alt")
|
||||
write(src / "b.pdf")
|
||||
result = run(engine, snap(rule_id, action=RuleAction.COPY), src, dst, dry_run=True)
|
||||
assert {i.src: (i.status, i.dst) for i in result.items} == {
|
||||
"a.pdf": (ItemStatus.PLANNED, str(dst / "a-1.pdf")),
|
||||
"b.pdf": (ItemStatus.PLANNED, str(dst / "b.pdf")),
|
||||
}
|
||||
assert sorted(p.name for p in dst.iterdir()) == ["a.pdf"]
|
||||
assert sorted(p.name for p in src.iterdir()) == ["a.pdf", "b.pdf"]
|
||||
with session_factory() as s:
|
||||
assert s.scalars(select(CopiedFile)).all() == []
|
||||
|
||||
|
||||
def test_quarantine_after_repeated_failures(
|
||||
engine: Engine,
|
||||
rule_id: int,
|
||||
src: Path,
|
||||
dst: Path,
|
||||
tmp_path: Path,
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
|
||||
def broken(self: object, *a: object, **kw: object) -> None:
|
||||
raise TransferError("kaputt")
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "_link_noreplace", broken)
|
||||
write(src / "defekt.pdf")
|
||||
quarantine = src / "quarantaene"
|
||||
rule = snap(rule_id, quarantine_after=3, recursive=True)
|
||||
msgs = []
|
||||
for _ in range(3):
|
||||
r = run(engine, rule, src, dst, quarantine_root=quarantine)
|
||||
msgs.append(r.items[0].message)
|
||||
assert msgs[-1] == "kaputt (Versuch 3)"
|
||||
r = run(engine, rule, src, dst, quarantine_root=quarantine)
|
||||
assert statuses(r) == {"defekt.pdf": ItemStatus.QUARANTINED}
|
||||
assert (quarantine / "defekt.pdf").exists()
|
||||
# Quarantäne liegt in der Quelle, wird aber trotz Rekursion nicht erneut gescannt
|
||||
r = run(engine, rule, src, dst, quarantine_root=quarantine)
|
||||
assert r.seen == 0
|
||||
|
||||
|
||||
def test_mount_error_mid_run_aborts_instead_of_failing_every_file(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
monkeypatch.setattr(LocalDestination, "move_same_device", lambda *a, **k: None)
|
||||
|
||||
def stale(self: object, *a: object, **kw: object) -> None:
|
||||
raise OSError(errno.ESTALE, "Stale file handle")
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "_upload", stale)
|
||||
write(src / "a.pdf")
|
||||
write(src / "b.pdf")
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert result.status is RunStatus.ERROR
|
||||
assert "gestört" in (result.message or "")
|
||||
assert (src / "a.pdf").exists() and (src / "b.pdf").exists()
|
||||
|
||||
|
||||
def test_cancel_stops_before_next_file(engine: Engine, rule_id: int, src: Path, dst: Path) -> None:
|
||||
write(src / "a.pdf")
|
||||
token = CancelToken()
|
||||
token.cancel()
|
||||
result = engine.execute(snap(rule_id), src, LocalDestination(dst), token)
|
||||
assert result.status is RunStatus.ERROR
|
||||
assert (src / "a.pdf").exists()
|
||||
|
||||
|
||||
def test_missing_source_dir(engine: Engine, rule_id: int, tmp_path: Path, dst: Path) -> None:
|
||||
result = run(engine, snap(rule_id), tmp_path / "fehlt", dst)
|
||||
assert result.status is RunStatus.ERROR and "existiert nicht" in (result.message or "")
|
||||
|
||||
|
||||
def test_local_destination_must_exist(
|
||||
engine: Engine, rule_id: int, src: Path, tmp_path: Path
|
||||
) -> None:
|
||||
write(src / "a.pdf")
|
||||
result = run(engine, snap(rule_id), src, LocalDestination(tmp_path / "fehlt"))
|
||||
assert result.status is RunStatus.ERROR
|
||||
assert (src / "a.pdf").exists()
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("cross_device")
|
||||
def test_fsync_before_final_rename(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
events: list[str] = []
|
||||
real_fsync, real_link = os.fsync, os.link
|
||||
|
||||
def fsync(fd: int) -> None:
|
||||
events.append("fsync")
|
||||
real_fsync(fd)
|
||||
|
||||
def link(a: object, b: object, *args: object, **kw: object) -> None:
|
||||
events.append(f"link:{Path(str(b)).name}")
|
||||
real_link(a, b) # type: ignore[arg-type]
|
||||
|
||||
monkeypatch.setattr("netmover.mover.destinations.os.fsync", fsync)
|
||||
monkeypatch.setattr("netmover.mover.destinations.os.link", link)
|
||||
write(src / "a.pdf")
|
||||
run(engine, snap(rule_id), src, dst)
|
||||
assert events.index("fsync") < events.index("link:a.pdf")
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("cross_device")
|
||||
def test_name_taken_between_check_and_link_is_never_overwritten(
|
||||
engine: Engine, rule_id: int, src: Path, dst: Path, monkeypatch: pytest.MonkeyPatch
|
||||
) -> None:
|
||||
"""Enges Wettlauf-Fenster: exists() sagt „frei“, link() meldet EEXIST."""
|
||||
write(dst / "scan.pdf", "fremd")
|
||||
write(src / "scan.pdf", "neu")
|
||||
real_exists = LocalDestination.exists
|
||||
lied: list[str] = []
|
||||
|
||||
def stale_exists(self: LocalDestination, rel: str) -> bool:
|
||||
if rel == "scan.pdf" and not lied and self.root == dst:
|
||||
lied.append(rel)
|
||||
return False # veraltete Antwort
|
||||
return real_exists(self, rel)
|
||||
|
||||
monkeypatch.setattr(LocalDestination, "plan", lambda self, rel, policy: rel)
|
||||
monkeypatch.setattr(LocalDestination, "exists", stale_exists)
|
||||
result = run(engine, snap(rule_id), src, dst)
|
||||
assert lied
|
||||
assert result.items[0].dst == str(dst / "scan-1.pdf")
|
||||
assert (dst / "scan.pdf").read_text() == "fremd"
|
||||
assert (dst / "scan-1.pdf").read_text() == "neu"
|
||||
Reference in New Issue
Block a user