← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_lazy_import.py

  • Committer: John Arbash Meinel
  • Date: 2006-09-10 20:59:01 UTC
  • mto: This revision was merged to the branch mainline in revision 2004.
  • Revision ID: john@arbash-meinel.com-20060910205901-ceb5929c1497f81f
start working on some lazy importing code

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_lazy_import.py
16
16
 
17
17
"""Test the lazy_import functionality."""
18
18
 
 
19
import os
 
20
import sys
19
21
 
20
22
from bzrlib import (
21
23
    lazy_import,
 
24
    osutils,
22
25
    )
23
26
from bzrlib.tests import TestCase, TestCaseInTempDir
24
27
 
199
202
                          'func',
200
203
                         ], actions)
201
204
 
 
205
 
202
206
class TestImportReplacer(TestCaseInTempDir):
203
207
    """Test the ability to have a lazily imported module or object"""
204
208
 
 
209
    def setUp(self):
 
210
        TestCaseInTempDir.setUp(self)
 
211
        self.create_modules()
 
212
        base_path = self.test_dir + '/base'
 
213
 
 
214
        sys.path.append(base_path)
 
215
        def cleanup():
 
216
            sys.path.remove(base_path)
 
217
 
 
218
    def create_modules(self):
 
219
        """Create some random modules to be imported.
 
220
 
 
221
        Each entry has a random suffix, and the full names are saved
 
222
 
 
223
        These are setup as follows:
 
224
         base/ <= used to ensure not in default search path
 
225
            root-XXX/
 
226
                __init__.py <= This will contain var1, func1
 
227
                mod-XXX.py <= This will contain var2, func2
 
228
                sub-XXX/
 
229
                    __init__.py <= Contains var3, func3
 
230
                    submod-XXX.py <= contains var4, func4
 
231
        """
 
232
        rand_suffix = osutils.rand_chars(4)
 
233
        root_name = 'root_' + rand_suffix
 
234
        mod_name = 'mod_' + rand_suffix
 
235
        sub_name = 'sub_' + rand_suffix
 
236
        submod_name = 'submod_' + rand_suffix
 
237
        os.mkdir('base')
 
238
        root_path = osutils.pathjoin('base', root_name)
 
239
        os.mkdir(root_path)
 
240
        root_init = osutils.pathjoin(root_path, '__init__.py')
 
241
        f = open(osutils.pathjoin(root_path, '__init__.py'), 'wb')
 
242
        try:
 
243
            f.write('var1 = 1\ndef func1(a):\n  return a\n')
 
244
        finally:
 
245
            f.close()
 
246
        mod_path = osutils.pathjoin(root_path, mod_name + '.py')
 
247
        f = open(mod_path, 'wb')
 
248
        try:
 
249
            f.write('var2 = 2\ndef func2(a):\n  return a\n')
 
250
        finally:
 
251
            f.close()
 
252
 
 
253
        sub_path = osutils.pathjoin(root_path, sub_name)
 
254
        os.mkdir(sub_path)
 
255
        f = open(osutils.pathjoin(sub_path, '__init__.py'), 'wb')
 
256
        try:
 
257
            f.write('var3 = 3\ndef func3(a):\n  return a\n')
 
258
        finally:
 
259
            f.close()
 
260
        submod_path = osutils.pathjoin(sub_path, submod_name + '.py')
 
261
        f = open(submod_path, 'wb')
 
262
        try:
 
263
            f.write('var4 = 4\ndef func4(a):\n  return a\n')
 
264
        finally:
 
265
            f.close()
 
266
        self.root_name = root_name
 
267
        self.mod_name = mod_name
 
268
        self.sub_name = sub_name
 
269
        self.submod_name = submod_name
 
270
 
 
271
    def test_basic_import(self):
 
272
        root = __import__('.'.join([self.root_name, self.sub_name,
 
273
                                    self.submod_name]),
 
274
                          globals(), locals(), [])
 
275
        self.assertEqual(1, root.var1)
 
276
        self.assertEqual(3, getattr(root, self.sub_name).var3)
 
277
        self.assertEqual(4, getattr(getattr(root, self.sub_name),
 
278
                                    self.submod_name).var4)