Skip to content

Commit 8113e32

Browse files
committed
use tempfile
1 parent eb08e19 commit 8113e32

1 file changed

Lines changed: 7 additions & 7 deletions

File tree

sdk/core/azure-core/tests/test_retry_policy.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
HttpResponse,
2020
HttpTransport,
2121
)
22-
import uuid
22+
import tempfile
2323
import os
2424

2525
def test_retry_code_class_variables():
@@ -143,19 +143,19 @@ def send(self, request, **kwargs): # type: (PipelineRequest, Any) -> PipelineRe
143143
assert not position
144144
return HttpResponse(request, None)
145145

146-
file_name = str(uuid.uuid4())
147-
with open(file_name, "w+") as f:
148-
f.write('Lots of dataaaa')
146+
file = tempfile.NamedTemporaryFile(delete=False)
147+
file.write(b'Lots of dataaaa')
148+
file.close()
149149
http_request = HttpRequest('GET', 'http://127.0.0.1/')
150150
headers = {'Content-Type': "multipart/form-data"}
151151
http_request.headers = headers
152-
with open(file_name, 'rb') as f:
152+
with open(file.name, 'rb') as f:
153153
form_data_content = {
154154
'fileContent': f,
155-
'fileName': file_name,
155+
'fileName': f.name,
156156
}
157157
http_request.set_formdata_body(form_data_content)
158158
http_retry = RetryPolicy(retry_total=1)
159159
pipeline = Pipeline(MockTransport(), [http_retry])
160160
pipeline.run(http_request)
161-
os.remove(file_name)
161+
os.unlink(f.name)

0 commit comments

Comments
 (0)