3
3
from git .test .lib import *
4
4
import shutil
5
5
import tempfile
6
+ import logging
6
7
7
8
from git .db import (
8
9
GitCmdObjectDB ,
18
19
#} END invariants
19
20
20
21
21
- #{ Utilities
22
- def resolve_or_fail (env_var ):
23
- """:return: resolved environment variable or raise EnvironmentError"""
24
- try :
25
- return os .environ [env_var ]
26
- except KeyError :
27
- raise EnvironmentError ("Please set the %r envrionment variable and retry" % env_var )
28
- # END exception handling
29
-
30
- #} END utilities
31
-
32
-
33
22
#{ Base Classes
34
23
35
24
class TestBigRepoR (TestBase ):
@@ -51,12 +40,19 @@ class TestBigRepoR(TestBase):
51
40
head_sha_50 = '32347c375250fd470973a5d76185cac718955fd5'
52
41
#} END invariants
53
42
54
- @classmethod
55
- def setUp (cls ):
56
- super (TestBigRepoR , cls ).setUp ()
57
- repo_path = resolve_or_fail (k_env_git_repo )
58
- cls .gitrorepo = Repo (repo_path , odbt = GitCmdObjectDB )
59
- cls .puregitrorepo = Repo (repo_path , odbt = GitDB )
43
+ def setUp (self ):
44
+ try :
45
+ super (TestBigRepoR , self ).setUp ()
46
+ except AttributeError :
47
+ pass
48
+
49
+ repo_path = os .environ .get (k_env_git_repo )
50
+ if repo_path is None :
51
+ logging .info ("You can set the %s environment variable to a .git repository of your choice - defaulting to the gitpython repository" , k_env_git_repo )
52
+ repo_path = os .path .dirname (__file__ )
53
+ # end set some repo path
54
+ self .gitrorepo = Repo (repo_path , odbt = GitCmdObjectDB )
55
+ self .puregitrorepo = Repo (repo_path , odbt = GitDB )
60
56
61
57
62
58
class TestBigRepoRW (TestBigRepoR ):
@@ -65,16 +61,17 @@ class TestBigRepoRW(TestBigRepoR):
65
61
66
62
Provides ``self.gitrwrepo`` and ``self.puregitrwrepo``"""
67
63
68
- @classmethod
69
- def setUp (cls ):
70
- super (TestBigRepoRW , cls ).setUp ()
64
+ def setUp (self ):
65
+ try :
66
+ super (TestBigRepoRW , self ).setUp ()
67
+ except AttributeError :
68
+ pass
71
69
dirname = tempfile .mktemp ()
72
70
os .mkdir (dirname )
73
- cls .gitrwrepo = cls .gitrorepo .clone (dirname , shared = True , bare = True , odbt = GitCmdObjectDB )
74
- cls .puregitrwrepo = Repo (dirname , odbt = GitDB )
71
+ self .gitrwrepo = self .gitrorepo .clone (dirname , shared = True , bare = True , odbt = GitCmdObjectDB )
72
+ self .puregitrwrepo = Repo (dirname , odbt = GitDB )
75
73
76
- @classmethod
77
- def tearDownAll (cls ):
78
- shutil .rmtree (cls .gitrwrepo .working_dir )
74
+ def tearDown (self ):
75
+ shutil .rmtree (self .gitrwrepo .working_dir )
79
76
80
77
#} END base classes
0 commit comments