Lib/test/test_shlex.py

   1 # -*- coding: iso-8859-1 -*-
   2 import unittest
   3 import os, sys
   4 import shlex
   5
   6 from test import test_support
   7
   8 try:
   9     from cStringIO import StringIO
  10 except ImportError:
  11     from StringIO import StringIO
  12
  13
  14 # The original test data set was from shellwords, by Hartmut Goebel.
  15
  16 data = r"""x|x|
  17 foo bar|foo|bar|
  18  foo bar|foo|bar|
  19  foo bar |foo|bar|
  20 foo   bar    bla     fasel|foo|bar|bla|fasel|
  21 x y  z              xxxx|x|y|z|xxxx|
  22 \x bar|\|x|bar|
  23 \ x bar|\|x|bar|
  24 \ bar|\|bar|
  25 foo \x bar|foo|\|x|bar|
  26 foo \ x bar|foo|\|x|bar|
  27 foo \ bar|foo|\|bar|
  28 foo "bar" bla|foo|"bar"|bla|
  29 "foo" "bar" "bla"|"foo"|"bar"|"bla"|
  30 "foo" bar "bla"|"foo"|bar|"bla"|
  31 "foo" bar bla|"foo"|bar|bla|
  32 foo 'bar' bla|foo|'bar'|bla|
  33 'foo' 'bar' 'bla'|'foo'|'bar'|'bla'|
  34 'foo' bar 'bla'|'foo'|bar|'bla'|
  35 'foo' bar bla|'foo'|bar|bla|
  36 blurb foo"bar"bar"fasel" baz|blurb|foo"bar"bar"fasel"|baz|
  37 blurb foo'bar'bar'fasel' baz|blurb|foo'bar'bar'fasel'|baz|
  38 ""|""|
  39 ''|''|
  40 foo "" bar|foo|""|bar|
  41 foo '' bar|foo|''|bar|
  42 foo "" "" "" bar|foo|""|""|""|bar|
  43 foo '' '' '' bar|foo|''|''|''|bar|
  44 \""|\|""|
  45 "\"|"\"|
  46 "foo\ bar"|"foo\ bar"|
  47 "foo\\ bar"|"foo\\ bar"|
  48 "foo\\ bar\"|"foo\\ bar\"|
  49 "foo\\" bar\""|"foo\\"|bar|\|""|
  50 "foo\\ bar\" dfadf"|"foo\\ bar\"|dfadf"|
  51 "foo\\\ bar\" dfadf"|"foo\\\ bar\"|dfadf"|
  52 "foo\\\x bar\" dfadf"|"foo\\\x bar\"|dfadf"|
  53 "foo\x bar\" dfadf"|"foo\x bar\"|dfadf"|
  54 \''|\|''|
  55 'foo\ bar'|'foo\ bar'|
  56 'foo\\ bar'|'foo\\ bar'|
  57 "foo\\\x bar\" df'a\ 'df'|"foo\\\x bar\"|df'a|\|'df'|
  58 \"foo"|\|"foo"|
  59 \"foo"\x|\|"foo"|\|x|
  60 "foo\x"|"foo\x"|
  61 "foo\ "|"foo\ "|
  62 foo\ xx|foo|\|xx|
  63 foo\ x\x|foo|\|x|\|x|
  64 foo\ x\x\""|foo|\|x|\|x|\|""|
  65 "foo\ x\x"|"foo\ x\x"|
  66 "foo\ x\x\\"|"foo\ x\x\\"|
  67 "foo\ x\x\\""foobar"|"foo\ x\x\\"|"foobar"|
  68 "foo\ x\x\\"\''"foobar"|"foo\ x\x\\"|\|''|"foobar"|
  69 "foo\ x\x\\"\'"fo'obar"|"foo\ x\x\\"|\|'"fo'|obar"|
  70 "foo\ x\x\\"\'"fo'obar" 'don'\''t'|"foo\ x\x\\"|\|'"fo'|obar"|'don'|\|''|t'|
  71 'foo\ bar'|'foo\ bar'|
  72 'foo\\ bar'|'foo\\ bar'|
  73 foo\ bar|foo|\|bar|
  74 foo#bar\nbaz|foobaz|
  75 :-) ;-)|:|-|)|;|-|)|
  76 áéíóú|á|é|í|ó|ú|
  77 """
  78
  79 posix_data = r"""x|x|
  80 foo bar|foo|bar|
  81  foo bar|foo|bar|
  82  foo bar |foo|bar|
  83 foo   bar    bla     fasel|foo|bar|bla|fasel|
  84 x y  z              xxxx|x|y|z|xxxx|
  85 \x bar|x|bar|
  86 \ x bar| x|bar|
  87 \ bar| bar|
  88 foo \x bar|foo|x|bar|
  89 foo \ x bar|foo| x|bar|
  90 foo \ bar|foo| bar|
  91 foo "bar" bla|foo|bar|bla|
  92 "foo" "bar" "bla"|foo|bar|bla|
  93 "foo" bar "bla"|foo|bar|bla|
  94 "foo" bar bla|foo|bar|bla|
  95 foo 'bar' bla|foo|bar|bla|
  96 'foo' 'bar' 'bla'|foo|bar|bla|
  97 'foo' bar 'bla'|foo|bar|bla|
  98 'foo' bar bla|foo|bar|bla|
  99 blurb foo"bar"bar"fasel" baz|blurb|foobarbarfasel|baz|
 100 blurb foo'bar'bar'fasel' baz|blurb|foobarbarfasel|baz|
 101 ""||
 102 ''||
 103 foo "" bar|foo||bar|
 104 foo '' bar|foo||bar|
 105 foo "" "" "" bar|foo||||bar|
 106 foo '' '' '' bar|foo||||bar|
 107 \"|"|
 108 "\""|"|
 109 "foo\ bar"|foo\ bar|
 110 "foo\\ bar"|foo\ bar|
 111 "foo\\ bar\""|foo\ bar"|
 112 "foo\\" bar\"|foo\|bar"|
 113 "foo\\ bar\" dfadf"|foo\ bar" dfadf|
 114 "foo\\\ bar\" dfadf"|foo\\ bar" dfadf|
 115 "foo\\\x bar\" dfadf"|foo\\x bar" dfadf|
 116 "foo\x bar\" dfadf"|foo\x bar" dfadf|
 117 \'|'|
 118 'foo\ bar'|foo\ bar|
 119 'foo\\ bar'|foo\\ bar|
 120 "foo\\\x bar\" df'a\ 'df"|foo\\x bar" df'a\ 'df|
 121 \"foo|"foo|
 122 \"foo\x|"foox|
 123 "foo\x"|foo\x|
 124 "foo\ "|foo\ |
 125 foo\ xx|foo xx|
 126 foo\ x\x|foo xx|
 127 foo\ x\x\"|foo xx"|
 128 "foo\ x\x"|foo\ x\x|
 129 "foo\ x\x\\"|foo\ x\x\|
 130 "foo\ x\x\\""foobar"|foo\ x\x\foobar|
 131 "foo\ x\x\\"\'"foobar"|foo\ x\x\'foobar|
 132 "foo\ x\x\\"\'"fo'obar"|foo\ x\x\'fo'obar|
 133 "foo\ x\x\\"\'"fo'obar" 'don'\''t'|foo\ x\x\'fo'obar|don't|
 134 "foo\ x\x\\"\'"fo'obar" 'don'\''t' \\|foo\ x\x\'fo'obar|don't|\|
 135 'foo\ bar'|foo\ bar|
 136 'foo\\ bar'|foo\\ bar|
 137 foo\ bar|foo bar|
 138 foo#bar\nbaz|foo|baz|
 139 :-) ;-)|:-)|;-)|
 140 áéíóú|áéíóú|
 141 """
 142
 143 class ShlexTest(unittest.TestCase):
 144     def setUp(self):
 145         self.data = [x.split("|")[:-1]
 146                      for x in data.splitlines()]
 147         self.posix_data = [x.split("|")[:-1]
 148                            for x in posix_data.splitlines()]
 149         for item in self.data:
 150             item[0] = item[0].replace(r"\n", "\n")
 151         for item in self.posix_data:
 152             item[0] = item[0].replace(r"\n", "\n")
 153
 154     def splitTest(self, data, comments):
 155         for i in range(len(data)):
 156             l = shlex.split(data[i][0], comments=comments)
 157             self.assertEqual(l, data[i][1:],
 158                              "%s: %s != %s" %
 159                              (data[i][0], l, data[i][1:]))
 160
 161     def oldSplit(self, s):
 162         ret = []
 163         lex = shlex.shlex(StringIO(s))
 164         tok = lex.get_token()
 165         while tok:
 166             ret.append(tok)
 167             tok = lex.get_token()
 168         return ret
 169
 170     def testSplitPosix(self):
 171         """Test data splitting with posix parser"""
 172         self.splitTest(self.posix_data, comments=True)
 173
 174     def testCompat(self):
 175         """Test compatibility interface"""
 176         for i in range(len(self.data)):
 177             l = self.oldSplit(self.data[i][0])
 178             self.assertEqual(l, self.data[i][1:],
 179                              "%s: %s != %s" %
 180                              (self.data[i][0], l, self.data[i][1:]))
 181
 182 # Allow this test to be used with old shlex.py
 183 if not getattr(shlex, "split", None):
 184     for methname in dir(ShlexTest):
 185         if methname.startswith("test") and methname != "testCompat":
 186             delattr(ShlexTest, methname)
 187
 188 def test_main():
 189     test_support.run_unittest(ShlexTest)
 190
 191 if __name__ == "__main__":
 192     test_main()