outputs properly formmated json dump

mlibrary · Sep 27, 2024 · cd49e3d · cd49e3d
1 parent 24656eb
commit cd49e3d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/aim/cli/digifeeds.py b/aim/cli/digifeeds.py
@@ -2,6 +2,8 @@
 from aim.digifeeds.add_to_db import add_to_db as add_to_digifeeds_db
 from aim.digifeeds.list_barcodes_in_bucket import list_barcodes_in_bucket
 from aim.digifeeds.database import models, main
+import json
+import sys
 
 
 app = typer.Typer()
@@ -21,4 +23,4 @@ def load_statuses():
 
 @app.command()
 def list_barcodes_in_input_bucket():
-    print(list_barcodes_in_bucket(), end="")
+    json.dump(list_barcodes_in_bucket(), sys.stdout)
diff --git a/tests/cli/test_digifeeds.py b/tests/cli/test_digifeeds.py
@@ -70,4 +70,4 @@ def test_list_barcodes_in_input_bucket(mocker):
     result = runner.invoke(app, ["digifeeds", "list-barcodes-in-input-bucket"])
     assert list_barcodes_mock.call_count == 1
     assert result.exit_code == 0
-    assert "['barcode1', 'barcode2']" == result.stdout
+    assert '["barcode1", "barcode2"]' == result.stdout