From 8a6ee9bca902db076be12745967a69c7c184e044 Mon Sep 17 00:00:00 2001
From: Jake Swenson <jake@jakeswenson.com>
Date: Fri, 14 May 2021 07:33:45 -0700
Subject: split models

---
 src/lib.rs | 15 ++++++++-------
 1 file changed, 8 insertions(+), 7 deletions(-)

(limited to 'src/lib.rs')

diff --git a/src/lib.rs b/src/lib.rs
index 53e639f..7fbaa15 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,6 +1,5 @@
 use crate::models::search::SearchRequest;
 use crate::models::{Database, ListResponse};
-use std::collections::HashMap;
 
 mod models;
 
@@ -54,11 +53,15 @@ mod tests {
     use crate::NotionApi;
     const TEST_TOKEN: &'static str = include_str!(".api_token");
 
+    fn test_client() -> NotionApi {
+        NotionApi {
+            token: TEST_TOKEN.trim().to_string(),
+        }
+    }
+
     #[tokio::test]
     async fn list_databases() -> Result<(), Box<dyn std::error::Error>> {
-        let api = NotionApi {
-            token: TEST_TOKEN.to_string(),
-        };
+        let api = test_client();
 
         dbg!(api.list_databases().await?);
 
@@ -67,9 +70,7 @@ mod tests {
 
     #[tokio::test]
     async fn search() -> Result<(), Box<dyn std::error::Error>> {
-        let api = NotionApi {
-            token: TEST_TOKEN.to_string(),
-        };
+        let api = test_client();
 
         dbg!(
             api.search(NotionSearch::Filter {
-- 
cgit v1.2.3