Merge pull request #1031 from priv-kweihmann/move-test-2nd

[RFC/WIP] move tests and avoid packaging them

1 files changed, 94 insertions, 0 deletions

diff --git a/test/performance/lib.py b/test/performance/lib.py
new file mode 100644
index 00000000..86f87757
--- /dev/null
+++ b/test/performance/lib.py
@@ -0,0 +1,94 @@
+"""Contains library functions"""
+import logging
+import os
+import tempfile
+
+from git import (
+    Repo
+)
+from git.db import (
+    GitCmdObjectDB,
+    GitDB
+)
+from test.lib import (
+    TestBase
+)
+from git.util import rmtree
+import os.path as osp
+
+#{ Invariants
+
+k_env_git_repo = "GIT_PYTHON_TEST_GIT_REPO_BASE"
+
+#} END invariants
+
+
+#{ Base Classes
+
+class TestBigRepoR(TestBase):
+
+    """TestCase providing access to readonly 'big' repositories using the following
+    member variables:
+
+    * gitrorepo
+
+     * Read-Only git repository - actually the repo of git itself
+
+    * puregitrorepo
+
+     * As gitrepo, but uses pure python implementation
+    """
+
+    #{ Invariants
+    #} END invariants
+
+    def setUp(self):
+        try:
+            super(TestBigRepoR, self).setUp()
+        except AttributeError:
+            pass
+
+        repo_path = os.environ.get(k_env_git_repo)
+        if repo_path is None:
+            logging.info(
+                ("You can set the %s environment variable to a .git repository of" % k_env_git_repo) +
+                "your choice - defaulting to the gitpython repository")
+            repo_path = osp.dirname(__file__)
+        # end set some repo path
+        self.gitrorepo = Repo(repo_path, odbt=GitCmdObjectDB, search_parent_directories=True)
+        self.puregitrorepo = Repo(repo_path, odbt=GitDB, search_parent_directories=True)
+
+    def tearDown(self):
+        self.gitrorepo.git.clear_cache()
+        self.gitrorepo = None
+        self.puregitrorepo.git.clear_cache()
+        self.puregitrorepo = None
+
+
+class TestBigRepoRW(TestBigRepoR):
+
+    """As above, but provides a big repository that we can write to.
+
+    Provides ``self.gitrwrepo`` and ``self.puregitrwrepo``"""
+
+    def setUp(self):
+        self.gitrwrepo = None
+        try:
+            super(TestBigRepoRW, self).setUp()
+        except AttributeError:
+            pass
+        dirname = tempfile.mktemp()
+        os.mkdir(dirname)
+        self.gitrwrepo = self.gitrorepo.clone(dirname, shared=True, bare=True, odbt=GitCmdObjectDB)
+        self.puregitrwrepo = Repo(dirname, odbt=GitDB)
+
+    def tearDown(self):
+        super(TestBigRepoRW, self).tearDown()
+        if self.gitrwrepo is not None:
+            rmtree(self.gitrwrepo.working_dir)
+            self.gitrwrepo.git.clear_cache()
+        self.gitrwrepo = None
+        self.puregitrwrepo.git.clear_cache()
+        self.puregitrwrepo = None
+
+#} END base classes