Change product finding to handle lists of products

datacube/api/query.py

@@ -99,12 +99,12 @@ def __init__(self, index=None, product=None, geopolygon=None, like=None, **searc
             # Retrieve known keys for extra dimensions
             known_dim_keys = set()
             if product is not None:
-                datacube_product = index.products.get_by_name(product)
-                if datacube_product is not None:
-                    known_dim_keys.update(datacube_product.extra_dimensions.dims.keys())
+                datacube_products = index.products.search(product=product)
             else:
-                for datacube_product in index.products.get_all():
-                    known_dim_keys.update(datacube_product.extra_dimensions.dims.keys())
+                datacube_products = index.products.get_all()
+
+            for datacube_product in datacube_products:
+                known_dim_keys.update(datacube_product.extra_dimensions.dims.keys())
 
             remaining_keys -= known_dim_keys
 

tests/api/test_query.py

@@ -15,17 +15,20 @@
 from datacube.utils.geometry import CRS
 
 
+@pytest.fixture
+def mock_index():
+    from unittest.mock import MagicMock
+    return MagicMock()
+
+
 def test_datetime_to_timestamp():
     assert _datetime_to_timestamp((1990, 1, 7)) == 631670400
     assert _datetime_to_timestamp(datetime.datetime(1990, 1, 7)) == 631670400
     assert _datetime_to_timestamp(631670400) == 631670400
     assert _datetime_to_timestamp('1990-01-07T00:00:00.0Z') == 631670400
 
 
-def test_query_kwargs():
-    from unittest.mock import MagicMock
-
-    mock_index = MagicMock()
+def test_query_kwargs(mock_index):
     mock_index.datasets.get_field_names = lambda: {u'product', u'lat', u'sat_path', 'type_id', u'time', u'lon',
                                                    u'orbit', u'instrument', u'sat_row', u'platform', 'metadata_type',
                                                    u'gsi', 'type', 'id'}
@@ -213,3 +216,8 @@ def test_dateline_query_building():
 def test_query_issue_1146():
     q = Query(k='AB')
     assert q.search['k'] == 'AB'
+
+
+def test_query_multiple_products(mock_index):
+    q = Query(index=mock_index, product=['ls5_nbar_albers', 'ls7_nbar_albers'])
+    assert q.product == ['ls5_nbar_albers', 'ls7_nbar_albers']