Little sanity pass over the test_cli.py file

No longer importing, then deleting, then re-importing modules all over the
place. Just import at the top of the file.

Also removed some superfluous return statements.
This commit is contained in:
2017-04-21 11:44:43 +02:00
parent a926dfa4c1
commit 0304489db0

View File

@@ -13,7 +13,13 @@ Run a single test:
python3 -m unittest test_cli.BamCheckoutTest.test_checkout
"""
import argparse
import json
import os
import shutil
import sys
import unittest
VERBOSE = os.environ.get("VERBOSE", False)
if VERBOSE == "0":
VERBOSE = None
@@ -24,23 +30,19 @@ if VERBOSE:
# ------------------
# Ensure module path
import os
import sys
path = os.path.normpath(os.path.join(os.path.dirname(os.path.abspath(__file__)), ".."))
if path not in sys.path:
sys.path.append(path)
del os, sys, path
del path
# --------
# ------------------
# Ensure module path
import os
import sys
path = os.path.normpath(os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "webservice", "bam"))
if path not in sys.path:
sys.path.append(path)
del os, sys, path
del path
# --------
@@ -64,8 +66,6 @@ if not VERBOSE:
# So override `sys.exit` with one that prints to the original
# stdout/stderr on exit.
if 1:
import sys
def exit(status):
import io
globals().update(sys.exit.exit_data)
@@ -91,7 +91,6 @@ if 1:
}
sys.exit = exit
del exit
del sys
# --------
# --------------------------------------------
@@ -103,7 +102,6 @@ if 1:
# This monkey-patches in an exist function which simply raises an exception.
# We could do something a bit nicer here,
# but for now just use a basic exception.
import argparse
def argparse_fake_exit(self, status, message):
@@ -112,18 +110,12 @@ def argparse_fake_exit(self, status, message):
argparse.ArgumentParser.exit = argparse_fake_exit
del argparse_fake_exit
del argparse
# --------
# ----------------------------------------------------------------------------
# Real beginning of code!
import os
import sys
import shutil
import json
TEMP_LOCAL = "/tmp/bam_test"
# Separate tmp directory for server, since we don't reset the server at every test
TEMP_SERVER = "/tmp/bam_test_server"
@@ -261,25 +253,21 @@ def bam_run_as_json(argv, cwd=None):
stdout, stderr = bam_run(argv, cwd=cwd)
if stderr:
raise Exception(stderr)
return None
ret = None
import json
try:
ret = json.loads(stdout)
return json.loads(stdout)
except Exception as e:
print("---- JSON BEGIN (invalid) ----")
print(stdout)
print("---- JSON END ----")
raise e
return ret
def file_quick_write(path, filepart=None, data=None, append=False):
"""
Quick file creation utility.
"""
if data is None:
data = b''
@@ -579,8 +567,6 @@ def global_teardown(data, use_server=True):
# ------------------------------------------------------------------------------
# Unit Tests
import unittest
class BamSimpleTestCase(unittest.TestCase):
"""
@@ -882,7 +868,6 @@ class BamCheckoutTest(BamSessionTestCase):
variation_path = os.path.join(session_path, "variations")
if 1:
import shutil
# path cant already exist, ugh
shutil.copytree(
os.path.join(CURRENT_DIR, "blends", "variations"),
@@ -1003,7 +988,6 @@ class BamCheckoutTest(BamSessionTestCase):
blendfile, images)
# we are going to remove the maps directory, getting bam to handle a missing path
import shutil
shutil.rmtree(os.path.join(session_path, "maps"))
# commit and checkout
@@ -1121,7 +1105,6 @@ class BamRevertTest(BamSessionTestCase):
os.makedirs(os.path.join(session_path, "dir"))
# just a way to quickly get a lot of files.
import shutil
for d in ("abs", "subdir"):
# path cant already exist, ugh
shutil.copytree(
@@ -1520,7 +1503,6 @@ class BamRelativeAbsoluteTest(BamSessionTestCase):
proj_path, session_path = self.init_session(session_name)
if 1:
import shutil
for d in ("abs", "subdir"):
# path cant already exist, ugh
shutil.copytree(
@@ -1537,7 +1519,6 @@ class BamRelativeAbsoluteTest(BamSessionTestCase):
Same as test_absolute_relative_from_blendfiles(),
but start from a single file commit
"""
import shutil
session_name = "mysession"
proj_path, session_path = self.init_session(session_name)
@@ -1551,7 +1532,6 @@ class BamRelativeAbsoluteTest(BamSessionTestCase):
if not blendfile_template_create(blendfile_abs, session_path, "create_blank", None, []):
self.fail("blend file couldn't be created")
return
self.assertTrue(os.path.exists(blendfile_abs))
stdout, stderr = bam_run(["commit", "-m", "test message"], session_path)
@@ -1566,7 +1546,6 @@ class BamRelativeAbsoluteTest(BamSessionTestCase):
# they must now map back to the correct paths.
if 1:
import shutil
shutil.copytree(
os.path.join(CURRENT_DIR, "blends", "multi_level", "abs", "path"),
os.path.join(session_path, "_abs", "path"),
@@ -1615,7 +1594,6 @@ class BamRelativeAbsoluteTest(BamSessionTestCase):
blendfile = os.path.join("root", "level1", "level2", "level3", "level3.blend")
import shutil
shutil.copytree(
os.path.join(CURRENT_DIR, "blends", "multi_level_link"),
os.path.join(session_path, "root"),
@@ -1673,7 +1651,6 @@ class BamIgnoreTest(BamSessionTestCase):
# create some files
file_quick_write(session_path, file_name, file_data)
import os
subdir_path = os.path.join(session_path, "subdirectory")
os.makedirs(subdir_path)
file_quick_write(subdir_path, "testfile.blend1", file_data)