3 import os, sys, re, glob
5 def read_version_py(infname):
7 verstrline = open(infname, "rt").read()
8 except EnvironmentError:
11 VSRE = r"^verstr = ['\"]([^'\"]*)['\"]"
12 mo = re.search(VSRE, verstrline, re.M)
16 version = read_version_py(os.path.join('..', 'src', 'allmydata', '_version.py'))
19 raise AssertionError("We don't know which version we're supposed to be testing.")
21 APPNAME='allmydata-tahoe'
24 execfile(os.path.join('..', 'src', 'allmydata', '_auto_deps.py'), adglobals)
25 install_requires = adglobals['install_requires']
26 test_requires = adglobals.get('test_requires', ['mock'])
28 # setuptools/zetuptoolz looks in __main__.__requires__ for a list of
31 __requires__ = [APPNAME + '==' + version] + install_requires + test_requires
33 print "Requirements: %r" % (__requires__,)
35 eggz = glob.glob(os.path.join('..', 'setuptools-*.egg'))
37 egg = os.path.realpath(eggz[0])
38 print "Inserting egg on sys.path: %r" % (egg,)
39 sys.path.insert(0, egg)
44 for i in xrange(1, len(sys.argv)):
45 if not sys.argv[i].startswith('-'):
46 modulename = sys.argv[i]
49 if modulename is None:
50 raise AssertionError("no test module specified")
52 __import__(modulename)
53 srcfile = sys.modules[modulename].__file__
54 srcdir = os.path.dirname(os.path.realpath(srcfile))
55 for i in modulename.split('.'):
56 srcdir = os.path.dirname(srcdir)
58 if os.path.normcase(srcdir).endswith('.egg'):
59 srcdir = os.path.dirname(srcdir)
60 elif os.path.normcase(os.path.basename(srcdir)) == 'site-packages':
61 srcdir = os.path.dirname(srcdir)
62 if re.search(r'python.+\..+', os.path.normcase(os.path.basename(srcdir))):
63 srcdir = os.path.dirname(srcdir)
64 if os.path.normcase(os.path.basename(srcdir)) == 'lib':
65 srcdir = os.path.dirname(srcdir)
67 rootdir = os.path.normcase(os.path.normpath(srcdir))
68 if os.path.basename(rootdir) == 'src':
69 rootdir = os.path.dirname(rootdir)
71 root_from_cwd = os.path.normcase(os.path.normpath(os.getcwd()))
72 if os.path.basename(root_from_cwd) == 'src':
73 root_from_cwd = os.path.dirname(root_from_cwd)
75 same = (root_from_cwd == rootdir)
78 same = os.path.samefile(root_from_cwd, rootdir)
79 except AttributeError, e:
83 msg = ("We seem to be testing the code at %r\n"
84 "(according to the source filename %r),\n"
85 "but expected to be testing the code at %r.\n"
86 % (rootdir, srcfile, root_from_cwd))
88 root_from_cwdu = os.path.normcase(os.path.normpath(os.getcwdu()))
89 if os.path.basename(root_from_cwdu) == u'src':
90 root_from_cwdu = os.path.dirname(root_from_cwdu)
92 if not isinstance(root_from_cwd, unicode) and root_from_cwd.decode(sys.getfilesystemencoding(), 'replace') != root_from_cwdu:
93 msg += ("However, this may be a false alarm because the current directory path\n"
94 "is not representable in the filesystem encoding. This script needs to be\n"
95 "run from the source directory to be tested, at a non-Unicode path.")
97 msg += "This script needs to be run from the source directory to be tested."
99 raise AssertionError(msg)
101 from twisted.scripts.trial import run