kleinkram 0.48.0.dev20250723090520__py3-none-any.whl → 0.58.0.dev20260121112512__py3-none-any.whl

kleinkram/wrappers.py

@@ -9,7 +9,8 @@ conversion to the internal representation
 from __future__ import annotations
 
 from pathlib import Path
-from typing import Collection, Any
+from typing import Any
+from typing import Collection
 from typing import Dict
 from typing import List
 from typing import Literal
@@ -66,9 +67,7 @@ def _args_to_mission_query(
     return MissionQuery(
         ids=[parse_uuid_like(_id) for _id in mission_ids or []],
         patterns=list(mission_names or []),
-        project_query=_args_to_project_query(
-            project_names=project_names, project_ids=project_ids
-        ),
+        project_query=_args_to_project_query(project_names=project_names, project_ids=project_ids),
     )
 
 
@@ -85,9 +84,7 @@ def _verify_string_sequence(arg_name: str, arg_value: Optional[Sequence[Any]]) -
         if not isinstance(arg_value, Sequence):
             raise TypeError(f"{arg_name} must be a Sequence, None, or empty array.")
         if isinstance(arg_value, str):
-            raise TypeError(
-                f"{arg_name} cannot be a string, but a sequence of strings."
-            )
+            raise TypeError(f"{arg_name} cannot be a string, but a sequence of strings.")
         for item in arg_value:
             if not isinstance(item, str):
                 raise TypeError(f"{arg_name} must contain strings only.")
@@ -358,15 +355,11 @@ def create_mission(
 
 
 def create_project(project_name: str, description: str) -> None:
-    kleinkram.api.routes._create_project(
-        AuthenticatedClient(), project_name, description
-    )
+    kleinkram.api.routes._create_project(AuthenticatedClient(), project_name, description)
 
 
 def update_file(file_id: IdLike) -> None:
-    kleinkram.core.update_file(
-        client=AuthenticatedClient(), file_id=parse_uuid_like(file_id)
-    )
+    kleinkram.core.update_file(client=AuthenticatedClient(), file_id=parse_uuid_like(file_id))
 
 
 def update_mission(mission_id: IdLike, metadata: Dict[str, str]) -> None:
@@ -399,48 +392,34 @@ def delete_file(file_id: IdLike) -> None:
     """\
     delete a single file by id
     """
-    file = kleinkram.api.routes.get_file(
-        AuthenticatedClient(), FileQuery(ids=[parse_uuid_like(file_id)])
-    )
-    kleinkram.api.routes._delete_files(
-        AuthenticatedClient(), file_ids=[file.id], mission_id=file.mission_id
-    )
+    file = kleinkram.api.routes.get_file(AuthenticatedClient(), FileQuery(ids=[parse_uuid_like(file_id)]))
+    kleinkram.api.routes._delete_files(AuthenticatedClient(), file_ids=[file.id], mission_id=file.mission_id)
 
 
 def delete_mission(mission_id: IdLike) -> None:
-    kleinkram.core.delete_mission(
-        client=AuthenticatedClient(), mission_id=parse_uuid_like(mission_id)
-    )
+    kleinkram.core.delete_mission(client=AuthenticatedClient(), mission_id=parse_uuid_like(mission_id))
 
 
 def delete_project(project_id: IdLike) -> None:
-    kleinkram.core.delete_project(
-        client=AuthenticatedClient(), project_id=parse_uuid_like(project_id)
-    )
+    kleinkram.core.delete_project(client=AuthenticatedClient(), project_id=parse_uuid_like(project_id))
 
 
 def get_file(file_id: IdLike) -> File:
     """\
     get a file by its id
     """
-    return kleinkram.api.routes.get_file(
-        AuthenticatedClient(), FileQuery(ids=[parse_uuid_like(file_id)])
-    )
+    return kleinkram.api.routes.get_file(AuthenticatedClient(), FileQuery(ids=[parse_uuid_like(file_id)]))
 
 
 def get_mission(mission_id: IdLike) -> Mission:
     """\
     get a mission by its id
     """
-    return kleinkram.api.routes.get_mission(
-        AuthenticatedClient(), MissionQuery(ids=[parse_uuid_like(mission_id)])
-    )
+    return kleinkram.api.routes.get_mission(AuthenticatedClient(), MissionQuery(ids=[parse_uuid_like(mission_id)]))
 
 
 def get_project(project_id: IdLike) -> Project:
     """\
     get a project by its id
     """
-    return kleinkram.api.routes.get_project(
-        AuthenticatedClient(), ProjectQuery(ids=[parse_uuid_like(project_id)])
-    )
+    return kleinkram.api.routes.get_project(AuthenticatedClient(), ProjectQuery(ids=[parse_uuid_like(project_id)]))

{kleinkram-0.48.0.dev20250723090520.dist-info → kleinkram-0.58.0.dev20260121112512.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kleinkram
-Version: 0.48.0.dev20250723090520
+Version: 0.58.0.dev20260121112512
 Summary: give me your bags
 Author: Cyrill Püntener, Dominique Garmier, Johann Schwabe
 Author-email: pucyril@ethz.ch, dgarmier@ethz.ch, jschwab@ethz.ch
@@ -16,13 +16,14 @@ Requires-Python: >=3.8
 Description-Content-Type: text/markdown
 Requires-Dist: boto3
 Requires-Dist: botocore
+Requires-Dist: click
 Requires-Dist: httpx
 Requires-Dist: python-dateutil
 Requires-Dist: pyyaml
+Requires-Dist: requests
 Requires-Dist: rich
 Requires-Dist: tqdm
 Requires-Dist: typer
-Requires-Dist: click
 
 # Kleinkram: CLI
 
@@ -76,7 +77,7 @@ Instead of downloading files from a specified mission you can download arbitrary
 klein download --dest out *id1* *id2* *id3*
 ```
 
-For more information consult the [documentation](https://docs.datasets.leggedrobotics.com/usage/python/getting-started.html).
+For more information consult the [documentation](https://docs.datasets.leggedrobotics.com//usage/python/setup).
 
 ## Development
 
@@ -118,7 +119,7 @@ pytest
 ```
 For the latter you need to have an instance of the backend running locally.
 See instructions in the root of the repository for this.
-On top of that these tests require particular files to be present in the `cli/data/testing` directory.
-To see the exact files that are required, see `cli/testing/backend_fixtures.py`.
+On top of that these tests require particular files to be present in the `cli/tests/data` directory.
+These files are automatically generated by the `cli/tests/generate_test_data.py` script.
 
 You also need to make sure to be logged in with the cli with `klein login`.

kleinkram-0.58.0.dev20260121112512.dist-info/RECORD

@@ -0,0 +1,53 @@
+kleinkram/__init__.py,sha256=xIJqTJw2kbCGryGlCeAdpmtR1FTxmrW1MklUNQEaj74,1061
+kleinkram/__main__.py,sha256=B9RiZxfO4jpCmWPUHyKJ7_EoZlEG4sPpH-nz7T_YhhQ,125
+kleinkram/_version.py,sha256=QYJyRTcqFcJj4qWYpqs7WcoOP6jxDMqyvxLY-cD6KcE,129
+kleinkram/auth.py,sha256=zbjSIPBS5xzLzo75Ou2dgUluXLwBRxKxBdSwtIQdA5g,6835
+kleinkram/config.py,sha256=fCRyDdBKXbH7HQ_Kc3YGTamqGL9zkKLRIIQTQDql0Ec,7273
+kleinkram/core.py,sha256=AhFIxm2jjzRUlGXrOJksebYHOyuioAcbpCx0WmbIpbs,9790
+kleinkram/errors.py,sha256=Bevo8LNPlM46C9JY_uj1Z3FSvPJ051Vyd0M9IYENLtc,1140
+kleinkram/main.py,sha256=BTE0mZN__xd46wBhFi6iBlK9eGGQvJ1LdUMsbnysLi0,172
+kleinkram/models.py,sha256=YB3pcZGJ5nXeZ9gLTbpaaWsUCnvaDIcnvpXUnyWZAaY,2824
+kleinkram/printing.py,sha256=U8I43Ju3orA_jKqF5f-vmgl3cC8Ffjo5m2fgChmY6ik,19163
+kleinkram/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kleinkram/types.py,sha256=nfDjj8TB1Jn5vqO0Xg6qhLOuKom9DDhe62BrngqnVGM,185
+kleinkram/utils.py,sha256=iPZXUbiG4cYfkL3CG-Rs-dwVovNYkgybfAAvD0oamBY,6725
+kleinkram/wrappers.py,sha256=TprQWvEw-8QsgcDNsQxjBkcEwVFQyfmclTpt8wcZ4tA,12967
+kleinkram/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kleinkram/api/client.py,sha256=j-ZWBsHI-PQOk86RtDfqunaSVyKqCnKdqPMy0tCbdNI,5840
+kleinkram/api/deser.py,sha256=94O1ZeP0kW-LaKaZeP4o1PQLeHbxxZ_C7nU2FzunntY,10077
+kleinkram/api/file_transfer.py,sha256=j1iINYFV4iXMjAz59eM6y8G35Uq4GpPgqf1T0z73Jzs,19710
+kleinkram/api/pagination.py,sha256=NRVGorgNthKi4t_H9lOn87rDebx33WVOvTadA6-4QXw,1693
+kleinkram/api/query.py,sha256=WR_j0WRt9LDW9uyqc8pEdjmgN5eOLVQi1y3LSOh2Y9A,3498
+kleinkram/api/routes.py,sha256=zKgjEiuFP-v06pBnN87yLcdYyuAV4Jx-hbEf44qu_e4,17461
+kleinkram/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kleinkram/cli/_action.py,sha256=Gb3kIs4jcutINX4pmrnOjhetcEpeAGVs3llX0GR9gz0,4911
+kleinkram/cli/_download.py,sha256=n4d9y8ViTLnzrfn8MLaM_6Jlsuh9_Y_F0I-dfAhZEqI,2318
+kleinkram/cli/_endpoint.py,sha256=dFLTaARXP96ide4J4cOHx5qHc3_iFYGJmhz-fKLGbP8,1785
+kleinkram/cli/_file.py,sha256=CiTmy3Di7lVBH7mfsNbWpiwyDl8WGw0xi8MV0-EmAV4,2897
+kleinkram/cli/_file_validator.py,sha256=6SkQGNysq1ydZimfv7MmAqyRP7xiKwaWcoK7R-tL_M4,4552
+kleinkram/cli/_list.py,sha256=jDZlG76b0KWr8imBQdA--V2DC6O3j6jrMd6qyJm6cTw,3028
+kleinkram/cli/_mission.py,sha256=nZjk9gHsTCMkmwbtNMIb9fxfySyggc3nGJdTC0t45G4,5759
+kleinkram/cli/_project.py,sha256=C6mnNJM4oloFto9wDY4zgwdmXNsqR9ErbgZeFdhCENk,3338
+kleinkram/cli/_run.py,sha256=TRx08DPB4PRGIqZl2CjHEDODgQac3Jr7iNQqBN9A7nE,8008
+kleinkram/cli/_upload.py,sha256=PkD9O4ApBo5VUxdt70pu0XdJqZ88FrY_vZWQzTtN0pg,4030
+kleinkram/cli/_verify.py,sha256=j-bRmFqW2j88ZwVOSFEy0Dgok-JY0nAPkcTAOwgxQyY,3211
+kleinkram/cli/app.py,sha256=fpme23Y6pd2LeTadcFYBNfyjDn4AEGltSJVg79GCDAI,8940
+kleinkram/cli/error_handling.py,sha256=ZFcJhPXWITG1cGpnsiuDVVehjIqO5pbz47zWdL1QcQs,1907
+tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+tests/backend_fixtures.py,sha256=Kjr1W9NYQ1yCOqLheAIILYeTo9B59WW1M0FMr2MPKew,2308
+tests/conftest.py,sha256=0kbRbC4AJhwjOlZYdwkzQHLBmcrU5pB3o76PImunP4c,102
+tests/generate_test_data.py,sha256=ujp_CoSUr_v9ZrTrYDIjlsOcxqpsc2pzE307z_pS1qY,12116
+tests/test_config.py,sha256=RUIUEqTnAKg8EyqXVGohwfm9jtuaWo1hWqVxZMV6_bU,5843
+tests/test_core.py,sha256=qIPYuMpI3yh2G6OEqqqjRCWlvaM9bd9VC0bB5kI7PJs,5473
+tests/test_end_to_end.py,sha256=mySbCPZPmrszo2AjD18TDitrPAJJ73oz7DaBpTXIkiI,3337
+tests/test_error_handling.py,sha256=qPSMKF1qsAHyUME0-krxbIrk38iGKkhAyAah-KwN4NE,1300
+tests/test_fixtures.py,sha256=SaOe7naSqGH3nivdbwyNo1tcYOg_XU-NJy1cFv3__2o,1053
+tests/test_printing.py,sha256=Kfg0XjGV7ymvI4RgHtNc8wiwpqB3VG0wc1XYWO1ld_k,2217
+tests/test_query.py,sha256=fExmCKXLA7-9j2S2sF_sbvRX_2s6Cp3a7OTcqE25q9g,3864
+tests/test_utils.py,sha256=_DGxsQ60YEKeLNmrwLdJbevSFhWgIK9f38fH02VA8MI,4762
+tests/test_wrappers.py,sha256=Jr7qFbkEBbs21S-SkwF8A5cH-OcFsVz0Y0kBc6gGBuY,2547
+kleinkram-0.58.0.dev20260121112512.dist-info/METADATA,sha256=uZsYFjw4P-9121f4ImhNHVhjqWaAJQIR_OWNu2uNk0U,2862
+kleinkram-0.58.0.dev20260121112512.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+kleinkram-0.58.0.dev20260121112512.dist-info/entry_points.txt,sha256=SaB2l5aqhSr8gmaMw2kvQU90a8Bnl7PedU8cWYxkfYo,46
+kleinkram-0.58.0.dev20260121112512.dist-info/top_level.txt,sha256=G1Lj9vHAtZn402Ukkrfll-6BCmnDNy_HVtWeNvXzdDA,16
+kleinkram-0.58.0.dev20260121112512.dist-info/RECORD,,

{kleinkram-0.48.0.dev20250723090520.dist-info → kleinkram-0.58.0.dev20260121112512.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{kleinkram-0.48.0.dev20250723090520.dist-info → kleinkram-0.58.0.dev20260121112512.dist-info}/top_level.txt

@@ -1,3 +1,2 @@
 kleinkram
-testing
 tests

{testing → tests}/backend_fixtures.py

@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+import subprocess
+import sys
 import time
 from pathlib import Path
 from secrets import token_hex
@@ -14,7 +16,7 @@ from kleinkram import list_projects
 from kleinkram import upload
 
 # we expect the mission files to be in this folder that is not commited to the repo
-DATA_PATH = Path(__file__).parent.parent / "data" / "testing"
+DATA_PATH = Path(__file__).parent / "data"
 DATA_FILES = [
     DATA_PATH / "10_KB.bag",
     DATA_PATH / "50_KB.bag",
@@ -26,7 +28,7 @@ DATA_FILES = [
 PROJECT_DESCRIPTION = "This is a test project"
 
 
-WAIT_BEOFORE_DELETION = 5
+WAIT_BEFORE_DELETION = 5
 
 
 @pytest.fixture(scope="session")
@@ -38,7 +40,7 @@ def project():
 
     yield project
 
-    time.sleep(WAIT_BEOFORE_DELETION)
+    time.sleep(WAIT_BEFORE_DELETION)
     delete_project(project.id)
 
 
@@ -63,3 +65,25 @@ def empty_mission(project):
     mission = list_missions(project_ids=[project.id], mission_names=[mission_name])[0]
 
     yield mission
+
+
+@pytest.fixture(scope="session", autouse=True)
+def auto_login():
+    """
+    Automatically logs in using the CLI before running any tests.
+    """
+    try:
+        # Run: python3 -m kleinkram login --user 1
+        result = subprocess.run(
+            [sys.executable, "-m", "kleinkram", "login", "--user", "1"],
+            cwd=str(Path(__file__).parent.parent),  # Run from cli root
+            capture_output=True,
+            text=True,
+        )
+        if result.returncode != 0:
+            pytest.fail(
+                f"Failed to auto-login. Return code: {result.returncode}\nStdout: {result.stdout}\nStderr: {result.stderr}"
+            )
+
+    except Exception as e:
+        pytest.fail(f"Failed to auto-login: {e}")

tests/conftest.py

@@ -3,5 +3,5 @@ from __future__ import annotations
 import pytest
 
 pytest_plugins = [
-    "testing.backend_fixtures",
+    "tests.backend_fixtures",
 ]

tests/generate_test_data.py

@@ -0,0 +1,322 @@
+from __future__ import annotations
+
+import os
+import struct
+import time
+
+from rosbags.rosbag1 import Writer
+
+
+def serialize_string(data):
+    """Serialize a string for ROS1 (std_msgs/String)."""
+    encoded = data.encode("utf-8")
+    return struct.pack("<I", len(encoded)) + encoded
+
+
+def serialize_time(secs, nsecs):
+    return struct.pack("<II", secs, nsecs)
+
+
+def serialize_header(seq, secs, nsecs, frame_id):
+    return struct.pack("<I", seq) + serialize_time(secs, nsecs) + serialize_string(frame_id)
+
+
+def serialize_log(seq, secs, nsecs, frame_id, level, name, msg, file, function, line, topics):
+    # rosgraph_msgs/Log
+    # Header header
+    # byte level
+    # string name
+    # string msg
+    # string file
+    # string function
+    # uint32 line
+    # string[] topics
+    data = serialize_header(seq, secs, nsecs, frame_id)
+    data += struct.pack("<b", level)
+    data += serialize_string(name)
+    data += serialize_string(msg)
+    data += serialize_string(file)
+    data += serialize_string(function)
+    data += struct.pack("<I", line)
+    data += struct.pack("<I", len(topics))
+    for t in topics:
+        data += serialize_string(t)
+    return data
+
+
+def serialize_temperature(seq, secs, nsecs, frame_id, temp, variance):
+    # sensor_msgs/Temperature
+    # Header header
+    # float64 temperature
+    # float64 variance
+    data = serialize_header(seq, secs, nsecs, frame_id)
+    data += struct.pack("<dd", temp, variance)
+    return data
+
+
+def serialize_time_reference(seq, secs, nsecs, frame_id, ref_secs, ref_nsecs, source):
+    # sensor_msgs/TimeReference
+    # Header header
+    # time time_ref
+    # string source
+    data = serialize_header(seq, secs, nsecs, frame_id)
+    data += serialize_time(ref_secs, ref_nsecs)
+    data += serialize_string(source)
+    return data
+
+
+def serialize_twist_stamped(seq, secs, nsecs, frame_id, linear, angular):
+    # geometry_msgs/TwistStamped
+    # Header header
+    # Twist twist (Vector3 linear, Vector3 angular)
+    data = serialize_header(seq, secs, nsecs, frame_id)
+    data += struct.pack("<ddd", linear[0], linear[1], linear[2])
+    data += struct.pack("<ddd", angular[0], angular[1], angular[2])
+    return data
+
+
+def serialize_tf_message(transforms):
+    # tf2_msgs/TFMessage
+    # geometry_msgs/TransformStamped[] transforms
+    data = struct.pack("<I", len(transforms))
+    for t in transforms:
+        # TransformStamped
+        # Header header
+        # string child_frame_id
+        # Transform transform (Vector3 translation, Quaternion rotation)
+        data += serialize_header(t["seq"], t["secs"], t["nsecs"], t["frame_id"])
+        data += serialize_string(t["child_frame_id"])
+        data += struct.pack("<ddd", t["tx"], t["ty"], t["tz"])
+        data += struct.pack("<dddd", t["rx"], t["ry"], t["rz"], t["rw"])
+    return data
+
+
+def generate_bag(filename, target_size):
+    # Adjust payload to be smaller for small files
+    payload_size = 1024
+    if target_size > 1024 * 1024:
+        payload_size = 1024 * 1024
+
+    if target_size < 2000:  # Very small files
+        payload_size = 100
+
+    payload = "x" * payload_size
+    serialized_msg = serialize_string(payload)
+
+    # Calculate number of messages needed
+    # Approximate overhead per message in bag file is ~30-50 bytes (record header) + connection ref
+    # We'll assume overhead is small compared to payload for large files, but significant for small ones.
+    # We'll just write until we think we are close.
+
+    msg_size = len(serialized_msg) + 50  # rough estimate including record headers
+    num_msgs = int(target_size / msg_size) + 1
+    if num_msgs < 1:
+        num_msgs = 1
+
+    print(f"Generating {filename} (~{target_size} bytes) with {num_msgs} messages of payload {payload_size}...")
+
+    if os.path.exists(filename):
+        os.remove(filename)
+
+    with Writer(filename) as writer:
+        # Add a connection
+        # msg_def for std_msgs/String is just "string data"
+        conn = writer.add_connection(
+            topic="/test_topic",
+            msgtype="std_msgs/msg/String",
+            msgdef="string data",
+            md5sum="992ce8a1687cec8c8bd883ec73ca41d1",
+        )
+        timestamp = 1000
+        for i in range(num_msgs):
+            writer.write(conn, timestamp + i, serialized_msg)
+
+
+def generate_frontend_bag(filename):
+    print(f"Generating frontend test bag: {filename}")
+    if os.path.exists(filename):
+        os.remove(filename)
+
+    with Writer(filename) as writer:
+        # 1. rosgraph_msgs/Log
+        conn_log = writer.add_connection(
+            topic="/rosout",
+            msgtype="rosgraph_msgs/msg/Log",
+            msgdef="Header header\nbyte level\nstring name\nstring msg\nstring file\nstring function\nuint32 "
+            "line\nstring[] topics\n================================================================================\n"
+            "MSG: std_msgs/Header\nuint32 seq\ntime stamp\nstring frame_id",
+            md5sum="acffd30cd6b6de30f120938c17c593fb",
+        )
+        # 2. sensor_msgs/Temperature
+        conn_temp = writer.add_connection(
+            topic="/sensors/temperature",
+            msgtype="sensor_msgs/msg/Temperature",
+            msgdef="Header header\nfloat64 temperature\nfloat64 variance"
+            "\n================================================================================\n"
+            "MSG: std_msgs/Header\nuint32 seq\ntime stamp\nstring frame_id",
+            md5sum="ff71b307acdbe7c871a5a6d7edce2f6e",
+        )
+        # 3. sensor_msgs/TimeReference
+        conn_time = writer.add_connection(
+            topic="/time_ref",
+            msgtype="sensor_msgs/msg/TimeReference",
+            msgdef="Header header\ntime time_ref\nstring source\n"
+            "================================================================================\n"
+            "MSG: std_msgs/Header\nuint32 seq\ntime stamp\nstring frame_id",
+            md5sum="fded64a0265108ba86c3d38fb11c0c16",
+        )
+        # 4. geometry_msgs/TwistStamped
+        conn_twist = writer.add_connection(
+            topic="/cmd_vel",
+            msgtype="geometry_msgs/msg/TwistStamped",
+            msgdef="Header header\ngeometry_msgs/Twist twist\n"
+            "================================================================================\n"
+            "MSG: std_msgs/Header\nuint32 seq\ntime stamp\nstring frame_id\n"
+            "================================================================================\n"
+            "MSG: geometry_msgs/Twist\nVector3 linear\nVector3 angular\n"
+            "================================================================================\n"
+            "MSG: geometry_msgs/Vector3\nfloat64 x\nfloat64 y\nfloat64 z",
+            md5sum="98d34b0043a2093cf9d9345ab6eef12e",
+        )
+        # 5. tf2_msgs/TFMessage
+        conn_tf = writer.add_connection(
+            topic="/tf",
+            msgtype="tf2_msgs/msg/TFMessage",
+            msgdef="geometry_msgs/TransformStamped[] transforms\n"
+            "================================================================================\n"
+            "MSG: geometry_msgs/TransformStamped\nHeader header\nstring child_frame_id\n"
+            "geometry_msgs/Transform transform\n"
+            "================================================================================\n"
+            "MSG: std_msgs/Header\nuint32 seq\ntime stamp\nstring frame_id\n"
+            "================================================================================\n"
+            "MSG: geometry_msgs/Transform\ngeometry_msgs/Vector3 translation\n"
+            "geometry_msgs/Quaternion rotation\n"
+            "================================================================================\n"
+            "MSG: geometry_msgs/Vector3\nfloat64 x\nfloat64 y\nfloat64 z\n"
+            "================================================================================\n"
+            "MSG: geometry_msgs/Quaternion\nfloat64 x\nfloat64 y\nfloat64 z\nfloat64 w",
+            md5sum="94810edda583a504dfda3829e70d7eec",
+        )
+
+        # Write messages
+        for i in range(100):
+            timestamp = 1000 + i * 100000000  # 100ms steps
+            secs = int(timestamp / 1000000000)
+            nsecs = timestamp % 1000000000
+
+            # Log
+            writer.write(
+                conn_log,
+                timestamp,
+                serialize_log(
+                    i,
+                    secs,
+                    nsecs,
+                    "",
+                    2,
+                    "test_node",
+                    f"Log message {i}",
+                    "test.cpp",
+                    "main",
+                    i,
+                    ["/rosout"],
+                ),
+            )
+
+            # Temperature (sine wave)
+            import math
+
+            temp = 20.0 + 5.0 * math.sin(i * 0.1)
+            writer.write(
+                conn_temp,
+                timestamp,
+                serialize_temperature(i, secs, nsecs, "sensor_frame", temp, 0.1),
+            )
+
+            # TimeReference
+            writer.write(
+                conn_time,
+                timestamp,
+                serialize_time_reference(i, secs, nsecs, "time_frame", secs, nsecs, "GPS"),
+            )
+
+            # TwistStamped (circle)
+            writer.write(
+                conn_twist,
+                timestamp,
+                serialize_twist_stamped(i, secs, nsecs, "base_link", [1.0, 0.0, 0.0], [0.0, 0.0, 0.5]),
+            )
+
+            # TF
+            writer.write(
+                conn_tf,
+                timestamp,
+                serialize_tf_message(
+                    [
+                        {
+                            "seq": i,
+                            "secs": secs,
+                            "nsecs": nsecs,
+                            "frame_id": "map",
+                            "child_frame_id": "base_link",
+                            "tx": i * 0.1,
+                            "ty": 0.0,
+                            "tz": 0.0,
+                            "rx": 0.0,
+                            "ry": 0.0,
+                            "rz": 0.0,
+                            "rw": 1.0,
+                        }
+                    ]
+                ),
+            )
+
+
+def main():
+    data_dir = os.path.join(os.path.dirname(__file__), "data")
+    os.makedirs(data_dir, exist_ok=True)
+
+    files = {
+        "10_KB.bag": 10 * 1024,
+        "50_KB.bag": 50 * 1024,
+        "1_MB.bag": 1 * 1024 * 1024,
+        "17_MB.bag": 17 * 1024 * 1024,
+        "125_MB.bag": 125 * 1024 * 1024,
+    }
+
+    for filename, size in files.items():
+        filepath = os.path.join(data_dir, filename)
+        generate_bag(filepath, size)
+
+    # Generate backend fixtures
+    backend_fixtures_dir = os.path.join(os.path.dirname(__file__), "../../backend/tests/fixtures")
+    os.makedirs(backend_fixtures_dir, exist_ok=True)
+    generate_bag(os.path.join(backend_fixtures_dir, "test.bag"), 10 * 1024)
+    generate_bag(os.path.join(backend_fixtures_dir, "to_delete.bag"), 10 * 1024)
+    generate_bag(os.path.join(backend_fixtures_dir, "file1.bag"), 10 * 1024)
+    generate_bag(os.path.join(backend_fixtures_dir, "file2.bag"), 10 * 1024)
+    generate_bag(os.path.join(backend_fixtures_dir, "move_me.bag"), 10 * 1024)
+    generate_bag(os.path.join(backend_fixtures_dir, "state_test.bag"), 10 * 1024)
+
+    # Generate backend dummy MCAP and YAML
+    with open(os.path.join(backend_fixtures_dir, "config.yaml"), "w") as f:
+        f.write("test: true\nvalue: 123\n")
+    with open(os.path.join(backend_fixtures_dir, "config.yml"), "w") as f:
+        f.write("test: true\nvalue: 123\n")
+    with open(os.path.join(backend_fixtures_dir, "test.mcap"), "wb") as f:
+        f.write(b"\x89MCAP\x30\r\n")
+
+    generate_frontend_bag(os.path.join(data_dir, "frontend_test.bag"))
+
+    # Generate dummy MCAP and YAML
+    with open(os.path.join(data_dir, "test.yaml"), "w") as f:
+        f.write("test: true\nvalue: 123\n")
+
+    with open(os.path.join(data_dir, "test.mcap"), "wb") as f:
+        f.write(b"\x89MCAP\x30\r\n")  # Minimal magic bytes
+
+    print("Done.")
+
+
+if __name__ == "__main__":
+    main()

tests/test_config.py

@@ -72,9 +72,7 @@ def test_load_config_default(config_path):
     assert config.selected_endpoint == get_env().value
 
 
-def test_load_default_config_with_env_var_api_key_specified(
-    config_path, set_api_key_env
-):
+def test_load_default_config_with_env_var_api_key_specified(config_path, set_api_key_env):
     assert set_api_key_env is None
 
     config = _load_config(path=config_path)
@@ -87,9 +85,7 @@ def test_load_default_config_with_env_var_api_key_specified(
     assert not config_path.exists()
 
 
-def test_load_default_config_with_env_var_endpoints_specified(
-    config_path, set_endpoint_env
-):
+def test_load_default_config_with_env_var_endpoints_specified(config_path, set_endpoint_env):
     assert set_endpoint_env is None
     config = _load_config(path=config_path)