| #!/usr/bin/env python |
| # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import os |
| import sys |
| import unittest |
| |
| from local_file_system import LocalFileSystem |
| |
| class LocalFileSystemTest(unittest.TestCase): |
| def setUp(self): |
| self._file_system = LocalFileSystem(os.path.join(sys.path[0], |
| 'test_data', |
| 'file_system')) |
| |
| def testReadFiles(self): |
| expected = { |
| 'test1.txt': 'test1\n', |
| 'test2.txt': 'test2\n', |
| 'test3.txt': 'test3\n', |
| } |
| self.assertEqual( |
| expected, |
| self._file_system.Read(['test1.txt', 'test2.txt', 'test3.txt']).Get()) |
| |
| def testListDir(self): |
| expected = ['dir/'] |
| for i in range(7): |
| expected.append('file%d.html' % i) |
| self.assertEqual(expected, |
| sorted(self._file_system.ReadSingle('list/'))) |
| |
| if __name__ == '__main__': |
| unittest.main() |