From: Wesley Spikes <wesley.spikes@dreamhost.com>
Date: Fri, 15 Jul 2011 19:05:02 +0000 (-0700)
Subject: Adding last_read to RandomContentFile and first_write to FileVerifier, so we can... 
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=6afc7fdcdb143fb8bb071509cb1c6f7cdd24b87e;p=s3-tests.git

Adding last_read to RandomContentFile and first_write to FileVerifier, so we can measure latency.
---

diff --git a/s3tests/realistic.py b/s3tests/realistic.py
index b3f47ba3..d025c91a 100644
--- a/s3tests/realistic.py
+++ b/s3tests/realistic.py
@@ -15,6 +15,7 @@ class RandomContentFile(object):
         self.hash = hashlib.md5()
         self.digest_size = self.hash.digest_size
         self.digest = None
+        self.last_read = 0
 
     def seek(self, offset):
         assert offset == 0
@@ -59,6 +60,7 @@ class RandomContentFile(object):
             size -= digest_count
             data = self.digest[:digest_count]
             r.append(data)
+            self.last_read = time.time()
 
         return ''.join(r)
 
@@ -67,8 +69,11 @@ class FileVerifier(object):
         self.size = 0
         self.hash = hashlib.md5()
         self.buf = ''
+        self.first_write = 0
 
     def write(self, data):
+        if self.size == 0:
+            self.first_write = time.time()
         self.size += len(data)
         self.buf += data
         digsz = -1*self.hash.digest_size