| OLD | NEW |
|---|
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # Copyright (c) 2011 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 """Tests for pyauto_utils.""" | |
| 7 | |
| 8 import glob | |
| 9 import os | |
| 10 import shutil | |
| 11 import tempfile | |
| 12 import unittest | |
| 13 | |
| 14 import pyauto_utils | |
| 15 | |
| 16 | |
| 17 class ExistingPathReplacerTest(unittest.TestCase): | |
| 18 """Tests for ExistingPathReplacer.""" | |
| 19 | |
| 20 def setUp(self): | |
| 21 self._workdir = tempfile.mkdtemp() | |
| 22 self.assertEqual(0, len(os.listdir(self._workdir))) | |
| 23 | |
| 24 def tearDown(self): | |
| 25 shutil.rmtree(self._workdir, ignore_errors=True) | |
| 26 | |
| 27 def _CreateFile(self, path): | |
| 28 fp = open(path, 'w') | |
| 29 fp.write('magic') | |
| 30 fp.close() | |
| 31 | |
| 32 def _IsOrigFile(self, path): | |
| 33 if not os.path.isfile(path): | |
| 34 return False | |
| 35 return open(path).read() == 'magic' | |
| 36 | |
| 37 def testNonExistingFile(self): | |
| 38 """Test when the requested file does not exist.""" | |
| 39 myfile = os.path.join(self._workdir, 'myfile.txt') | |
| 40 self.assertFalse(os.path.isfile(myfile)) | |
| 41 r = pyauto_utils.ExistingPathReplacer(myfile, path_type='file') | |
| 42 self.assertTrue(os.path.isfile(myfile)) | |
| 43 del r | |
| 44 self.assertEqual(0, len(os.listdir(self._workdir))) | |
| 45 | |
| 46 def testExistingFile(self): | |
| 47 """Test when the requested file exists.""" | |
| 48 myfile = os.path.join(self._workdir, 'myfile.txt') | |
| 49 self._CreateFile(myfile) | |
| 50 self.assertTrue(self._IsOrigFile(myfile)) | |
| 51 r = pyauto_utils.ExistingPathReplacer(myfile, path_type='file') | |
| 52 self.assertFalse(self._IsOrigFile(myfile)) | |
| 53 self.assertEqual(2, len(os.listdir(self._workdir))) | |
| 54 del r | |
| 55 self.assertEqual(1, len(os.listdir(self._workdir))) | |
| 56 self.assertTrue(self._IsOrigFile(myfile)) | |
| 57 | |
| 58 def testNonExistingDir(self): | |
| 59 """Test when the requested dir does not exist.""" | |
| 60 mydir = os.path.join(self._workdir, 'mydir') | |
| 61 self.assertFalse(os.path.isdir(mydir)) | |
| 62 r = pyauto_utils.ExistingPathReplacer(mydir, path_type='dir') | |
| 63 self.assertTrue(os.path.isdir(mydir)) | |
| 64 self.assertEqual(0, len(os.listdir(mydir))) | |
| 65 del r | |
| 66 self.assertFalse(os.path.isdir(mydir)) | |
| 67 | |
| 68 def testExistingDir(self): | |
| 69 """Test when the requested dir exists.""" | |
| 70 # Create a dir with one file | |
| 71 mydir = os.path.join(self._workdir, 'mydir') | |
| 72 os.makedirs(mydir) | |
| 73 self.assertEqual(1, len(os.listdir(self._workdir))) | |
| 74 myfile = os.path.join(mydir, 'myfile.txt') | |
| 75 open(myfile, 'w').close() | |
| 76 self.assertTrue(os.path.isfile(myfile)) | |
| 77 r = pyauto_utils.ExistingPathReplacer(mydir) | |
| 78 self.assertEqual(2, len(os.listdir(self._workdir))) | |
| 79 self.assertFalse(os.path.isfile(myfile)) | |
| 80 del r | |
| 81 self.assertEqual(1, len(os.listdir(self._workdir))) | |
| 82 self.assertTrue(os.path.isfile(myfile)) | |
| 83 | |
| 84 | |
| 85 if __name__ == '__main__': | |
| 86 unittest.main() | |
| OLD | NEW |
|---|
Chromium Code Reviews
