Fix up client ips to read from pending data

This commit is contained in:
Erik Johnston 2017-06-27 14:46:12 +01:00
parent ed3d0170d9
commit a0a561ae85
4 changed files with 62 additions and 27 deletions

View file

@ -43,10 +43,7 @@ class ClientIpStoreTestCase(tests.unittest.TestCase):
"access_token", "ip", "user_agent", "device_id",
)
# deliberately use an iterable here to make sure that the lookup
# method doesn't iterate it twice
device_list = iter(((user_id, "device_id"),))
result = yield self.store.get_last_client_ip_by_device(device_list)
result = yield self.store.get_last_client_ip_by_device(user_id, "device_id")
r = result[(user_id, "device_id")]
self.assertDictContainsSubset(