OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright 2013 The Chromium Authors. All rights reserved. | 2 # Copyright 2013 The Chromium Authors. All rights reserved. |
3 # Use of this source code is governed by a BSD-style license that can be | 3 # Use of this source code is governed by a BSD-style license that can be |
4 # found in the LICENSE file. | 4 # found in the LICENSE file. |
5 | 5 |
6 import json | 6 import json |
7 from copy import deepcopy | 7 from copy import deepcopy |
8 from cStringIO import StringIO | 8 from cStringIO import StringIO |
9 from functools import partial | 9 from functools import partial |
10 from hashlib import sha1 | 10 from hashlib import sha1 |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
61 'changing-repo', create_mock_url_fetcher, path='') | 61 'changing-repo', create_mock_url_fetcher, path='') |
62 assert len(fetchers) == 1 | 62 assert len(fetchers) == 1 |
63 return gfs, fetchers[0] | 63 return gfs, fetchers[0] |
64 | 64 |
65 def Mutate(self): | 65 def Mutate(self): |
66 fake_version = self._GenerateHash() | 66 fake_version = self._GenerateHash() |
67 fake_data = self._GenerateHash() | 67 fake_data = self._GenerateHash() |
68 self.files['zipfile/hello.txt'] = fake_data | 68 self.files['zipfile/hello.txt'] = fake_data |
69 self.files['zipfile/new-file'] = fake_data | 69 self.files['zipfile/new-file'] = fake_data |
70 self.files['zipfile/dir/file1'] = fake_data | 70 self.files['zipfile/dir/file1'] = fake_data |
| 71 # XXX(kalman): These don't work anymore because TestFileSystem no longer |
| 72 # just uses the object it was given, but instead mutates it on |
| 73 # construction. For now the tests that rely on this (i.e. Mutate) are |
| 74 # disabled, and in fact NewGithubFileSystem isn't really used anymore, so |
| 75 # rather than fixing this we may just want to delete it all. |
71 self._test_files['test_owner']['changing-repo']['zipball'] = ( | 76 self._test_files['test_owner']['changing-repo']['zipball'] = ( |
72 self._ZipFromFiles(self.files)) | 77 self._ZipFromFiles(self.files)) |
73 self._test_files['test_owner']['changing-repo']['commits']['HEAD'] = ( | 78 self._test_files['test_owner']['changing-repo']['commits']['HEAD'] = ( |
74 self._MakeShaJson(fake_version)) | 79 self._MakeShaJson(fake_version)) |
75 return fake_version, fake_data | 80 return fake_version, fake_data |
76 | 81 |
77 def _GenerateHash(self): | 82 def _GenerateHash(self): |
78 '''Generates an arbitrary SHA1 hash. | 83 '''Generates an arbitrary SHA1 hash. |
79 ''' | 84 ''' |
80 return sha1(str(random())).hexdigest() | 85 return sha1(str(random())).hexdigest() |
(...skipping 92 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
173 self.assertEqual( | 178 self.assertEqual( |
174 initial_cgfs_read_two, | 179 initial_cgfs_read_two, |
175 self._cgfs.Read(['README.md', 'requirements.txt']).Get()) | 180 self._cgfs.Read(['README.md', 'requirements.txt']).Get()) |
176 | 181 |
177 def testWithoutRefresh(self): | 182 def testWithoutRefresh(self): |
178 # Without refreshing it will still read the content from blobstore, and it | 183 # Without refreshing it will still read the content from blobstore, and it |
179 # does this via the magic of the FakeURLFSFetcher. | 184 # does this via the magic of the FakeURLFSFetcher. |
180 self.assertEqual(['__init__.notpy', 'hello.notpy'], | 185 self.assertEqual(['__init__.notpy', 'hello.notpy'], |
181 sorted(self._gfs.ReadSingle('src/').Get())) | 186 sorted(self._gfs.ReadSingle('src/').Get())) |
182 | 187 |
183 def testRefresh(self): | 188 def DISABLED_testRefresh(self): |
184 test_bundle = _TestBundle() | 189 test_bundle = _TestBundle() |
185 gfs, fetcher = test_bundle.CreateGfsAndFetcher() | 190 gfs, fetcher = test_bundle.CreateGfsAndFetcher() |
186 | 191 |
187 # It shouldn't fetch until Refresh does so; then it will do 2, one for the | 192 # It shouldn't fetch until Refresh does so; then it will do 2, one for the |
188 # stat, and another for the read. | 193 # stat, and another for the read. |
189 self.assertTrue(*fetcher.CheckAndReset()) | 194 self.assertTrue(*fetcher.CheckAndReset()) |
190 gfs.Refresh().Get() | 195 gfs.Refresh().Get() |
191 self.assertTrue(*fetcher.CheckAndReset(fetch_count=1, | 196 self.assertTrue(*fetcher.CheckAndReset(fetch_count=1, |
192 fetch_async_count=1, | 197 fetch_async_count=1, |
193 fetch_resolve_count=1)) | 198 fetch_resolve_count=1)) |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
227 self.assertTrue(*fetcher.CheckAndReset(fetch_count=1, fetch_async_count=1)) | 232 self.assertTrue(*fetcher.CheckAndReset(fetch_count=1, fetch_async_count=1)) |
228 | 233 |
229 self.assertEqual(data, gfs.ReadSingle('new-file').Get()) | 234 self.assertEqual(data, gfs.ReadSingle('new-file').Get()) |
230 self.assertEqual(test_bundle.files['zipfile/dir/file1'], | 235 self.assertEqual(test_bundle.files['zipfile/dir/file1'], |
231 gfs.ReadSingle('dir/file1').Get()) | 236 gfs.ReadSingle('dir/file1').Get()) |
232 self.assertEqual(StatInfo(version), gfs.Stat('new-file')) | 237 self.assertEqual(StatInfo(version), gfs.Stat('new-file')) |
233 | 238 |
234 refresh_future.Get() | 239 refresh_future.Get() |
235 self.assertTrue(*fetcher.CheckAndReset(fetch_resolve_count=1)) | 240 self.assertTrue(*fetcher.CheckAndReset(fetch_resolve_count=1)) |
236 | 241 |
237 def testGetThenRefreshOnStartup(self): | 242 def DISABLED_testGetThenRefreshOnStartup(self): |
238 # Regression test: Test that calling Get() but never resolving the future, | 243 # Regression test: Test that calling Get() but never resolving the future, |
239 # then Refresh()ing the data, causes the data to be refreshed. | 244 # then Refresh()ing the data, causes the data to be refreshed. |
240 test_bundle = _TestBundle() | 245 test_bundle = _TestBundle() |
241 gfs, fetcher = test_bundle.CreateGfsAndFetcher() | 246 gfs, fetcher = test_bundle.CreateGfsAndFetcher() |
242 self.assertTrue(*fetcher.CheckAndReset()) | 247 self.assertTrue(*fetcher.CheckAndReset()) |
243 | 248 |
244 # Get a predictable version. | 249 # Get a predictable version. |
245 version, data = test_bundle.Mutate() | 250 version, data = test_bundle.Mutate() |
246 | 251 |
247 read_future = gfs.ReadSingle('hello.txt') | 252 read_future = gfs.ReadSingle('hello.txt') |
(...skipping 14 matching lines...) Expand all Loading... |
262 self.assertTrue(*fetcher.CheckAndReset()) | 267 self.assertTrue(*fetcher.CheckAndReset()) |
263 | 268 |
264 # Read data should not have changed. | 269 # Read data should not have changed. |
265 self.assertEqual(data, gfs.ReadSingle('hello.txt').Get()) | 270 self.assertEqual(data, gfs.ReadSingle('hello.txt').Get()) |
266 self.assertEqual(StatInfo(version), gfs.Stat('hello.txt')) | 271 self.assertEqual(StatInfo(version), gfs.Stat('hello.txt')) |
267 self.assertTrue(*fetcher.CheckAndReset()) | 272 self.assertTrue(*fetcher.CheckAndReset()) |
268 | 273 |
269 | 274 |
270 if __name__ == '__main__': | 275 if __name__ == '__main__': |
271 unittest.main() | 276 unittest.main() |
OLD | NEW |