1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
""" Base class for Egg-based test cases. """
# Standard library imports.
import unittest
import pkg_resources
from os.path import dirname, join
class EggBasedTestCase(unittest.TestCase):
""" Base class for Egg-based test cases. """
###########################################################################
# 'TestCase' interface.
###########################################################################
def setUp(self):
""" Prepares the test fixture before each test method is called. """
# The location of the 'eggs' directory.
self.egg_dir = join(dirname(__file__), 'eggs')
return
def tearDown(self):
""" Called immediately after each test method has been called. """
return
###########################################################################
# Private interface.
###########################################################################
def _add_egg(self, filename, working_set=None):
""" Create and add a distribution from the specified '.egg'. """
if working_set is None:
working_set = pkg_resources.working_set
# The eggs must be in our egg directory!
filename = join(dirname(__file__), 'eggs', filename)
# Create a distribution for the egg.
distributions = pkg_resources.find_distributions(filename)
# Add the distributions to the working set (this makes any Python
# modules in the eggs available for importing).
map(working_set.add, distributions)
return
def _add_eggs_on_path(self, path, working_set=None):
""" Add all eggs found on the path to a working set. """
if working_set is None:
working_set = pkg_resources.working_set
environment = pkg_resources.Environment(path)
# 'find_plugins' identifies those distributions that *could* be added
# to the working set without version conflicts or missing requirements.
distributions, errors = working_set.find_plugins(environment)
if len(errors) > 0:
raise SystemError('Cannot find eggs %s' % errors)
# Add the distributions to the working set (this makes any Python
# modules in the eggs available for importing).
map(working_set.add, distributions)
return
#### EOF ######################################################################
|
