Changeset 65787e5 in trunk

Timestamp:

2022-01-20T17:57:52Z (3 years ago)

Author:

Itamar Turner-Trauring <itamar@…>

Branches:

master

Children:

c4d71a46

Parents:

4b5c71f

Message:

Get rid of inlineCallbacks.

File:

• src/allmydata/test/test_storage_http.py (modified) (10 diffs)

src/allmydata/test/test_storage_http.py ¶

@@ -17 +17 @@
 from base64 import b64encode
 from os import urandom
-
-from twisted.internet.defer import inlineCallbacks
 
 from hypothesis import assume, given, strategies as st
@@ -165 +163 @@
 
 
+def result_of(d):
+    """
+    Synchronously extract the result of a Deferred.
+    """
+    result = []
+    error = []
+    d.addCallbacks(result.append, error.append)
+    if result:
+        return result[0]
+    if error:
+        error[0].raiseException()
+    raise RuntimeError(
+        "We expected given Deferred to have result already, but it wasn't. "
+        + "This is probably a test design issue."
+    )
+
+
 class RoutingTests(SyncTestCase):
     """
@@ -183 +198 @@
         )
 
-    @inlineCallbacks
     def test_authorization_enforcement(self):
         """
@@ -190 +204 @@
         """
         # Without secret, get a 400 error.
-        response = yield self.client._request(
-            "GET",
-            "http://127.0.0.1/upload_secret",
+        response = result_of(
+            self.client._request(
+                "GET",
+                "http://127.0.0.1/upload_secret",
+            )
         )
         self.assertEqual(response.code, 400)
 
         # With secret, we're good.
-        response = yield self.client._request(
-            "GET", "http://127.0.0.1/upload_secret", upload_secret=b"MAGIC"
+        response = result_of(
+            self.client._request(
+                "GET", "http://127.0.0.1/upload_secret", upload_secret=b"MAGIC"
+            )
         )
         self.assertEqual(response.code, 200)
-        self.assertEqual((yield response.content()), b"GOOD SECRET")
+        self.assertEqual(result_of(response.content()), b"GOOD SECRET")
 
 
@@ -233 +251 @@
         self.http = self.useFixture(HttpTestFixture())
 
-    @inlineCallbacks
     def test_bad_authentication(self):
         """
@@ -245 +262 @@
         )
         with self.assertRaises(ClientException) as e:
-            yield client.get_version()
+            result_of(client.get_version())
         self.assertEqual(e.exception.args[0], 401)
 
-    @inlineCallbacks
     def test_version(self):
         """
@@ -256 +272 @@
         might change across calls.
         """
-        version = yield self.http.client.get_version()
+        version = result_of(self.http.client.get_version())
         version[b"http://allmydata.org/tahoe/protocols/storage/v1"].pop(
             b"available-space"
@@ -284 +300 @@
         self.http = self.useFixture(HttpTestFixture())
 
-    @inlineCallbacks
     def test_upload_can_be_downloaded(self):
         """
@@ -303 +318 @@
         lease_secret = urandom(32)
         storage_index = b"".join(bytes([i]) for i in range(16))
-        created = yield im_client.create(
-            storage_index, {1}, 100, upload_secret, lease_secret, lease_secret
+        created = result_of(
+            im_client.create(
+                storage_index, {1}, 100, upload_secret, lease_secret, lease_secret
+            )
         )
         self.assertEqual(
@@ -320 +337 @@
             )
 
-        finished = yield write(10, 10)
-        self.assertFalse(finished)
-        finished = yield write(30, 10)
-        self.assertFalse(finished)
-        finished = yield write(50, 10)
+        finished = result_of(write(10, 10))
+        self.assertFalse(finished)
+        finished = result_of(write(30, 10))
+        self.assertFalse(finished)
+        finished = result_of(write(50, 10))
         self.assertFalse(finished)
 
         # Then, an overlapping write with matching data (15-35):
-        finished = yield write(15, 20)
+        finished = result_of(write(15, 20))
         self.assertFalse(finished)
 
         # Now fill in the holes:
-        finished = yield write(0, 10)
-        self.assertFalse(finished)
-        finished = yield write(40, 10)
-        self.assertFalse(finished)
-        finished = yield write(60, 40)
+        finished = result_of(write(0, 10))
+        self.assertFalse(finished)
+        finished = result_of(write(40, 10))
+        self.assertFalse(finished)
+        finished = result_of(write(60, 40))
         self.assertTrue(finished)
 
         # We can now read:
         for offset, length in [(0, 100), (10, 19), (99, 1), (49, 200)]:
-            downloaded = yield im_client.read_share_chunk(
-                storage_index, 1, offset, length
+            downloaded = result_of(
+                im_client.read_share_chunk(storage_index, 1, offset, length)
             )
             self.assertEqual(downloaded, expected_data[offset : offset + length])