fox-it · yunzheng · Aug 11, 2025 · Aug 5, 2025 · Aug 6, 2025
diff --git a/.gitattributes b/.gitattributes
@@ -0,0 +1 @@
+tests/_data/** filter=lfs diff=lfs merge=lfs -text
diff --git a/.gitignore b/.gitignore
@@ -8,6 +8,6 @@ __pycache__/
 .pytest_cache/
 
 flow/record/version.py
-tests/docs/api
-tests/docs/build
+tests/_docs/api
+tests/_docs/build
 .tox/
diff --git a/tests/selector_explain_example.py → examples/selectors.py b/tests/selector_explain_example.py → examples/selectors.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -125,7 +125,7 @@ select = [
 ignore = ["E203", "B904", "UP024", "ANN002", "ANN003", "ANN204", "ANN401", "SIM105", "TRY003", "TRY400"]
 
 [tool.ruff.lint.per-file-ignores]
-"tests/docs/**" = ["INP001"]
+"tests/_docs/**" = ["INP001"]
 
 [tool.ruff.lint.isort]
 known-first-party = ["flow.record"]

diff --git a/tests/_data/.gitkeep b/tests/_data/.gitkeep
diff --git a/tests/docs/Makefile → tests/_docs/Makefile b/tests/docs/Makefile → tests/_docs/Makefile
diff --git a/tests/docs/conf.py → tests/_docs/conf.py b/tests/docs/conf.py → tests/_docs/conf.py
diff --git a/tests/docs/index.rst → tests/_docs/index.rst b/tests/docs/index.rst → tests/_docs/index.rst
diff --git a/tests/adapter/__init__.py b/tests/adapter/__init__.py
diff --git a/tests/test_avro.py → tests/adapter/test_avro.py b/tests/test_avro.py → tests/adapter/test_avro.py
@@ -5,9 +5,10 @@
 
 import pytest
 
-from flow.record import RecordDescriptor, RecordReader
+from flow.record import RecordDescriptor, RecordReader, RecordWriter
 from flow.record.adapter.avro import AvroReader, AvroWriter
 from flow.record.base import HAS_AVRO
+from tests._utils import generate_plain_records
 
 if TYPE_CHECKING:
     from collections.abc import Iterator
@@ -68,3 +69,51 @@ def test_avrostream_filelike_object(tmp_path: Path) -> None:
         assert rec.name == f"record{index}"
         assert rec.foo == "bar"
         assert rec.bar == "baz"
+
+
+def test_avro_adapter(tmpdir: Path) -> None:
+    json_file = tmpdir.join("records.avro")
+    record_adapter_path = f"avro://{json_file}"
+    writer = RecordWriter(record_adapter_path)
+    nr_records = 1337
+
+    for record in generate_plain_records(nr_records):
+        writer.write(record)
+    writer.flush()
+
+    nr_received_records = 0
+    reader = RecordReader(record_adapter_path)
+    for _ in reader:
+        nr_received_records += 1
+
+    assert nr_records == nr_received_records
+
+
+def test_avro_adapter_contextmanager(tmpdir: Path) -> None:
+    json_file = tmpdir.join("records.avro")
+    record_adapter_path = f"avro://{json_file}"
+    with RecordWriter(record_adapter_path) as writer:
+        nr_records = 1337
+        for record in generate_plain_records(nr_records):
+            writer.write(record)
+
+    nr_received_records = 0
+    with RecordReader(record_adapter_path) as reader:
+        for _ in reader:
+            nr_received_records += 1
+
+        assert nr_records == nr_received_records
+
+
+def test_avro_adapter_empty(tmpdir: Path) -> None:
+    json_file = tmpdir.join("records.avro")
+    record_adapter_path = f"avro://{json_file}"
+    with RecordWriter(record_adapter_path):
+        pass
+
+    nr_received_records = 0
+    with RecordReader(record_adapter_path) as reader:
+        for _ in reader:
+            nr_received_records += 1
+
+        assert nr_received_records == 0
diff --git a/tests/test_csv_adapter.py → tests/adapter/test_csv.py b/tests/test_csv_adapter.py → tests/adapter/test_csv.py
diff --git a/tests/test_elastic_adapter.py → tests/adapter/test_elastic.py b/tests/test_elastic_adapter.py → tests/adapter/test_elastic.py
diff --git a/tests/test_json_record_adapter.py → tests/adapter/test_json.py b/tests/test_json_record_adapter.py → tests/adapter/test_json.py
@@ -6,8 +6,7 @@
 import pytest
 
 from flow.record import RecordReader, RecordWriter
-
-from ._utils import generate_records
+from tests._utils import generate_records
 
 if TYPE_CHECKING:
     from pathlib import Path

diff --git a/tests/test_adapter_line.py → tests/adapter/test_line.py b/tests/test_adapter_line.py → tests/adapter/test_line.py
diff --git a/tests/test_splunk_adapter.py → tests/adapter/test_splunk.py b/tests/test_splunk_adapter.py → tests/adapter/test_splunk.py
diff --git a/tests/test_sqlite_duckdb_adapter.py → tests/adapter/test_sqlite_duckdb.py b/tests/test_sqlite_duckdb_adapter.py → tests/adapter/test_sqlite_duckdb.py
diff --git a/tests/test_adapter_text.py → tests/adapter/test_text.py b/tests/test_adapter_text.py → tests/adapter/test_text.py
diff --git a/tests/test_xlsx_adapter.py → tests/adapter/test_xlsx.py b/tests/test_xlsx_adapter.py → tests/adapter/test_xlsx.py
diff --git a/tests/fieldtypes/__init__.py b/tests/fieldtypes/__init__.py
diff --git a/tests/test_fieldtypes.py → tests/fieldtypes/test_fieldtypes.py b/tests/test_fieldtypes.py → tests/fieldtypes/test_fieldtypes.py
diff --git a/tests/test_fieldtype_ip.py → tests/fieldtypes/test_ip.py b/tests/test_fieldtype_ip.py → tests/fieldtypes/test_ip.py
diff --git a/tests/packer/__init__.py b/tests/packer/__init__.py
diff --git a/tests/test_json_packer.py → tests/packer/test_json_packer.py b/tests/test_json_packer.py → tests/packer/test_json_packer.py
diff --git a/tests/test_packer.py → tests/packer/test_packer.py b/tests/test_packer.py → tests/packer/test_packer.py
diff --git a/tests/record/__init__.py b/tests/record/__init__.py
diff --git a/tests/test_record_adapter.py → tests/record/test_adapter.py b/tests/test_record_adapter.py → tests/record/test_adapter.py
@@ -28,8 +28,7 @@
     ZSTD_MAGIC,
 )
 from flow.record.selector import CompiledSelector, Selector
-
-from ._utils import generate_records
+from tests._utils import generate_records
 
 if TYPE_CHECKING:
     from pathlib import Path

diff --git a/tests/test_record_descriptor.py → tests/record/test_descriptor.py b/tests/test_record_descriptor.py → tests/record/test_descriptor.py
diff --git a/tests/test_multi_timestamp.py → tests/record/test_multi_timestamp.py b/tests/test_multi_timestamp.py → tests/record/test_multi_timestamp.py
diff --git a/tests/test_record.py → tests/record/test_record.py b/tests/test_record.py → tests/record/test_record.py
diff --git a/tests/selector/__init__.py b/tests/selector/__init__.py
diff --git a/tests/test_compiled_selector.py → tests/selector/test_compiled.py b/tests/test_compiled_selector.py → tests/selector/test_compiled.py
diff --git a/tests/test_selector.py → tests/selector/test_selectors.py b/tests/test_selector.py → tests/selector/test_selectors.py
diff --git a/tests/standalone_test.py b/tests/standalone_test.py
diff --git a/tests/test_avro_adapter.py b/tests/test_avro_adapter.py
diff --git a/tests/test_regression.py → tests/test_regressions.py b/tests/test_regression.py → tests/test_regressions.py
diff --git a/tests/tools/__init__.py b/tests/tools/__init__.py
diff --git a/tests/test_rdump.py → tests/tools/test_rdump.py b/tests/test_rdump.py → tests/tools/test_rdump.py
diff --git a/tox.ini b/tox.ini
@@ -57,12 +57,12 @@ deps =
     sphinx-design
     furo
 commands =
-    make -C tests/docs clean
-    make -C tests/docs html
+    make -C tests/_docs clean
+    make -C tests/_docs html
 
 [testenv:docs-linkcheck]
 allowlist_externals = make
 deps = {[testenv:docs-build]deps}
 commands =
-    make -C tests/docs clean
-    make -C tests/docs linkcheck
+    make -C tests/_docs clean
+    make -C tests/_docs linkcheck
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		tests/_data/** filter=lfs diff=lfs merge=lfs -text