Add async and sync iterable put tests

Author: kylebarron

tests/test_put.py

@@ -1,3 +1,5 @@
+import itertools
+
 import pytest
 
 import obstore as obs
@@ -39,3 +41,33 @@ def test_put_mode():
         obs.put(store, "file1.txt", b"foo", mode="create")
 
     assert obs.get(store, "file1.txt").bytes() == b"bar"
+
+
+@pytest.mark.asyncio
+async def test_put_async_iterable():
+    store = MemoryStore()
+
+    data = b"the quick brown fox jumps over the lazy dog," * 50_000
+    path = "big-data.txt"
+
+    await obs.put_async(store, path, data)
+
+    resp = await obs.get_async(store, path)
+    stream = resp.stream(min_chunk_size=0)
+    new_path = "new-path.txt"
+    await obs.put_async(store, new_path, stream)
+
+    assert obs.get(store, new_path).bytes() == data
+
+
+def test_put_sync_iterable():
+    store = MemoryStore()
+
+    b = b"the quick brown fox jumps over the lazy dog,"
+    iterator = itertools.repeat(b, 50_000)
+    data = b * 50_000
+    path = "big-data.txt"
+
+    obs.put(store, path, iterator)
+
+    assert obs.get(store, path).bytes() == data