Index: scripts/master/buildbucket/unittests/integration_test.py |
diff --git a/scripts/master/buildbucket/unittests/integration_test.py b/scripts/master/buildbucket/unittests/integration_test.py |
index f6fe5d8525d572ed725d3be97b2d6e3d5d530108..1807a9f8be4d50484f923d5d2071a8aeaf7a920f 100755 |
--- a/scripts/master/buildbucket/unittests/integration_test.py |
+++ b/scripts/master/buildbucket/unittests/integration_test.py |
@@ -122,13 +122,12 @@ class IntegratorTest(unittest.TestCase): |
self.integrator = integration.BuildBucketIntegrator( |
self.buckets, max_lease_count=max_lease_count, |
build_params_hook=build_params_hook, |
- heartbeat_interval=datetime.timedelta(microseconds=1)) |
+ heartbeat_interval=datetime.timedelta(microseconds=1), |
+ change_store_factory=lambda bb: self.changes) |
def log(msg, level=logging.INFO): |
logging.log(level, msg) |
self.integrator.log = log |
- self.integrator.start( |
- self.buildbot, self.buildbucket, |
- change_store_factory=lambda bb: self.changes) |
+ self.integrator.start(self.buildbot, self.buildbucket) |
try: |
yield |
finally: |