add tests

Author: asonnenschein

tests/integration/test_orders_cli.py

@@ -514,6 +514,49 @@ def test_cli_orders_request_product_bundle_incompatible(mock_bundles, invoke):
     assert result.exit_code == 2
 
 
+@pytest.mark.parametrize(
+    "bundle, fallback_bundle",
+    [
+        ('analytic_8b_udm2', 'analytic_udm2'),
+        ('analytic_8b_udm2', 'analytic_udm2,analytic_3b_udm2'),
+    ])
+def test_cli_orders_request_fallback_bundle_success(mock_bundles,
+                                                    invoke,
+                                                    bundle,
+                                                    fallback_bundle):
+    result = invoke([
+        'request',
+        '--item-type=PSScene',
+        f'--bundle={bundle}',
+        f'--fallback-bundle={fallback_bundle}',
+        '--name=my order',
+        '20250130_035211_69_2516'
+    ])
+    assert result.exit_code == 0
+
+
+@pytest.mark.parametrize(
+    "bundle, fallback_bundle",
+    [
+        ('analytic_8b_udm2', ''),
+        ('analytic_8b_udm2', 'analytic_udm2;analytic_3b_udm2'),
+        ('analytic_8b_udm2', 'analytic_udm2,analytic_3b_udm2,fake_bundle'),
+    ])
+def test_cli_orders_request_fallback_bundle_fail(mock_bundles,
+                                                 invoke,
+                                                 bundle,
+                                                 fallback_bundle):
+    result = invoke([
+        'request',
+        '--item-type=PSScene',
+        f'--bundle={bundle}',
+        f'--fallback-bundle={fallback_bundle}',
+        '--name=my order',
+        '20250130_035211_69_2516'
+    ])
+    assert result.exit_code == 2
+
+
 def test_cli_orders_request_id_empty(mock_bundles, invoke):
     result = invoke([
         'request', '--item-type=PSScene', '--bundle=visual', '--name=test', ''

tests/unit/test_order_request.py

@@ -404,3 +404,38 @@ def test_sentinel_hub_collection_configuration():
         }
     }
     assert sh_config == expected
+
+
+@pytest.mark.parametrize(
+    "bundle, fallback_bundle",
+    [
+        ('analytic_8b_udm2', 'analytic_udm2'),
+        ('analytic_8b_udm2', 'analytic_udm2,analytic_3b_udm2'),
+    ])
+def test_fallback_bundle_success(bundle, fallback_bundle):
+    product = order_request.product(["20250130_035211_69_2516"],
+                                    bundle,
+                                    "PSScene",
+                                    fallback_bundle=fallback_bundle)
+
+    expected = {
+        "item_ids": ["20250130_035211_69_2516"],
+        "item_type": "PSScene",
+        "product_bundle": f"{bundle},{fallback_bundle}"
+    }
+    assert product == expected
+
+
+@pytest.mark.parametrize(
+    "bundle, fallback_bundle",
+    [
+        ('analytic_8b_udm2', ''),
+        ('analytic_8b_udm2', 'analytic_udm2;analytic_3b_udm2'),
+        ('analytic_8b_udm2', 'analytic_udm2,analytic_3b_udm2,fake_bundle'),
+    ])
+def test_fallback_bundle_invalid(bundle, fallback_bundle):
+    with pytest.raises(specs.SpecificationException):
+        _ = order_request.product(["20250130_035211_69_2516"],
+                                  bundle,
+                                  "PSScene",
+                                  fallback_bundle=fallback_bundle)