Lib/test/test_uu.py

   1 """
   2 Tests for uu module.
   3 Nick Mathewson
   4 """
   5
   6 from test_support import verify, TestFailed, verbose, TESTFN
   7 import sys, os
   8 import uu
   9 from StringIO import StringIO
  10
  11 teststr = "The smooth-scaled python crept over the sleeping dog\n"
  12 expected = """\
  13 M5&AE('-M;V]T:\"US8V%L960@<'ET:&]N(&-R97!T(&]V97(@=&AE('-L965P
  14 (:6YG(&1O9PH """
  15 encoded1 = "begin 666 t1\n"+expected+"\n \nend\n"
  16 if verbose:
  17     print '1. encode file->file'
  18 inp = StringIO(teststr)
  19 out = StringIO()
  20 uu.encode(inp, out, "t1")
  21 verify(out.getvalue() == encoded1)
  22 inp = StringIO(teststr)
  23 out = StringIO()
  24 uu.encode(inp, out, "t1", 0644)
  25 verify(out.getvalue() == "begin 644 t1\n"+expected+"\n \nend\n")
  26
  27 if verbose:
  28     print '2. decode file->file'
  29 inp = StringIO(encoded1)
  30 out = StringIO()
  31 uu.decode(inp, out)
  32 verify(out.getvalue() == teststr)
  33 inp = StringIO("""UUencoded files may contain many lines,
  34                   even some that have 'begin' in them.\n"""+encoded1)
  35 out = StringIO()
  36 uu.decode(inp, out)
  37 verify(out.getvalue() == teststr)
  38
  39 stdinsave = sys.stdin
  40 stdoutsave = sys.stdout
  41 try:
  42     if verbose:
  43         print '3. encode stdin->stdout'
  44     sys.stdin = StringIO(teststr)
  45     sys.stdout = StringIO()
  46     uu.encode("-", "-", "t1", 0666)
  47     verify(sys.stdout.getvalue() == encoded1)
  48     if verbose:
  49         print >>stdoutsave, '4. decode stdin->stdout'
  50     sys.stdin = StringIO(encoded1)
  51     sys.stdout = StringIO()
  52     uu.decode("-", "-")
  53     verify(sys.stdout.getvalue() == teststr)
  54 finally:
  55     sys.stdin = stdinsave
  56     sys.stdout = stdoutsave
  57
  58 if verbose:
  59     print '5. encode file->file'
  60 tmpIn  = TESTFN + "i"
  61 tmpOut = TESTFN + "o"
  62 try:
  63     fin = open(tmpIn, 'wb')
  64     fin.write(teststr)
  65     fin.close()
  66
  67     fin = open(tmpIn, 'rb')
  68     fout = open(tmpOut, 'w')
  69     uu.encode(fin, fout, tmpIn, mode=0644)
  70     fin.close()
  71     fout.close()
  72
  73     fout = open(tmpOut, 'r')
  74     s = fout.read()
  75     fout.close()
  76     verify(s == 'begin 644 ' + tmpIn + '\n' + expected + '\n \nend\n')
  77
  78     os.unlink(tmpIn)
  79     if verbose:
  80         print '6. decode file-> file'
  81     uu.decode(tmpOut)
  82     fin = open(tmpIn, 'rb')
  83     s = fin.read()
  84     fin.close()
  85     verify(s == teststr)
  86     # XXX is there an xp way to verify the mode?
  87
  88 finally:
  89     try:
  90         fin.close()
  91     except:
  92         pass
  93     try:
  94         fout.close()
  95     except:
  96         pass
  97     try:
  98         os.unlink(tmpIn)
  99     except:
 100         pass
 101     try:
 102         os.unlink(tmpOut)
 103     except:
 104         pass
 105
 106 if verbose:
 107     print '7. error: truncated input'
 108 inp = StringIO("begin 644 t1\n"+expected)
 109 out = StringIO()
 110 try:
 111     uu.decode(inp, out)
 112     raise TestFailed("No exception thrown")
 113 except uu.Error, e:
 114     verify(str(e) == 'Truncated input file')
 115
 116 if verbose:
 117     print '8. error: missing begin'
 118 inp = StringIO("")
 119 out = StringIO()
 120 try:
 121     uu.decode(inp, out)
 122     raise TestFailed("No exception thrown")
 123 except uu.Error, e:
 124     verify(str(e) == 'No valid begin line found in input file')
 125
 126 # Test to verify that decode() will refuse to overwrite an existing file
 127 import tempfile
 128 outfile = tempfile.mktemp()
 129 inp = StringIO('Here is a message to be uuencoded')
 130 out = StringIO()
 131 uu.encode(inp, out, outfile)
 132 out.seek(0)
 133 try:
 134     if verbose:
 135         print '9. decode w/file not exists is okay'
 136     uu.decode(out)
 137     if not os.path.exists(outfile):
 138         raise TestFailed('uudecode w/ out_file=None failed')
 139     fp = open(outfile)
 140     data = fp.read()
 141     fp.close()
 142     if data <> inp.getvalue():
 143         raise TestFailed('uudecode stored something weird')
 144     # Try to write it again, which should cause a failure
 145     if verbose:
 146         print '10. uudecode w/file exists fails'
 147     out.seek(0)
 148     try:
 149         uu.decode(out)
 150     except uu.Error:
 151         pass
 152     else:
 153         raise TestFailed('expected to get a "file exists" error')
 154 finally:
 155     try:
 156         os.unlink(outfile)
 157     except OSError:
 158         pass