tests(webhdfs): address unawaited coroutine warnings

Author: SilviaSWR

tests/test_provider_webhdfs.py

@@ -112,14 +112,17 @@ async def test_do_get_file(mock_provider, fake_request):
     }
 
     mock_provider._get_dav_property_d0 = AsyncMock(
-        return_value=(
-            200,
-            await mock_provider._create_dav_property_obj(
-                fake_request, DAVPath("/testfile.txt"), fake_status
-            ),
-        )
+        return_value=(200, await mock_provider._create_dav_property_obj(
+            fake_request, DAVPath("/testfile.txt"), fake_status
+        ))
+    )
+
+    async def fake_generator():
+        yield b"data"
+
+    mock_provider._dav_response_data_generator = MagicMock(
+        return_value=fake_generator()
     )
-    mock_provider._dav_response_data_generator = AsyncMock(return_value=AsyncMock())
 
     fake_request.ranges = [DAVRequestRange(DAVRangeType.RANGE, 0, 100, 200)]
     status, basic_data, generator, _ = await mock_provider._do_get(fake_request)
@@ -138,14 +141,17 @@ async def test_do_get_file_non_range(mock_provider, fake_request):
     }
 
     mock_provider._get_dav_property_d0 = AsyncMock(
-        return_value=(
-            200,
-            await mock_provider._create_dav_property_obj(
-                fake_request, DAVPath("/testfile.txt"), fake_status
-            ),
-        )
+        return_value=(200, await mock_provider._create_dav_property_obj(
+            fake_request, DAVPath("/testfile.txt"), fake_status
+        ))
+    )
+
+    async def fake_generator():
+        yield b"data"
+
+    mock_provider._dav_response_data_generator = MagicMock(
+        return_value=fake_generator()
     )
-    mock_provider._dav_response_data_generator = AsyncMock(return_value=AsyncMock())
 
     fake_request.ranges = None
     status, basic_data, generator, _ = await mock_provider._do_get(fake_request)
@@ -361,24 +367,28 @@ async def __aexit__(self, exc_type, exc_val, exc_tb):
 
 @pytest.mark.asyncio
 async def test_do_put(mock_provider, fake_request):
-    mock_response = MagicMock()
-    mock_response.status_code = 200
-    mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = MagicMock()
+    mock_provider._precheck_source = AsyncMock(return_value=(True, False, False))
 
-    mock_provider.client.get.return_value = mock_response
+    first_put = AsyncMock()
+    first_put.status_code = 307
+    first_put.headers = {"location": "http://fake-hdfs:9870/webhdfs/v1/file_put_location"}
+    first_put.raise_for_status = MagicMock()
 
-    response = await mock_provider._do_put(fake_request)
+    second_put = AsyncMock()
+    second_put.status_code = 201
+    second_put.raise_for_status = MagicMock()
 
-    assert response == 204
+    mock_provider.client.put = AsyncMock(side_effect=[first_put, second_put])
 
+    response = await mock_provider._do_put(fake_request)
+    assert response == 201
 
 @pytest.mark.asyncio
 async def test_do_copy(mock_provider, fake_request):
     mock_response = MagicMock()
     mock_response.status_code = 200
     mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = AsyncMock()
+    mock_response.raise_for_status = MagicMock()
 
     mock_provider.client.get.return_value = mock_response
 
@@ -391,7 +401,7 @@ async def test_do_mkcol(mock_provider, fake_request):
     mock_response = MagicMock()
     mock_response.status_code = 200
     mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = AsyncMock()
+    mock_response.raise_for_status = MagicMock()
 
     mock_provider.client.get.return_value = mock_response
 
@@ -405,7 +415,7 @@ async def test_do_propfind(mock_provider, fake_request):
     mock_response = MagicMock()
     mock_response.status_code = 200
     mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = AsyncMock()
+    mock_response.raise_for_status = MagicMock()
 
     mock_provider.client.get.return_value = mock_response
 
@@ -477,7 +487,7 @@ async def test_do_head(mock_provider, fake_request):
     mock_response = MagicMock()
     mock_response.status_code = 200
     mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = AsyncMock()
+    mock_response.raise_for_status = MagicMock()
 
     mock_provider.client.get.return_value = mock_response
 
@@ -491,7 +501,7 @@ async def test_do_precheck_destination(mock_provider, fake_request):
     mock_response = MagicMock()
     mock_response.status_code = 200
     mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = AsyncMock()
+    mock_response.raise_for_status = MagicMock()
 
     mock_provider.client.get.return_value = mock_response
 
@@ -502,16 +512,24 @@ async def test_do_precheck_destination(mock_provider, fake_request):
 
 @pytest.mark.asyncio
 async def test_do_move(mock_provider, fake_request):
-    mock_response = MagicMock()
-    mock_response.status_code = 200
-    mock_response.json = MagicMock(return_value={"FileStatus": {}})
-    mock_response.raise_for_status = MagicMock()
+    fake_request.overwrite = True
 
-    mock_provider.client.get.return_value = mock_response
+    mock_provider._precheck_destination = AsyncMock(return_value=(True, False, True))
+
+    mock_delete_response = MagicMock()
+    mock_delete_response.status_code = 204
+    mock_delete_response.raise_for_status = MagicMock()
 
-    response = await mock_provider._do_move(fake_request)
+    mock_provider.client.delete = AsyncMock(return_value=mock_delete_response)
 
-    assert response == 204
+    mock_put_response = MagicMock()
+    mock_put_response.status_code = 201
+    mock_put_response.raise_for_status = MagicMock()
+
+    mock_provider.client.put = AsyncMock(return_value=mock_put_response)
+
+    result = await mock_provider._do_move(fake_request)
+    assert result == 204
 
 
 @pytest.mark.asyncio