From d914a7b9a52adccf59a3ab445875f2bc963aedae Mon Sep 17 00:00:00 2001
From: Campbell Barton <ideasman42@gmail.com>
Date: Wed, 19 Nov 2014 20:35:11 +0100
Subject: [PATCH] add testing for blend-files

---
 tests/blendfile_templates.py | 63 ++++++++++++++++++++++++++++++++++++
 tests/test_cli.py            | 61 ++++++++++++++++++++++++++++++++--
 2 files changed, 122 insertions(+), 2 deletions(-)
 create mode 100644 tests/blendfile_templates.py

diff --git a/tests/blendfile_templates.py b/tests/blendfile_templates.py
new file mode 100644
index 0000000..fdd5936
--- /dev/null
+++ b/tests/blendfile_templates.py
@@ -0,0 +1,63 @@
+# Apache License, Version 2.0
+
+# Pass in:
+#
+#    blender --python blendfile_templates.py -- destination/file.blend create_command return_code
+#
+# 'return_code' can be any number, use to check the script completes.
+
+
+def _clear_blend():
+    import bpy
+
+    # copied from batch_import.py
+    unique_obs = set()
+    for scene in bpy.data.scenes:
+        for obj in scene.objects[:]:
+            scene.objects.unlink(obj)
+            unique_obs.add(obj)
+
+    # remove obdata, for now only worry about the startup scene
+    for bpy_data_iter in (bpy.data.objects, bpy.data.meshes, bpy.data.lamps, bpy.data.cameras):
+        for id_data in bpy_data_iter:
+            bpy_data_iter.remove(id_data)
+
+
+def create_blank():
+    pass
+
+
+def create_image_single():
+    import bpy
+    image = bpy.data.images.new(name="MyImage", width=512, height=512)
+    image.filepath_raw = "//my_image.png"
+    image.file_format = 'PNG'
+    image.use_fake_user = True
+    image.save()
+
+
+if __name__ == "__main__":
+    import sys
+    blendfile, create_id, returncode = sys.argv[-3:]
+    returncode = int(returncode)
+    create_fn = globals()[create_id]
+
+    # ----
+    import bpy
+    # no need for blend1's
+    bpy.context.user_preferences.filepaths.save_version = 0
+    # WEAK! (but needed)
+    # save the file first, so we can do real, relative dirs
+    bpy.ops.wm.save_mainfile('EXEC_DEFAULT', filepath=blendfile)
+    del bpy
+    # ----
+
+    _clear_blend()
+    create_fn()
+
+    import bpy
+    bpy.ops.wm.save_mainfile('EXEC_DEFAULT', filepath=blendfile)
+
+    import sys
+    sys.exit(returncode)
+
diff --git a/tests/test_cli.py b/tests/test_cli.py
index 4e38579..8c72a33 100755
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -13,7 +13,7 @@ Run a single test:
    python3 -m unittest test_cli.BamCommitTest.test_checkout
 """
 
-VERBOSE = 0
+VERBOSE = 1
 
 # ------------------
 # Ensure module path
@@ -97,6 +97,9 @@ TEMP_SERVER = "/tmp/bam_test_server"
 PORT = 5555
 PROJECT_NAME = "test_project"
 
+# running scripts next to this one!
+CURRENT_DIR = os.path.dirname(__file__)
+
 
 def run(cmd, cwd=None):
     if VERBOSE:
@@ -111,12 +114,14 @@ def run(cmd, cwd=None):
 
     proc = subprocess.Popen(cmd, **kwargs)
     stdout, stderr = proc.communicate()
+    returncode = proc.returncode
 
     if VERBOSE:
         sys.stdout.write("   stdout:  %s\n" % stdout.strip())
         sys.stdout.write("   stderr:  %s\n" % stderr.strip())
+        sys.stdout.write("   return:  %d\n" % returncode)
 
-    return stdout, stderr
+    return stdout, stderr, returncode
 
 
 class CHDir:
@@ -489,6 +494,58 @@ class BamCommitTest(BamSessionTestCase):
         self.assertEqual(file_data, file_data_test)
 
 
+class BamBlendTest(BamSessionTestCase):
+
+    def __init__(self, *args):
+        self.init_defaults()
+        super().__init__(*args)
+
+    @staticmethod
+    def create_blend_id(blendfile, create_id, returncode_test):
+        os.makedirs(os.path.dirname(blendfile), exist_ok=True)
+        stdout, stderr, returncode = run(
+                ("blender",
+                 "--background",
+                 "--factory-startup",
+                 "-noaudio",
+                 "--python",
+                 os.path.join(CURRENT_DIR, "blendfile_templates.py"),
+                 "--",
+                 blendfile,
+                 create_id,
+                 str(returncode_test),
+                 ))
+        print("  stdout=%s" % stdout.decode())
+        print("  stderr=%s" % stderr.decode())
+        return stdout, stderr, returncode
+
+    def test_create_all(self):
+        """ This simply tests all the create functions run without error.
+        """
+        import blendfile_templates
+        returncode_test = 42
+        for create_id, create_fn in blendfile_templates.__dict__.items():
+            if create_id.startswith("create_"):
+                if create_fn.__class__.__name__ == "function":
+                    blendfile = os.path.join(TEMP, create_id + ".blend")
+
+                    stdout, stderr, returncode = self.create_blend_id(blendfile, create_id, returncode_test)
+
+                    self.assertEqual(b'', stderr)
+                    self.assertEqual(True, os.path.exists(blendfile))
+                    self.assertEqual(returncode, returncode_test)
+                    with open(blendfile, 'rb') as blendfile_handle:
+                        self.assertEqual(b'BLENDER', blendfile_handle.read(7))
+                    os.remove(blendfile)
+
+    def test_empty(self):
+        blendfile = os.path.join(TEMP, "test.blend")
+        returncode_test = 13
+        stdout, stderr, returncode = self.create_blend_id(blendfile, "create_blank", returncode_test)
+        self.assertEqual(True, os.path.exists(blendfile))
+        self.assertEqual(returncode_test, returncode)
+
+
 if __name__ == '__main__':
     data = global_setup()
     unittest.main(exit=False)