tests/testalias.py

   1 #!/usr/bin/env python2.4
   2 from basetest import BaseTest
   3 import sys, tempfile, os, shutil, logging
   4 from StringIO import StringIO
   5 import unittest
   6 from logging import getLogger, DEBUG, INFO
   7
   8 sys.path.insert(0, '..')
   9
  10 from zeroinstall import alias
  11
  12 expected_script = """#!/bin/sh
  13 if [ "$*" = "--versions" ]; then
  14   exec 0launch -gd 'http://example.com/foo.xml' "$@"
  15 else
  16   exec 0launch  'http://example.com/foo.xml' "$@"
  17 fi
  18 """
  19
  20 expected_script_main = """#!/bin/sh
  21 if [ "$*" = "--versions" ]; then
  22   exec 0launch -gd 'http://example.com/foo.xml' "$@"
  23 else
  24   exec 0launch --main 'a'\\'''\\''\\test' 'http://example.com/foo.xml' "$@"
  25 fi
  26 """
  27
  28 class TestAlias(BaseTest):
  29         def setUp(self):
  30                 BaseTest.setUp(self)
  31
  32         def testWrite(self):
  33                 buf = StringIO()
  34                 alias.write_script(buf, 'http://example.com/foo.xml', None)
  35                 self.assertEquals(expected_script, buf.getvalue())
  36
  37                 buf = StringIO()
  38                 alias.write_script(buf, 'http://example.com/foo.xml', 'a\'\'\\test')
  39                 self.assertEquals(expected_script_main, buf.getvalue())
  40
  41         def testParse(self):
  42                 tmp = tempfile.NamedTemporaryFile()
  43                 tmp.write(expected_script)
  44                 tmp.flush()
  45                 tmp.seek(0)
  46                 uri, main = alias.parse_script(tmp.name)
  47                 self.assertEquals('http://example.com/foo.xml', uri)
  48                 self.assertEquals(None, main)
  49
  50                 tmp = tempfile.NamedTemporaryFile()
  51                 tmp.write(expected_script_main)
  52                 tmp.flush()
  53                 tmp.seek(0)
  54                 uri, main = alias.parse_script(tmp.name)
  55                 self.assertEquals('http://example.com/foo.xml', uri)
  56                 self.assertEquals('a\'\'\\test', main)
  57
  58         def testParseException(self):
  59                 tmp = tempfile.NamedTemporaryFile()
  60                 tmp.write('hi' + expected_script)
  61                 tmp.flush()
  62                 tmp.seek(0)
  63                 try:
  64                         alias.parse_script(tmp.name)
  65                         assert False
  66                 except alias.NotAnAliasScript:
  67                         pass
  68
  69 suite = unittest.makeSuite(TestAlias)
  70 if __name__ == '__main__':
  71         sys.argv.append('-v')
  72         unittest.main()