1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#!/usr/bin/python
# 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."""
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
else:
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
|
