blob: d4373ce62012fea4747d48a171a398a86a44413a [file] [log] [blame]
# Copyright (c) 2012 The Native Client Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Fake (similar in memory implementation) version of GSDStorage."""
from pynacl import file_tools
class FakeStorage(object):
"""Testing replacement for GSDStorage that stores to memory."""
def __init__(self):
self._store = {}
self._write_count = 0
def Exists(self, key):
return self._store.get(key, False)
def PutData(self, data, key, clobber=True):
if clobber or key not in self._store:
self._store[key] = data
self._write_count += 1
def PutFile(self, path, key, clobber=True):
self.PutData(file_tools.ReadFile(path), key, clobber=clobber)
# Use form: fake://<key> for make-believe URLs.
return 'fake://' + key
def GetData(self, key):
return self._store.get(key)
def GetFile(self, key, path):
data = self.GetData(key)
if data is not None:
file_tools.WriteFile(data, path)
# Use form: fake://<key> for make-believe URLs.
return 'fake://' + key
return None
def ItemCount(self):
"""Return a count of how many total unique keys have been written."""
return len(self._store)
def WriteCount(self):
"""Return a count of how many total writes have been done."""
return self._write_count