local_file_system_test.py revision 4e180b6a0b4720a9b8e9e959a882386f690f08ff 
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
6import os
7import sys
8import unittest
9
10from local_file_system import LocalFileSystem
11
12class LocalFileSystemTest(unittest.TestCase):
13  def setUp(self):
14    self._file_system = LocalFileSystem(os.path.join(sys.path[0],
15                                                     'test_data',
16                                                     'file_system'))
17
18  def testReadFiles(self):
19    expected = {
20      'test1.txt': 'test1\n',
21      'test2.txt': 'test2\n',
22      'test3.txt': 'test3\n',
23    }
24    self.assertEqual(
25        expected,
26        self._file_system.Read(['test1.txt', 'test2.txt', 'test3.txt']).Get())
27
28  def testListDir(self):
29    expected = ['dir/']
30    for i in range(7):
31      expected.append('file%d.html' % i)
32    self.assertEqual(expected,
33                     sorted(self._file_system.ReadSingle('list/').Get()))
34
35if __name__ == '__main__':
36  unittest.main()
37