Austin/firecrawl
1
0
Fork 0
mirror of https://github.com/mendableai/firecrawl.git synced 2024-11-16 11:42:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: update tests

Browse Source
This commit is contained in:
sanix-darker 2024-07-12 14:07:58 +02:00
parent d86de4f5a6
commit 21983011b9
No known key found for this signature in database
GPG Key ID: 02BD1EB7C7D57CD3
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
apps/rust-sdk/tests/e2e_with_auth.rs
View File

@ -154,14 +154,14 @@ async fn test_crawl_url_with_idempotency_key_e2e() {
"excludes": ["blog/*"] "excludes": ["blog/*"]
} }
}); });
let result = app.crawl_url("https://roastmywebsite.ai", Some(params), true, 2, Some(unique_idempotency_key.clone())).await.unwrap(); let result = app.crawl_url("https://roastmywebsite.ai", Some(params.clone()), true, 2, Some(unique_idempotency_key.clone())).await.unwrap();
let result_as_str = result.as_object().unwrap(); let result_as_str = result.as_object().unwrap();
assert!(result_as_str.len() > 0); assert!(result_as_str.len() > 0);
// assert!(result[0].contains_key("content")); // assert!(result[0].contains_key("content"));
// assert!(result[0]["content"].as_str().unwrap().contains("_Roast_")); // assert!(result[0]["content"].as_str().unwrap().contains("_Roast_"));
let conflict_result = app.crawl_url("https://firecrawl.dev", Some(params), true, 2, Some(unique_idempotency_key)).await; let conflict_result = app.crawl_url("https://firecrawl.dev", Some(params.clone()), true, 2, Some(unique_idempotency_key)).await;
assert_matches!(conflict_result, Err(e) if e.to_string() == "Conflict: Failed to start crawl job due to a conflict. Idempotency key already used"); assert_matches!(conflict_result, Err(e) if e.to_string() == "Conflict: Failed to start crawl job due to a conflict. Idempotency key already used");
} }
@ -191,11 +191,13 @@ async fn test_check_crawl_status_e2e() {
async fn test_search_e2e() { async fn test_search_e2e() {
dotenv().ok(); dotenv().ok();
let api_url = env::var("API_URL").unwrap(); let api_url = env::var("API_URL").unwrap();
let api_key = env::var("TEST_API_KEY").unwrap(); let api_key = env::var("API_KEY").unwrap();
let app = FirecrawlApp::new(Some(api_key), Some(api_url)).unwrap(); let app = FirecrawlApp::new(Some(api_key), Some(api_url)).unwrap();
let result = app.search("test query").await.unwrap();
assert!(result.as_object().unwrap().len() > 2); let result = app.search("test query", None).await.unwrap();
//assert!(result.as_object().unwrap()[0].contains_key("content")); assert!(result.is_array());
assert!(!result.as_array().unwrap().is_empty());
assert!(result[0].get("content").is_some());
} }
#[tokio::test] #[tokio::test]
@ -203,7 +205,7 @@ async fn test_search_invalid_api_key() {
dotenv().ok(); dotenv().ok();
let api_url = env::var("API_URL").unwrap(); let api_url = env::var("API_URL").unwrap();
let app = FirecrawlApp::new(Some("invalid_api_key".to_string()), Some(api_url)).unwrap(); let app = FirecrawlApp::new(Some("invalid_api_key".to_string()), Some(api_url)).unwrap();
let result = app.search("test query").await; let result = app.search("test query", None).await;
assert_matches!(result, Err(e) if e.to_string() == "Unexpected error during search: Status code 401. Unauthorized: Invalid token"); assert_matches!(result, Err(e) if e.to_string() == "Unexpected error during search: Status code 401. Unauthorized: Invalid token");
} }