convert missing tests

Author: yoshuawuyts

sdk/data_cosmos/examples/create_delete_database.rs

@@ -44,8 +44,8 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
 
     // create collection!
     {
-        let db_client = client.database(database_name.clone());
-        let create_collection_response = db_client
+        let database = client.database(database_name.clone());
+        let create_collection_response = database
             .create_collection("panzadoro", "/id")
             .into_future()
             .await?;
@@ -55,12 +55,12 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
             create_collection_response
         );
 
-        let db_collection = db_client.collection("panzadoro");
+        let db_collection = database.collection("panzadoro");
 
         let get_collection_response = db_collection.get_collection().into_future().await?;
         println!("get_collection_response == {:#?}", get_collection_response);
 
-        let mut stream = db_client.list_collections().into_stream();
+        let mut stream = database.list_collections().into_stream();
         while let Some(res) = stream.next().await {
             let res = res?;
             println!("res == {:#?}", res);

sdk/data_cosmos/examples/permission_00.rs

@@ -33,7 +33,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
 
     let database = client.database(database_name);
     let collection = database.collection(collection_name);
-    let collection2_client = database.collection(collection_name2);
+    let collection2 = database.collection(collection_name2);
     let user = database.user(user_name);
 
     let get_database_response = database.get_database().into_future().await?;
@@ -42,7 +42,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
     let get_collection_response = collection.get_collection().into_future().await?;
     println!("get_collection_response == {:#?}", get_collection_response);
 
-    let get_collection2_response = collection2_client.get_collection().into_future().await?;
+    let get_collection2_response = collection2.get_collection().into_future().await?;
     println!(
         "get_collection2_response == {:#?}",
         get_collection2_response

sdk/data_cosmos/src/clients/mod.rs

@@ -14,7 +14,7 @@
 //!
 //! // Create an http client, then a `CosmosClient`, and then a `DatabaseClient`
 //! let client = CosmosClient::new(account, authorization_token, CosmosOptions::default());
-//! let client = client.into_database(database_name);
+//! let client = client.database(database_name);
 //! ```
 
 mod attachment;

sdk/data_cosmos/src/lib.rs

@@ -64,9 +64,9 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync>> {
     let client = CosmosClient::new(account.clone(), authorization_token, CosmosOptions::default());
 
     // We know the database so we can obtain a database client.
-    let database = client.into_database(database_name);
+    let database = client.database(database_name);
     // We know the collection so we can obtain a collection client.
-    let collection = database.into_collection(collection_name);
+    let collection = database.collection(collection_name);
 
     // Insert 10 documents
     println!("Inserting 10 documents...");

sdk/data_cosmos/src/requests/execute_stored_procedure_builder.rs

@@ -41,7 +41,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
         .await
         .unwrap();
 
-    let database = client.into_database(DATABASE_NAME);
+    let database = client.database(DATABASE_NAME);
 
     // create a temp collection
     let _create_collection_response = {
@@ -72,7 +72,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
             .unwrap()
     };
 
-    let collection = database.clone().into_collection(COLLECTION_NAME);
+    let collection = database.collection(COLLECTION_NAME);
 
     let id = format!("unique_id{}", 100);
 
@@ -90,7 +90,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
         .await?
         .into();
 
-    let document = collection.into_document(id.clone(), &doc.id)?;
+    let document = collection.document(id.clone(), &doc.id)?;
 
     // list attachments, there must be none.
     let ret = document
@@ -103,7 +103,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
     assert_eq!(0, ret.attachments.len());
 
     // create reference attachment
-    let attachment = document.into_attachment("reference");
+    let attachment = document.attachment("reference");
     let resp = attachment
         .create_attachment("https://www.bing.com", "image/jpeg")
         .consistency_level(&ret)
@@ -118,7 +118,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
         .await?;
 
     // create slug attachment
-    let attachment = document.into_attachment("slug");
+    let attachment = document.attachment("slug");
     let resp = attachment
         .create_slug("something cool here".into())
         .consistency_level(&resp)
@@ -138,8 +138,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
 
     // get reference attachment, it must have the updated media link
     let reference_attachment = document
-        .clone()
-        .into_attachment("reference")
+        .attachment("reference")
         .get()
         .consistency_level(&ret)
         .into_future()
@@ -152,8 +151,7 @@ async fn attachment() -> Result<(), azure_data_cosmos::Error> {
     // get slug attachment, it must have the text/plain content type
     println!("getting slug attachment");
     let slug_attachment = document
-        .clone()
-        .into_attachment("slug")
+        .attachment("slug")
         .get()
         .consistency_level(&reference_attachment)
         .into_future()