Add squash testing code to registry tests

Fixes #896
2015-11-20 14:47:56 -05:00 · 2015-11-20 14:47:56 -05:00 · e9b577104d
commit e9b577104d
parent 352a2b7280
2 changed files with 50 additions and 1 deletions
--- a/storage/fakestorage.py
+++ b/storage/fakestorage.py
@ -33,7 +33,7 @@ class FakeStorage(BaseStorageV2):
      yield buf

  def stream_read_file(self, path):
-    return StringIO(_FAKE_STORAGE_MAP[path])
+    return StringIO.StringIO(self.get_content(path))

  def stream_write(self, path, fp, content_type=None, content_encoding=None):
    out_fp = _FAKE_STORAGE_MAP[path]