Add: scenarios for testing sp-filter, dot and roadmap graphs

1 files changed, 12 insertions, 0 deletions

diff --git a/subplot.py b/subplot.py
index bdd6236..fa03e54 100644
--- a/subplot.py
+++ b/subplot.py
@@ -1,4 +1,5 @@
 import os
+import re
 import subprocess
 
 def create_file(ctx, filename=None):
@@ -16,9 +17,20 @@ def run_codegen(ctx, filename=None):
     runcmd(ctx, [codegen, filename, '-o', 'test.py', '--run', '--templates', tmpldir])
     exit_code_zero(ctx)
 
+def run_pandoc_with_filter(ctx, filename=None, output=None):
+    sp_filter = binary('sp-filter')
+    runcmd(ctx, ['pandoc', '--filter', sp_filter, filename, '-o', output])
+    exit_code_zero(ctx)
+
 def file_exists(ctx, filename=None):
     assert_eq(os.path.exists(filename), True)
 
+def file_matches(ctx, filename=None, regex=None):
+    with open(filename) as f:
+        content = f.read()
+    m = re.search(regex, content)
+    assert_eq(bool(m), True)
+
 def scenario_was_run(ctx, name=None):
     stdout_matches(ctx, '\nrunning scenario: {}\n'.format(name))