update

tests/integration/test_files.py

@@ -212,43 +212,43 @@ def create_volume(w, catalog, schema, volume):
         return ResourceWithCleanup(lambda: w.volumes.delete(res.full_name))
 
 
-# def test_files_api_upload_download(ucws, random):
-#     w = ucws
-#     schema = 'filesit-' + random()
-#     volume = 'filesit-' + random()
-#     with ResourceWithCleanup.create_schema(w, 'main', schema):
-#         with ResourceWithCleanup.create_volume(w, 'main', schema, volume):
-#             f = io.BytesIO(b"some text data")
-#             target_file = f'/Volumes/main/{schema}/{volume}/filesit-{random()}.txt'
-#             w.files.upload(target_file, f)
-#
-#             res = w.files.get_status(target_file)
-#             assert not res.is_dir
-#
-#             with w.files.download(target_file).contents as f:
-#                 assert f.read() == b"some text data"
-#
-#             w.files.delete(target_file)
-#
-#
-# def test_files_api_read_twice_from_one_download(ucws, random):
-#     w = ucws
-#     schema = 'filesit-' + random()
-#     volume = 'filesit-' + random()
-#     with ResourceWithCleanup.create_schema(w, 'main', schema):
-#         with ResourceWithCleanup.create_volume(w, 'main', schema, volume):
-#             f = io.BytesIO(b"some text data")
-#             target_file = f'/Volumes/main/{schema}/{volume}/filesit-{random()}.txt'
-#             w.files.upload(target_file, f)
-#
-#             res = w.files.download(target_file).contents
-#
-#             with res:
-#                 assert res.read() == b"some text data"
-#
-#             with pytest.raises(ValueError):
-#                 with res:
-#                     res.read()
+def test_files_api_upload_download(ucws, random):
+    w = ucws
+    schema = 'filesit-' + random()
+    volume = 'filesit-' + random()
+    with ResourceWithCleanup.create_schema(w, 'main', schema):
+        with ResourceWithCleanup.create_volume(w, 'main', schema, volume):
+            f = io.BytesIO(b"some text data")
+            target_file = f'/Volumes/main/{schema}/{volume}/filesit-{random()}.txt'
+            w.files.upload(target_file, f)
+
+            res = w.files.get_status(target_file)
+            assert not res.is_dir
+
+            with w.files.download(target_file).contents as f:
+                assert f.read() == b"some text data"
+
+            w.files.delete(target_file)
+
+
+def test_files_api_read_twice_from_one_download(ucws, random):
+    w = ucws
+    schema = 'filesit-' + random()
+    volume = 'filesit-' + random()
+    with ResourceWithCleanup.create_schema(w, 'main', schema):
+        with ResourceWithCleanup.create_volume(w, 'main', schema, volume):
+            f = io.BytesIO(b"some text data")
+            target_file = f'/Volumes/main/{schema}/{volume}/filesit-{random()}.txt'
+            w.files.upload(target_file, f)
+
+            res = w.files.download(target_file).contents
+
+            with res:
+                assert res.read() == b"some text data"
+
+            with pytest.raises(ValueError):
+                with res:
+                    res.read()
 
 
 @pytest.mark.benchmark