← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/memory.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-03-16 16:58:03 UTC
  • mfrom: (3224.3.1 news-typo)
  • Revision ID: pqm@pqm.ubuntu.com-20080316165803-tisoc9mpob9z544o
(Matt Nordhoff) Trivial NEWS typo fix

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/memory.py
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""Implementation of Transport that uses memory for its storage.
18
18
 
27
27
from cStringIO import StringIO
28
28
import warnings
29
29
 
30
 
from bzrlib import (
31
 
    transport,
32
 
    urlutils,
33
 
    )
34
30
from bzrlib.errors import (
35
31
    FileExists,
36
32
    LockError,
43
39
    AppendBasedFileStream,
44
40
    _file_streams,
45
41
    LateReadError,
 
42
    register_transport,
 
43
    Server,
 
44
    Transport,
46
45
    )
 
46
import bzrlib.urlutils as urlutils
47
47
 
48
48
 
49
49
 
61
61
            self.st_mode = S_IFDIR | perms
62
62
 
63
63
 
64
 
class MemoryTransport(transport.Transport):
 
64
class MemoryTransport(Transport):
65
65
    """This is an in memory file system for transient data storage."""
66
66
 
67
67
    def __init__(self, url=""):
81
81
 
82
82
    def clone(self, offset=None):
83
83
        """See Transport.clone()."""
84
 
        path = urlutils.URL._combine_paths(self._cwd, offset)
 
84
        path = self._combine_paths(self._cwd, offset)
85
85
        if len(path) == 0 or path[-1] != '/':
86
86
            path += '/'
87
87
        url = self._scheme + path
88
 
        result = self.__class__(url)
 
88
        result = MemoryTransport(url)
89
89
        result._dirs = self._dirs
90
90
        result._files = self._files
91
91
        result._locks = self._locks
183
183
        for file in self._files:
184
184
            if file.startswith(self._cwd):
185
185
                yield urlutils.escape(file[len(self._cwd):])
186
 
 
 
186
    
187
187
    def list_dir(self, relpath):
188
188
        """See Transport.list_dir()."""
189
189
        _abspath = self._abspath(relpath)
222
222
                    del container[path]
223
223
        do_renames(self._files)
224
224
        do_renames(self._dirs)
225
 
 
 
225
    
226
226
    def rmdir(self, relpath):
227
227
        """See Transport.rmdir."""
228
228
        _abspath = self._abspath(relpath)
243
243
        """See Transport.stat()."""
244
244
        _abspath = self._abspath(relpath)
245
245
        if _abspath in self._files:
246
 
            return MemoryStat(len(self._files[_abspath][0]), False,
 
246
            return MemoryStat(len(self._files[_abspath][0]), False, 
247
247
                              self._files[_abspath][1])
248
248
        elif _abspath in self._dirs:
249
249
            return MemoryStat(0, True, self._dirs[_abspath])
283
283
    """This makes a lock."""
284
284
 
285
285
    def __init__(self, path, transport):
 
286
        assert isinstance(transport, MemoryTransport)
286
287
        self.path = path
287
288
        self.transport = transport
288
289
        if self.path in self.transport._locks:
289
290
            raise LockError('File %r already locked' % (self.path,))
290
291
        self.transport._locks[self.path] = self
291
292
 
 
293
    def __del__(self):
 
294
        # Should this warn, or actually try to cleanup?
 
295
        if self.transport:
 
296
            warnings.warn("MemoryLock %r not explicitly unlocked" % (self.path,))
 
297
            self.unlock()
 
298
 
292
299
    def unlock(self):
293
300
        del self.transport._locks[self.path]
294
301
        self.transport = None
295
302
 
296
303
 
297
 
class MemoryServer(transport.Server):
 
304
class MemoryServer(Server):
298
305
    """Server for the MemoryTransport for testing with."""
299
306
 
300
 
    def start_server(self):
 
307
    def setUp(self):
 
308
        """See bzrlib.transport.Server.setUp."""
301
309
        self._dirs = {'/':None}
302
310
        self._files = {}
303
311
        self._locks = {}
304
312
        self._scheme = "memory+%s:///" % id(self)
305
313
        def memory_factory(url):
306
 
            from bzrlib.transport import memory
307
 
            result = memory.MemoryTransport(url)
 
314
            result = MemoryTransport(url)
308
315
            result._dirs = self._dirs
309
316
            result._files = self._files
310
317
            result._locks = self._locks
311
318
            return result
312
 
        self._memory_factory = memory_factory
313
 
        transport.register_transport(self._scheme, self._memory_factory)
 
319
        register_transport(self._scheme, memory_factory)
314
320
 
315
 
    def stop_server(self):
 
321
    def tearDown(self):
 
322
        """See bzrlib.transport.Server.tearDown."""
316
323
        # unregister this server
317
 
        transport.unregister_transport(self._scheme, self._memory_factory)
318
324
 
319
325
    def get_url(self):
320
326
        """See bzrlib.transport.Server.get_url."""
321
327
        return self._scheme
322
328
 
323
 
    def get_bogus_url(self):
324
 
        raise NotImplementedError
325
 
 
326
329
 
327
330
def get_test_permutations():
328
331
    """Return the permutations to be used in testing."""