summary refs log tree commit diff
path: root/tests/appservice/test_scheduler.py
diff options
context:
space:
mode:
authorKegan Dougal <kegan@matrix.org>2015-03-06 16:17:01 +0000
committerKegan Dougal <kegan@matrix.org>2015-03-06 16:17:01 +0000
commit10766f1e93f884d8a71af43f565183a54786a3ca (patch)
tree562450404d9397e6c38b829bedd6c53ea9f7840b /tests/appservice/test_scheduler.py
parentApply clarity and docstrings (diff)
downloadsynapse-10766f1e93f884d8a71af43f565183a54786a3ca.tar.xz
Update UTs
Diffstat (limited to 'tests/appservice/test_scheduler.py')
-rw-r--r--tests/appservice/test_scheduler.py10
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/appservice/test_scheduler.py b/tests/appservice/test_scheduler.py
index f75a6f5d95..9532bf66b8 100644
--- a/tests/appservice/test_scheduler.py
+++ b/tests/appservice/test_scheduler.py
@@ -155,18 +155,18 @@ class ApplicationServiceSchedulerRecovererTestCase(unittest.TestCase):
 
         def take_txn(*args, **kwargs):
             return defer.succeed(txns.pop(0))
-        self.store.get_oldest_txn = Mock(side_effect=take_txn)
+        self.store.get_oldest_unsent_txn = Mock(side_effect=take_txn)
 
         self.recoverer.recover()
         # shouldn't have called anything prior to waiting for exp backoff
-        self.assertEquals(0, self.store.get_oldest_txn.call_count)
+        self.assertEquals(0, self.store.get_oldest_unsent_txn.call_count)
         txn.send = Mock(return_value=True)
         # wait for exp backoff
         self.clock.advance_time(2000)
         self.assertEquals(1, txn.send.call_count)
         self.assertEquals(1, txn.complete.call_count)
         # 2 because it needs to get None to know there are no more txns
-        self.assertEquals(2, self.store.get_oldest_txn.call_count)
+        self.assertEquals(2, self.store.get_oldest_unsent_txn.call_count)
         self.callback.assert_called_once_with(self.recoverer)
         self.assertEquals(self.recoverer.service, self.service)
 
@@ -180,10 +180,10 @@ class ApplicationServiceSchedulerRecovererTestCase(unittest.TestCase):
                 return defer.succeed(txns.pop(0))
             else:
                 return defer.succeed(txn)
-        self.store.get_oldest_txn = Mock(side_effect=take_txn)
+        self.store.get_oldest_unsent_txn = Mock(side_effect=take_txn)
 
         self.recoverer.recover()
-        self.assertEquals(0, self.store.get_oldest_txn.call_count)
+        self.assertEquals(0, self.store.get_oldest_unsent_txn.call_count)
         txn.send = Mock(return_value=False)
         self.clock.advance_time(2000)
         self.assertEquals(1, txn.send.call_count)