The Tor Browser / file comparison
comparison: config/tests/unit-mozunit.py
config/tests/unit-mozunit.py
- changeset 0
- 6474c204b198
equal
deleted
inserted
replaced
| -1:000000000000 | 0:5bfa24579d40 |
|---|---|
| 1 # This Source Code Form is subject to the terms of the Mozilla Public | |
| 2 # License, v. 2.0. If a copy of the MPL was not distributed with this | |
| 3 # file, You can obtain one at http://mozilla.org/MPL/2.0/. | |
| 4 | |
| 5 import sys | |
| 6 import os | |
| 7 from mozunit import main, MockedOpen | |
| 8 import unittest | |
| 9 from tempfile import mkstemp | |
| 10 | |
| 11 class TestMozUnit(unittest.TestCase): | |
| 12 def test_mocked_open(self): | |
| 13 # Create a temporary file on the file system. | |
| 14 (fd, path) = mkstemp() | |
| 15 with os.fdopen(fd, 'w') as file: | |
| 16 file.write('foobar'); | |
| 17 | |
| 18 self.assertFalse(os.path.exists('file1')) | |
| 19 self.assertFalse(os.path.exists('file2')) | |
| 20 | |
| 21 with MockedOpen({'file1': 'content1', | |
| 22 'file2': 'content2'}): | |
| 23 self.assertTrue(os.path.exists('file1')) | |
| 24 self.assertTrue(os.path.exists('file2')) | |
| 25 self.assertFalse(os.path.exists('foo/file1')) | |
| 26 | |
| 27 # Check the contents of the files given at MockedOpen creation. | |
| 28 self.assertEqual(open('file1', 'r').read(), 'content1') | |
| 29 self.assertEqual(open('file2', 'r').read(), 'content2') | |
| 30 | |
| 31 # Check that overwriting these files alters their content. | |
| 32 with open('file1', 'w') as file: | |
| 33 file.write('foo') | |
| 34 self.assertTrue(os.path.exists('file1')) | |
| 35 self.assertEqual(open('file1', 'r').read(), 'foo') | |
| 36 | |
| 37 # ... but not until the file is closed. | |
| 38 file = open('file2', 'w') | |
| 39 file.write('bar') | |
| 40 self.assertEqual(open('file2', 'r').read(), 'content2') | |
| 41 file.close() | |
| 42 self.assertEqual(open('file2', 'r').read(), 'bar') | |
| 43 | |
| 44 # Check that appending to a file does append | |
| 45 with open('file1', 'a') as file: | |
| 46 file.write('bar') | |
| 47 self.assertEqual(open('file1', 'r').read(), 'foobar') | |
| 48 | |
| 49 self.assertFalse(os.path.exists('file3')) | |
| 50 | |
| 51 # Opening a non-existing file ought to fail. | |
| 52 self.assertRaises(IOError, open, 'file3', 'r') | |
| 53 self.assertFalse(os.path.exists('file3')) | |
| 54 | |
| 55 # Check that writing a new file does create the file. | |
| 56 with open('file3', 'w') as file: | |
| 57 file.write('baz') | |
| 58 self.assertEqual(open('file3', 'r').read(), 'baz') | |
| 59 self.assertTrue(os.path.exists('file3')) | |
| 60 | |
| 61 # Check the content of the file created outside MockedOpen. | |
| 62 self.assertEqual(open(path, 'r').read(), 'foobar') | |
| 63 | |
| 64 # Check that overwriting a file existing on the file system | |
| 65 # does modify its content. | |
| 66 with open(path, 'w') as file: | |
| 67 file.write('bazqux') | |
| 68 self.assertEqual(open(path, 'r').read(), 'bazqux') | |
| 69 | |
| 70 with MockedOpen(): | |
| 71 # Check that appending to a file existing on the file system | |
| 72 # does modify its content. | |
| 73 with open(path, 'a') as file: | |
| 74 file.write('bazqux') | |
| 75 self.assertEqual(open(path, 'r').read(), 'foobarbazqux') | |
| 76 | |
| 77 # Check that the file was not actually modified on the file system. | |
| 78 self.assertEqual(open(path, 'r').read(), 'foobar') | |
| 79 os.remove(path) | |
| 80 | |
| 81 # Check that the file created inside MockedOpen wasn't actually | |
| 82 # created. | |
| 83 self.assertRaises(IOError, open, 'file3', 'r') | |
| 84 | |
| 85 if __name__ == "__main__": | |
| 86 main() |
