fix doctests

andrusha · Jun 6, 2024 · 39996b2 · 39996b2
1 parent 09b78cc
commit 39996b2
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 6 deletions.
diff --git a/snowflake-api/README.md b/snowflake-api/README.md
@@ -44,21 +44,24 @@ snowflake-api = "0.7.0"
 
 Check [examples](./examples) for working programs using the library.
 
-
 ```rust
 use anyhow::Result;
-use snowflake_api::{QueryResult, SnowflakeApi};
+use snowflake_api::{QueryResult, AuthArgs, PasswordArgs, AuthType, SnowflakeApi, SnowflakeApiBuilder};
 
 async fn run_query(sql: &str) -> Result<QueryResult> {
-    let mut api = SnowflakeApi::with_password_auth(
+
+    let auth = AuthArgs::new(
         "ACCOUNT_IDENTIFIER",
         Some("WAREHOUSE"),
         Some("DATABASE"),
         Some("SCHEMA"),
         "USERNAME",
         Some("ROLE"),
-        "PASSWORD",
-    )?;
+        AuthType::Password(PasswordArgs { password: "password".to_string() })
+    );
+
+    let mut api: SnowflakeApi = SnowflakeApiBuilder::new(auth)
+    .build()?;
     let res = api.exec(sql).await?;
 
     Ok(res)
@@ -68,7 +71,7 @@ async fn run_query(sql: &str) -> Result<QueryResult> {
 Or using environment variables:
 
 ```rust
- use anyhow::Result;
+use anyhow::Result;
 use snowflake_api::{QueryResult, SnowflakeApi};
 
 async fn run_query(sql: &str) -> Result<QueryResult> {

diff --git a/snowflake-api/src/lib.rs b/snowflake-api/src/lib.rs
@@ -193,6 +193,26 @@ pub struct AuthArgs {
 }
 
 impl AuthArgs {
+    pub fn new(
+        account_identifier: &str,
+        warehouse: Option<&str>,
+        database: Option<&str>,
+        schema: Option<&str>,
+        username: &str,
+        role: Option<&str>,
+        auth_type: AuthType,
+    ) -> Self {
+        Self {
+            account_identifier: account_identifier.to_string(),
+            warehouse: warehouse.map(str::to_string),
+            database: database.map(str::to_string),
+            schema: schema.map(str::to_string),
+            username: username.to_string(),
+            role: role.map(str::to_string),
+            auth_type,
+        }
+    }
+
     pub fn from_env() -> Result<AuthArgs, SnowflakeApiError> {
         let auth_type = if let Ok(password) = std::env::var("SNOWFLAKE_PASSWORD") {
             Ok(AuthType::Password(PasswordArgs { password }))