PyORAm

[iotcloud.git] / PyORAM / src / pyoram / tests / test_examples.py

diff --git a/PyORAM/src/pyoram/tests/test_examples.py b/PyORAM/src/pyoram/tests/test_examples.py
new file mode 100644 (file)
index 0000000..9446c40
--- /dev/null
+++ b/PyORAM/src/pyoram/tests/test_examples.py
@@ -0,0 +1,61 @@
+import os
+import glob
+import sys
+import unittest2
+
+thisfile = os.path.abspath(__file__)
+thisdir = os.path.dirname(thisfile)
+topdir = os.path.dirname(
+    os.path.dirname(
+        os.path.dirname(thisdir)))
+exdir = os.path.join(topdir, 'examples')
+examples = glob.glob(os.path.join(exdir,"*.py"))
+
+assert os.path.exists(exdir)
+assert thisfile not in examples
+
+tdict = {}
+for fname in examples:
+    basename = os.path.basename(fname)
+    assert basename.endswith('.py')
+    assert len(basename) >= 3
+    basename = basename[:-3]
+    tname = 'test_'+basename
+    tdict[tname] = fname, basename
+
+assert len(tdict) == len(examples)
+
+assert 'test_encrypted_storage_s3' in tdict
+assert 'test_path_oram_s3' in tdict
+if 'PYORAM_AWS_TEST_BUCKET' not in os.environ:
+    del tdict['test_encrypted_storage_s3']
+    del tdict['test_path_oram_s3']
+assert 'test_encrypted_storage_sftp' in tdict
+assert 'test_path_oram_sftp' in tdict
+assert 'test_path_oram_sftp_setup' in tdict
+assert 'test_path_oram_sftp_test' in tdict
+if 'PYORAM_SSH_TEST_HOST' not in os.environ:
+    del tdict['test_encrypted_storage_sftp']
+    del tdict['test_path_oram_sftp']
+    del tdict['test_path_oram_sftp_setup']
+    del tdict['test_path_oram_sftp_test']
+
+def _execute_example(example_name):
+    filename, basename = tdict[example_name]
+    assert os.path.exists(filename)
+    try:
+        sys.path.insert(0, exdir)
+        m = __import__(basename)
+        m.main()
+    finally:
+        sys.path.remove(exdir)
+
+# this is recognized by nosetests as
+# a dynamic test generator
+def test_generator():
+    for example_name in sorted(tdict):
+        yield _execute_example, example_name
+
+if __name__ == "__main__":
+    for tfunc, tname in test_generator():              # pragma: no cover
+        tfunc(tname)                                   # pragma: no cover