test: check the URL is returned.

nitrictech · Nov 7, 2022 · 0d23d91 · 0d23d91
1 parent ffcc097
commit 0d23d91
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/api/test_storage.py b/tests/api/test_storage.py
@@ -103,7 +103,7 @@ async def test_sign_url(self):
         with patch("nitricapi.nitric.storage.v1.StorageServiceStub.pre_sign_url", mock_pre_sign_url):
             bucket = Storage().bucket("test-bucket")
             file = bucket.file("test-file")
-            await file.sign_url()
+            url = await file.sign_url()
 
         # Check expected values were passed to Stub
         mock_pre_sign_url.assert_called_once_with(
@@ -115,6 +115,9 @@ async def test_sign_url(self):
             )
         )
 
+        # check the URL is returned
+        assert url == "www.example.com"
+
     async def test_write_error(self):
         mock_write = AsyncMock()
         mock_write.side_effect = GRPCError(Status.UNKNOWN, "test error")