home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / test_pkg.py < prev    next >
Text File  |  2003-12-30  |  7KB  |  260 lines
  1. # Test packages (dotted-name import)
  2.  
  3. import sys, os, tempfile, traceback
  4. from os import mkdir, rmdir, extsep          # Can't test if these fail
  5. del mkdir, rmdir
  6. from test.test_support import verify, verbose, TestFailed
  7.  
  8. # Helpers to create and destroy hierarchies.
  9.  
  10. def mkhier(root, descr):
  11.     if not os.path.isdir(root):
  12.         mkdir(root)
  13.     for name, contents in descr:
  14.         comps = name.split()
  15.         fullname = root
  16.         for c in comps:
  17.             fullname = os.path.join(fullname, c)
  18.         if contents is None:
  19.             mkdir(fullname)
  20.         else:
  21.             if verbose: print "write", fullname
  22.             f = open(fullname, "w")
  23.             f.write(contents)
  24.             if contents and contents[-1] != '\n':
  25.                 f.write('\n')
  26.             f.close()
  27.  
  28. def mkdir(x):
  29.     if verbose: print "mkdir", x
  30.     os.mkdir(x)
  31.  
  32. def cleanout(root):
  33.     names = os.listdir(root)
  34.     for name in names:
  35.         fullname = os.path.join(root, name)
  36.         if os.path.isdir(fullname) and not os.path.islink(fullname):
  37.             cleanout(fullname)
  38.         else:
  39.             os.remove(fullname)
  40.     rmdir(root)
  41.  
  42. def rmdir(x):
  43.     if verbose: print "rmdir", x
  44.     os.rmdir(x)
  45.  
  46. def fixdir(lst):
  47.     try:
  48.         lst.remove('__builtins__')
  49.     except ValueError:
  50.         pass
  51.     return lst
  52.  
  53. # Helper to run a test
  54.  
  55. def runtest(hier, code):
  56.     root = tempfile.mkdtemp()
  57.     mkhier(root, hier)
  58.     savepath = sys.path[:]
  59.     fd, fname = tempfile.mkstemp(text=True)
  60.     os.write(fd, code)
  61.     os.close(fd)
  62.     try:
  63.         sys.path.insert(0, root)
  64.         if verbose: print "sys.path =", sys.path
  65.         try:
  66.             execfile(fname, globals(), {})
  67.         except:
  68.             traceback.print_exc(file=sys.stdout)
  69.     finally:
  70.         sys.path[:] = savepath
  71.         os.unlink(fname)
  72.         try:
  73.             cleanout(root)
  74.         except (os.error, IOError):
  75.             pass
  76.  
  77. # Test descriptions
  78.  
  79. tests = [
  80.     ("t1", [("t1", None), ("t1 __init__"+os.extsep+"py", "")], "import t1"),
  81.  
  82.     ("t2", [
  83.     ("t2", None),
  84.     ("t2 __init__"+os.extsep+"py", "'doc for t2'; print __name__, 'loading'"),
  85.     ("t2 sub", None),
  86.     ("t2 sub __init__"+os.extsep+"py", ""),
  87.     ("t2 sub subsub", None),
  88.     ("t2 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"),
  89.     ],
  90. """
  91. import t2
  92. print t2.__doc__
  93. import t2.sub
  94. import t2.sub.subsub
  95. print t2.__name__, t2.sub.__name__, t2.sub.subsub.__name__
  96. import t2
  97. from t2 import *
  98. print dir()
  99. from t2 import sub
  100. from t2.sub import subsub
  101. from t2.sub.subsub import spam
  102. print sub.__name__, subsub.__name__
  103. print sub.subsub.__name__
  104. print dir()
  105. import t2.sub
  106. import t2.sub.subsub
  107. print t2.__name__, t2.sub.__name__, t2.sub.subsub.__name__
  108. from t2 import *
  109. print dir()
  110. """),
  111.  
  112.     ("t3", [
  113.     ("t3", None),
  114.     ("t3 __init__"+os.extsep+"py", "print __name__, 'loading'"),
  115.     ("t3 sub", None),
  116.     ("t3 sub __init__"+os.extsep+"py", ""),
  117.     ("t3 sub subsub", None),
  118.     ("t3 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"),
  119.     ],
  120. """
  121. import t3.sub.subsub
  122. print t3.__name__, t3.sub.__name__, t3.sub.subsub.__name__
  123. reload(t3)
  124. reload(t3.sub)
  125. reload(t3.sub.subsub)
  126. """),
  127.  
  128.     ("t4", [
  129.     ("t4"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (t4"+os.extsep+"py)'"),
  130.     ("t4", None),
  131.     ("t4 __init__"+os.extsep+"py", "print __name__, 'loading'"),
  132.     ("t4 sub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (sub"+os.extsep+"py)'"),
  133.     ("t4 sub", None),
  134.     ("t4 sub __init__"+os.extsep+"py", ""),
  135.     ("t4 sub subsub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (subsub"+os.extsep+"py)'"),
  136.     ("t4 sub subsub", None),
  137.     ("t4 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"),
  138.     ],
  139. """
  140. from t4.sub.subsub import *
  141. print "t4.sub.subsub.spam =", spam
  142. """),
  143.  
  144.     ("t5", [
  145.     ("t5", None),
  146.     ("t5 __init__"+os.extsep+"py", "import t5.foo"),
  147.     ("t5 string"+os.extsep+"py", "print __name__, 'loading'; spam = 1"),
  148.     ("t5 foo"+os.extsep+"py",
  149.      "print __name__, 'loading'; import string; print string.spam"),
  150.      ],
  151. """
  152. import t5
  153. from t5 import *
  154. print dir()
  155. import t5
  156. print fixdir(dir(t5))
  157. print fixdir(dir(t5.foo))
  158. print fixdir(dir(t5.string))
  159. """),
  160.  
  161.     ("t6", [
  162.     ("t6", None),
  163.     ("t6 __init__"+os.extsep+"py", "__all__ = ['spam', 'ham', 'eggs']"),
  164.     ("t6 spam"+os.extsep+"py", "print __name__, 'loading'"),
  165.     ("t6 ham"+os.extsep+"py", "print __name__, 'loading'"),
  166.     ("t6 eggs"+os.extsep+"py", "print __name__, 'loading'"),
  167.     ],
  168. """
  169. import t6
  170. print fixdir(dir(t6))
  171. from t6 import *
  172. print fixdir(dir(t6))
  173. print dir()
  174. """),
  175.  
  176.     ("t7", [
  177.     ("t7"+os.extsep+"py", "print 'Importing t7"+os.extsep+"py'"),
  178.     ("t7", None),
  179.     ("t7 __init__"+os.extsep+"py", "print __name__, 'loading'"),
  180.     ("t7 sub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (sub"+os.extsep+"py)'"),
  181.     ("t7 sub", None),
  182.     ("t7 sub __init__"+os.extsep+"py", ""),
  183.     ("t7 sub subsub"+os.extsep+"py", "print 'THIS SHOULD NOT BE PRINTED (subsub"+os.extsep+"py)'"),
  184.     ("t7 sub subsub", None),
  185.     ("t7 sub subsub __init__"+os.extsep+"py", "print __name__, 'loading'; spam = 1"),
  186.     ],
  187. """
  188. t7, sub, subsub = None, None, None
  189. import t7 as tas
  190. print fixdir(dir(tas))
  191. verify(not t7)
  192. from t7 import sub as subpar
  193. print fixdir(dir(subpar))
  194. verify(not t7 and not sub)
  195. from t7.sub import subsub as subsubsub
  196. print fixdir(dir(subsubsub))
  197. verify(not t7 and not sub and not subsub)
  198. from t7.sub.subsub import spam as ham
  199. print "t7.sub.subsub.spam =", ham
  200. verify(not t7 and not sub and not subsub)
  201. """),
  202.  
  203. ]
  204.  
  205. nontests = [
  206.     ("x5", [], ("import a" + ".a"*400)),
  207.     ("x6", [], ("import a" + ".a"*499)),
  208.     ("x7", [], ("import a" + ".a"*500)),
  209.     ("x8", [], ("import a" + ".a"*1100)),
  210.     ("x9", [], ("import " + "a"*400)),
  211.     ("x10", [], ("import " + "a"*500)),
  212.     ("x11", [], ("import " + "a"*998)),
  213.     ("x12", [], ("import " + "a"*999)),
  214.     ("x13", [], ("import " + "a"*999)),
  215.     ("x14", [], ("import " + "a"*2000)),
  216. ]
  217.  
  218. """XXX Things to test
  219.  
  220. import package without __init__
  221. import package with __init__
  222. __init__ importing submodule
  223. __init__ importing global module
  224. __init__ defining variables
  225. submodule importing other submodule
  226. submodule importing global module
  227. submodule import submodule via global name
  228. from package import submodule
  229. from package import subpackage
  230. from package import variable (defined in __init__)
  231. from package import * (defined in __init__)
  232. """
  233.  
  234. # Run the tests
  235.  
  236. args = []
  237. if __name__ == '__main__':
  238.     args = sys.argv[1:]
  239.     if args and args[0] == '-q':
  240.         verbose = 0
  241.         del args[0]
  242.  
  243. for name, hier, code in tests:
  244.     if args and name not in args:
  245.         print "skipping test", name
  246.         continue
  247.     print "running test", name
  248.     runtest(hier, code)
  249.  
  250. # Test
  251. import sys
  252. import imp
  253. try:
  254.     import sys.imp
  255. except ImportError:
  256.     # This is what we expect
  257.     pass
  258. else:
  259.     raise TestFailed, "No ImportError exception on 'import sys.imp'"
  260.