feat: default database and collection for MetadataDbClient

Author: helen-m-lin

src/aind_data_access_api/document_db.py

@@ -362,6 +362,36 @@ def __exit__(self, exc_type, exc_val, exc_tb):
 class MetadataDbClient(Client):
     """Class to manage reading and writing to metadata db"""
 
+    def __init__(
+        self,
+        host: str,
+        database: str = "metadata_index",
+        collection: str = "data_assets",
+        version: str = "v1",
+        boto: Optional[BotoSession] = None,
+        session: Optional[Session] = None,
+    ):
+        """
+        Instantiate a MetadataDbClient.
+
+        Parameters
+        ----------
+        host : str
+        database : str
+        collection : str
+        version : str
+        boto : Optional[BotoSession]
+        session : Optional[Session]
+        """
+        super().__init__(
+            host=host,
+            database=database,
+            collection=collection,
+            version=version,
+            boto=boto,
+            session=session,
+        )
+
     def retrieve_docdb_records(
         self,
         filter_query: Optional[dict] = None,

tests/test_document_db.py

@@ -437,8 +437,6 @@ class TestMetadataDbClient(unittest.TestCase):
 
     example_client_args = {
         "host": "example.com/",
-        "database": "metadata_db",
-        "collection": "data_assets",
     }
 
     example_record_list = [
@@ -452,6 +450,26 @@ class TestMetadataDbClient(unittest.TestCase):
         for id_num in range(0, 10)
     ]
 
+    def test_metadatadbclient_constructor(self):
+        """Tests class constructor"""
+        client = MetadataDbClient(**self.example_client_args)
+
+        self.assertEqual("example.com", client.host)
+        self.assertEqual("metadata_index", client.database)
+        self.assertEqual("data_assets", client.collection)
+        self.assertEqual("v1", client.version)
+        self.assertEqual(
+            "https://example.com/v1/metadata_index/data_assets",
+            client._base_url,
+        )
+
+        client = MetadataDbClient(**self.example_client_args, version="v2")
+        self.assertEqual("v2", client.version)
+        self.assertEqual(
+            "https://example.com/v2/metadata_index/data_assets",
+            client._base_url,
+        )
+
     @patch("aind_data_access_api.document_db.Client._find_records")
     def test_retrieve_docdb_records(
         self,