summary refs log tree commit diff
path: root/tests/storage/test_client_ips.py
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2017-06-27 14:46:12 +0100
committerErik Johnston <erik@matrix.org>2017-06-27 14:46:12 +0100
commita0a561ae852c6a031f8611859d95e2ad563770ca (patch)
treec6e4bd6bd58526f2fcf21d96965c977aacee4d73 /tests/storage/test_client_ips.py
parentBatch upsert user ips (diff)
downloadsynapse-a0a561ae852c6a031f8611859d95e2ad563770ca.tar.xz
Fix up client ips to read from pending data
Diffstat (limited to 'tests/storage/test_client_ips.py')
-rw-r--r--tests/storage/test_client_ips.py5
1 files changed, 1 insertions, 4 deletions
diff --git a/tests/storage/test_client_ips.py b/tests/storage/test_client_ips.py
index 1f0c0e7c37..03df697575 100644
--- a/tests/storage/test_client_ips.py
+++ b/tests/storage/test_client_ips.py
@@ -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(