summaryrefslogtreecommitdiff
path: root/subplot
diff options
context:
space:
mode:
authorLars Wirzenius <liw@liw.fi>2021-02-07 17:46:06 +0200
committerLars Wirzenius <liw@liw.fi>2021-02-07 18:45:57 +0200
commit028fd279652dbe167b12b32dff00c6a299c4fc02 (patch)
tree4328795e0e96a493ef8e71ae29c81fc3e9811914 /subplot
parent3ef5bd7a3ab445509216116bb2f8009ace2b1080 (diff)
downloadobnam2-028fd279652dbe167b12b32dff00c6a299c4fc02.tar.gz
test: add scenario for live data file being unreadable
Diffstat (limited to 'subplot')
-rw-r--r--subplot/data.py14
-rw-r--r--subplot/data.yaml6
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