Fix AS retries
parent
b0932b34cb
commit
310197bab5
|
@ -320,10 +320,10 @@ class ApplicationServiceTransactionStore(SQLBaseStore):
|
||||||
|
|
||||||
event_ids = json.loads(entry["event_ids"])
|
event_ids = json.loads(entry["event_ids"])
|
||||||
|
|
||||||
events = yield self.get_events(event_ids)
|
event_map = yield self.get_events(event_ids)
|
||||||
|
|
||||||
defer.returnValue(AppServiceTransaction(
|
defer.returnValue(AppServiceTransaction(
|
||||||
service=service, id=entry["txn_id"], events=events
|
service=service, id=entry["txn_id"], events=event_map.values()
|
||||||
))
|
))
|
||||||
|
|
||||||
def _get_oldest_unsent_txn(self, txn, service):
|
def _get_oldest_unsent_txn(self, txn, service):
|
||||||
|
|
|
@ -353,21 +353,21 @@ class ApplicationServiceTransactionStoreTestCase(unittest.TestCase):
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_get_oldest_unsent_txn(self):
|
def test_get_oldest_unsent_txn(self):
|
||||||
service = Mock(id=self.as_list[0]["id"])
|
service = Mock(id=self.as_list[0]["id"])
|
||||||
events = [Mock(event_id="e1"), Mock(event_id="e2")]
|
events = {"e1": Mock(event_id="e1"), "e2": Mock(event_id="e2")}
|
||||||
other_events = [Mock(event_id="e5"), Mock(event_id="e6")]
|
other_events = [Mock(event_id="e5"), Mock(event_id="e6")]
|
||||||
|
|
||||||
# we aren't testing store._base stuff here, so mock this out
|
# we aren't testing store._base stuff here, so mock this out
|
||||||
self.store.get_events = Mock(return_value=events)
|
self.store.get_events = Mock(return_value=events)
|
||||||
|
|
||||||
yield self._insert_txn(self.as_list[1]["id"], 9, other_events)
|
yield self._insert_txn(self.as_list[1]["id"], 9, other_events)
|
||||||
yield self._insert_txn(service.id, 10, events)
|
yield self._insert_txn(service.id, 10, events.values())
|
||||||
yield self._insert_txn(service.id, 11, other_events)
|
yield self._insert_txn(service.id, 11, other_events)
|
||||||
yield self._insert_txn(service.id, 12, other_events)
|
yield self._insert_txn(service.id, 12, other_events)
|
||||||
|
|
||||||
txn = yield self.store.get_oldest_unsent_txn(service)
|
txn = yield self.store.get_oldest_unsent_txn(service)
|
||||||
self.assertEquals(service, txn.service)
|
self.assertEquals(service, txn.service)
|
||||||
self.assertEquals(10, txn.id)
|
self.assertEquals(10, txn.id)
|
||||||
self.assertEquals(events, txn.events)
|
self.assertEquals(events.values(), txn.events)
|
||||||
|
|
||||||
@defer.inlineCallbacks
|
@defer.inlineCallbacks
|
||||||
def test_get_appservices_by_state_single(self):
|
def test_get_appservices_by_state_single(self):
|
||||||
|
|
Loading…
Reference in New Issue