← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/trace.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-09-01 06:53:18 UTC
  • mfrom: (3668.1.1 trunk)
  • Revision ID: pqm@pqm.ubuntu.com-20080901065318-3r92nk175nbu54dw
(vila, verterok) Fix some OSX failing tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/trace.py
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 traces transport operations.
18
18
 
20
20
and then delegates it.
21
21
"""
22
22
 
23
 
from __future__ import absolute_import
24
 
 
25
 
from bzrlib.transport import decorator
26
 
 
27
 
 
28
 
class TransportTraceDecorator(decorator.TransportDecorator):
 
23
from bzrlib.transport.decorator import TransportDecorator, DecoratorServer
 
24
 
 
25
 
 
26
class TransportTraceDecorator(TransportDecorator):
29
27
    """A tracing decorator for Transports.
30
28
 
31
29
    Calls that potentially perform IO are logged to self._activity. The
36
34
    operation please add a test to the tests of this transport, for the logging
37
35
    of the operation you want logged.
38
36
 
39
 
    See also TransportLogDecorator, that records a machine-readable log in 
40
 
    memory for eg testing.
 
37
    Another future enhancement would be to log to bzrlib.trace.mutter when
 
38
    trace+ is used from the command line (or perhaps as well/instead use
 
39
    -Dtransport), to make tracing operations of the entire program easily.
41
40
    """
42
41
 
43
42
    def __init__(self, url, _decorated=None, _from_transport=None):
44
43
        """Set the 'base' path where files will be stored.
45
 
 
 
44
        
46
45
        _decorated is a private parameter for cloning.
47
46
        """
48
 
        super(TransportTraceDecorator, self).__init__(url, _decorated)
 
47
        TransportDecorator.__init__(self, url, _decorated)
49
48
        if _from_transport is None:
50
49
            # newly created
51
50
            self._activity = []
77
76
 
78
77
    def get(self, relpath):
79
78
        """See Transport.get()."""
80
 
        self._trace(('get', relpath))
 
79
        self._activity.append(('get', relpath))
81
80
        return self._decorated.get(relpath)
82
81
 
83
82
    def get_smart_client(self):
93
92
 
94
93
    def mkdir(self, relpath, mode=None):
95
94
        """See Transport.mkdir()."""
96
 
        self._trace(('mkdir', relpath, mode))
97
95
        return self._decorated.mkdir(relpath, mode)
98
96
 
99
97
    def open_write_stream(self, relpath, mode=None):
103
101
    def put_file(self, relpath, f, mode=None):
104
102
        """See Transport.put_file()."""
105
103
        return self._decorated.put_file(relpath, f, mode)
106
 
 
 
104
    
107
105
    def put_bytes(self, relpath, bytes, mode=None):
108
106
        """See Transport.put_bytes()."""
109
 
        self._trace(('put_bytes', relpath, len(bytes), mode))
 
107
        self._activity.append(('put_bytes', relpath, len(bytes), mode))
110
108
        return self._decorated.put_bytes(relpath, bytes, mode)
111
109
 
112
 
    def put_bytes_non_atomic(self, relpath, bytes, mode=None,
113
 
        create_parent_dir=False, dir_mode=None):
114
 
        """See Transport.put_bytes_non_atomic."""
115
 
        self._trace(('put_bytes_non_atomic', relpath, len(bytes), mode,
116
 
            create_parent_dir, dir_mode))
117
 
        return self._decorated.put_bytes_non_atomic(relpath, bytes, mode=mode,
118
 
            create_parent_dir=create_parent_dir, dir_mode=dir_mode)
119
 
 
120
110
    def listable(self):
121
111
        """See Transport.listable."""
122
112
        return self._decorated.listable()
124
114
    def iter_files_recursive(self):
125
115
        """See Transport.iter_files_recursive()."""
126
116
        return self._decorated.iter_files_recursive()
127
 
 
 
117
    
128
118
    def list_dir(self, relpath):
129
119
        """See Transport.list_dir()."""
130
120
        return self._decorated.list_dir(relpath)
131
121
 
132
122
    def readv(self, relpath, offsets, adjust_for_latency=False,
133
123
        upper_limit=None):
134
 
        # we override at the readv() level rather than _readv() so that any
135
 
        # latency adjustments will be done by the underlying transport
136
 
        self._trace(('readv', relpath, offsets, adjust_for_latency,
 
124
        """See Transport.readv."""
 
125
        self._activity.append(('readv', relpath, offsets, adjust_for_latency,
137
126
            upper_limit))
138
127
        return self._decorated.readv(relpath, offsets, adjust_for_latency,
139
128
            upper_limit)
145
134
    def rename(self, rel_from, rel_to):
146
135
        self._activity.append(('rename', rel_from, rel_to))
147
136
        return self._decorated.rename(rel_from, rel_to)
148
 
 
 
137
    
149
138
    def rmdir(self, relpath):
150
139
        """See Transport.rmdir."""
151
 
        self._trace(('rmdir', relpath))
152
140
        return self._decorated.rmdir(relpath)
153
141
 
154
142
    def stat(self, relpath):
163
151
        """See Transport.lock_write."""
164
152
        return self._decorated.lock_write(relpath)
165
153
 
166
 
    def _trace(self, operation_tuple):
167
 
        """Record that a transport operation occured.
168
 
 
169
 
        :param operation: Tuple of transport call name and arguments.
170
 
        """
171
 
        self._activity.append(operation_tuple)
 
154
 
 
155
class TraceServer(DecoratorServer):
 
156
    """Server for the TransportTraceDecorator for testing with."""
 
157
 
 
158
    def get_decorator_class(self):
 
159
        return TransportTraceDecorator
172
160
 
173
161
 
174
162
def get_test_permutations():
175
163
    """Return the permutations to be used in testing."""
176
 
    from bzrlib.tests import test_server
177
 
    return [(TransportTraceDecorator, test_server.TraceServer)]
 
164
    return [(TransportTraceDecorator, TraceServer)]