native_client_sdk/src/tools/tests/oshelpers_test.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 import os
   7 import shutil
   8 import subprocess
   9 import sys
  10 import tempfile
  11 import unittest
  12 import zipfile
  13
  14 SCRIPT_DIR = os.path.dirname(os.path.abspath(__file__))
  15 PARENT_DIR = os.path.dirname(SCRIPT_DIR)
  16
  17 sys.path.append(PARENT_DIR)
  18
  19 import oshelpers
  20
  21 class RunError(subprocess.CalledProcessError):
  22   def __init__(self, retcode, command, output, error_output):
  23     subprocess.CalledProcessError.__init__(self, retcode, command)
  24     self.output = output
  25     self.error_output = error_output
  26
  27   def __str__(self):
  28     msg = subprocess.CalledProcessError.__str__(self)
  29     msg += '.\nstdout: """%s"""' % (self.output,)
  30     msg += '.\nstderr: """%s"""' % (self.error_output,)
  31     return msg
  32
  33
  34 def RunCmd(cmd, args, cwd, env=None):
  35   env = env or os.environ
  36   command = [sys.executable, 'oshelpers.py', cmd] + args
  37   process = subprocess.Popen(stdout=subprocess.PIPE,
  38                              stderr=subprocess.PIPE,
  39                              args=command,
  40                              cwd=cwd,
  41                              env=env)
  42   output, error_output = process.communicate()
  43   retcode = process.returncode
  44
  45   if retcode:
  46     raise RunError(retcode, command, output, error_output)
  47   return output, error_output
  48
  49
  50 class TestZip(unittest.TestCase):
  51   def setUp(self):
  52     # make zipname -> "testFooBar.zip"
  53     self.zipname = self.id().split('.')[-1] + '.zip'
  54     self.zipfile = None
  55     self.tempdir = tempfile.mkdtemp()
  56     shutil.copy(os.path.join(PARENT_DIR, 'oshelpers.py'),
  57         self.tempdir)
  58
  59   def tearDown(self):
  60     if self.zipfile:
  61       self.zipfile.close()
  62     shutil.rmtree(self.tempdir)
  63
  64   def GetTempPath(self, basename):
  65     return os.path.join(self.tempdir, basename)
  66
  67   def MakeFile(self, rel_path, size):
  68     with open(os.path.join(self.tempdir, rel_path), 'wb') as f:
  69       f.write('0' * size)
  70     return rel_path
  71
  72   def RunZip(self, *args):
  73     return RunCmd('zip', list(args), cwd=self.tempdir)
  74
  75   def OpenZipFile(self):
  76     self.zipfile = zipfile.ZipFile(self.GetTempPath(self.zipname), 'r')
  77
  78   def CloseZipFile(self):
  79     self.zipfile.close()
  80     self.zipfile = None
  81
  82   def GetZipInfo(self, path):
  83     return self.zipfile.getinfo(oshelpers.OSMakeZipPath(path))
  84
  85   def testNothingToDo(self):
  86     self.assertRaises(subprocess.CalledProcessError, self.RunZip,
  87         self.zipname, 'nonexistent_file')
  88     self.assertFalse(os.path.exists(self.zipname))
  89
  90   def testAddSomeFiles(self):
  91     file1 = self.MakeFile('file1', 1024)
  92     file2 = self.MakeFile('file2', 3354)
  93     self.RunZip(self.zipname, file1, file2)
  94     self.OpenZipFile()
  95     self.assertEqual(len(self.zipfile.namelist()), 2)
  96     self.assertEqual(self.GetZipInfo(file1).file_size, 1024)
  97     self.assertEqual(self.GetZipInfo(file2).file_size, 3354)
  98     # make sure files are added in order
  99     self.assertEqual(self.zipfile.namelist()[0], file1)
 100
 101   def testAddFilesWithGlob(self):
 102     self.MakeFile('file1', 1024)
 103     self.MakeFile('file2', 3354)
 104     self.RunZip(self.zipname, 'file*')
 105     self.OpenZipFile()
 106     self.assertEqual(len(self.zipfile.namelist()), 2)
 107
 108   def testAddDir(self):
 109     os.mkdir(self.GetTempPath('dir1'))
 110     self.RunZip(self.zipname, 'dir1')
 111     self.OpenZipFile()
 112     self.assertEqual(len(self.zipfile.namelist()), 1)
 113     self.assertRaises(KeyError, self.zipfile.getinfo, 'dir1')
 114     self.zipfile.getinfo('dir1/')
 115
 116   def testAddRecursive(self):
 117     os.mkdir(self.GetTempPath('dir1'))
 118     self.MakeFile(os.path.join('dir1', 'file1'), 256)
 119     os.mkdir(self.GetTempPath(os.path.join('dir1', 'dir2')))
 120     self.MakeFile(os.path.join('dir1', 'dir2', 'file2'), 1234)
 121     self.RunZip(self.zipname, '-r', 'dir1')
 122     self.OpenZipFile()
 123     self.assertEqual(len(self.zipfile.namelist()), 4)
 124
 125   def testUpdate(self):
 126     file1 = self.MakeFile('file1', 1223)
 127     self.RunZip(self.zipname, file1)
 128     self.OpenZipFile()
 129     self.assertEqual(self.GetZipInfo(file1).file_size, 1223)
 130
 131     file1 = self.MakeFile('file1', 2334)
 132     self.RunZip(self.zipname, file1)
 133     self.OpenZipFile()
 134     self.assertEqual(len(self.zipfile.namelist()), 1)
 135     self.assertEqual(self.GetZipInfo(file1).file_size, 2334)
 136
 137   def testUpdateOneFileOutOfMany(self):
 138     file1 = self.MakeFile('file1', 128)
 139     file2 = self.MakeFile('file2', 256)
 140     file3 = self.MakeFile('file3', 512)
 141     file4 = self.MakeFile('file4', 1024)
 142     self.RunZip(self.zipname, file1, file2, file3, file4)
 143     self.OpenZipFile()
 144     self.assertEqual(len(self.zipfile.namelist()), 4)
 145     self.CloseZipFile()
 146
 147     file3 = self.MakeFile('file3', 768)
 148     self.RunZip(self.zipname, file3)
 149     self.OpenZipFile()
 150     self.assertEqual(len(self.zipfile.namelist()), 4)
 151     self.assertEqual(self.zipfile.namelist()[0], file1)
 152     self.assertEqual(self.GetZipInfo(file1).file_size, 128)
 153     self.assertEqual(self.zipfile.namelist()[1], file2)
 154     self.assertEqual(self.GetZipInfo(file2).file_size, 256)
 155     self.assertEqual(self.zipfile.namelist()[2], file3)
 156     self.assertEqual(self.GetZipInfo(file3).file_size, 768)
 157     self.assertEqual(self.zipfile.namelist()[3], file4)
 158     self.assertEqual(self.GetZipInfo(file4).file_size, 1024)
 159
 160   def testUpdateSubdirectory(self):
 161     os.mkdir(self.GetTempPath('dir1'))
 162     file1 = self.MakeFile(os.path.join('dir1', 'file1'), 256)
 163     os.mkdir(self.GetTempPath(os.path.join('dir1', 'dir2')))
 164     self.MakeFile(os.path.join('dir1', 'dir2', 'file2'), 1234)
 165     self.RunZip(self.zipname, '-r', 'dir1')
 166     self.OpenZipFile()
 167     self.assertEqual(len(self.zipfile.namelist()), 4)
 168     self.assertEqual(self.GetZipInfo(file1).file_size, 256)
 169     self.CloseZipFile()
 170
 171     self.MakeFile(file1, 2560)
 172     self.RunZip(self.zipname, file1)
 173     self.OpenZipFile()
 174     self.assertEqual(len(self.zipfile.namelist()), 4)
 175     self.assertEqual(self.GetZipInfo(file1).file_size, 2560)
 176
 177   def testAppend(self):
 178     file1 = self.MakeFile('file1', 128)
 179     file2 = self.MakeFile('file2', 256)
 180     self.RunZip(self.zipname, file1, file2)
 181     self.OpenZipFile()
 182     self.assertEqual(len(self.zipfile.namelist()), 2)
 183     self.CloseZipFile()
 184
 185     file3 = self.MakeFile('file3', 768)
 186     self.RunZip(self.zipname, file3)
 187     self.OpenZipFile()
 188     self.assertEqual(len(self.zipfile.namelist()), 3)
 189
 190
 191 class TestWhich(unittest.TestCase):
 192   def setUp(self):
 193     self.path_list = []
 194     self.tempdir = tempfile.mkdtemp()
 195     shutil.copy(os.path.join(PARENT_DIR, 'oshelpers.py'),
 196         self.tempdir)
 197
 198   def tearDown(self):
 199     shutil.rmtree(self.tempdir)
 200
 201   def Mkdir(self, path):
 202     os.mkdir(os.path.join(self.tempdir, path))
 203
 204   def MakeExecutableFile(self, *path_components):
 205     path = os.path.join(self.tempdir, *path_components)
 206     if sys.platform == 'win32':
 207       path += '.exe'
 208
 209     with open(path, 'w') as f:
 210       f.write('')
 211     os.chmod(path, 0755)
 212
 213     return path
 214
 215   def RunWhich(self, *args):
 216     paths = os.pathsep.join(os.path.join(self.tempdir, p)
 217                             for p in self.path_list)
 218     env = {'PATH': paths}
 219     return RunCmd('which', list(args), cwd=self.tempdir, env=env)
 220
 221   def testNothing(self):
 222     self.assertRaises(RunError, self.RunWhich, 'foo')
 223
 224   def testBasic(self):
 225     self.Mkdir('bin')
 226     bin_cp = self.MakeExecutableFile('bin', 'cp')
 227     cp = os.path.basename(bin_cp)
 228
 229     self.path_list.append('bin')
 230     output, _ = self.RunWhich(cp)
 231     self.assertTrue(os.path.join(self.tempdir, 'bin', cp) in output)
 232
 233   def testMulti(self):
 234     self.Mkdir('bin')
 235     bin_cp = self.MakeExecutableFile('bin', 'cp')
 236     bin_ls = self.MakeExecutableFile('bin', 'ls')
 237     cp = os.path.basename(bin_cp)
 238     ls = os.path.basename(bin_ls)
 239
 240     self.path_list.append('bin')
 241     output, _ = self.RunWhich(cp, ls)
 242     self.assertTrue(os.path.join(self.tempdir, 'bin', cp) in output)
 243     self.assertTrue(os.path.join(self.tempdir, 'bin', ls) in output)
 244
 245   def testNonPath(self):
 246     self.Mkdir('bin')
 247     bin_cp = self.MakeExecutableFile('bin', 'cp')
 248     cp = os.path.basename(bin_cp)
 249
 250     # Note, "bin" not added to PATH.
 251     output, _ = self.RunWhich(bin_cp)
 252     self.assertTrue(os.path.join('bin', cp) in output)
 253
 254
 255 if __name__ == '__main__':
 256   unittest.main()