diff options
Diffstat (limited to 'subplot')
-rw-r--r-- | subplot/data.py | 14 | ||||
-rw-r--r-- | subplot/data.yaml | 6 |
2 files changed, 20 insertions, 0 deletions
diff --git a/subplot/data.py b/subplot/data.py index f3faf2b..f7fb903 100644 --- a/subplot/data.py +++ b/subplot/data.py @@ -54,6 +54,20 @@ def _create_manifest_of_directory(ctx, dirname=None, manifest=None): open(manifest, "w").write(stdout) +def file_is_restored(ctx, filename=None, restored=None): + filename = os.path.join(restored, "./" + filename) + exists = os.path.exists(filename) + logging.debug(f"restored? {filename} {exists}") + assert exists + + +def file_is_not_restored(ctx, filename=None, restored=None): + filename = os.path.join(restored, "./" + filename) + exists = os.path.exists(filename) + logging.debug(f"restored? {filename} {exists}") + assert not exists + + def files_match(ctx, first=None, second=None): assert_eq = globals()["assert_eq"] diff --git a/subplot/data.yaml b/subplot/data.yaml index 9538daa..1636e77 100644 --- a/subplot/data.yaml +++ b/subplot/data.yaml @@ -24,3 +24,9 @@ - then: "stdout, as JSON, matches file {filename}" function: match_stdout_to_json_file + +- then: "file {filename} is restored to {restored}" + function: file_is_restored + +- then: "file {filename} is not restored to {restored}" + function: file_is_not_restored |