fake_auth.rs 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869
  1. use std::env;
  2. use std::str::FromStr;
  3. use std::sync::Arc;
  4. use bip39::Mnemonic;
  5. use cashu::{MintAuthRequest, MintInfo};
  6. use cdk::amount::{Amount, SplitTarget};
  7. use cdk::mint_url::MintUrl;
  8. use cdk::nuts::nut00::{KnownMethod, ProofsMethods};
  9. use cdk::nuts::{
  10. AuthProof, AuthToken, BlindAuthToken, CheckStateRequest, CurrencyUnit, MeltQuoteBolt11Request,
  11. MeltQuoteState, MeltRequest, MintQuoteBolt11Request, MintRequest, PaymentMethod,
  12. RestoreRequest, State, SwapRequest,
  13. };
  14. use cdk::wallet::{AuthHttpClient, AuthMintConnector, HttpClient, MintConnector, WalletBuilder};
  15. use cdk::{Error, OidcClient};
  16. use cdk_fake_wallet::create_fake_invoice;
  17. use cdk_integration_tests::fund_wallet;
  18. use cdk_sqlite::wallet::memory;
  19. const MINT_URL: &str = "http://127.0.0.1:8087";
  20. const ENV_OIDC_USER: &str = "CDK_TEST_OIDC_USER";
  21. const ENV_OIDC_PASSWORD: &str = "CDK_TEST_OIDC_PASSWORD";
  22. fn get_oidc_credentials() -> (String, String) {
  23. let user = env::var(ENV_OIDC_USER).unwrap_or_else(|_| "test".to_string());
  24. let password = env::var(ENV_OIDC_PASSWORD).unwrap_or_else(|_| "test".to_string());
  25. (user, password)
  26. }
  27. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  28. async fn test_invalid_credentials() {
  29. let db = Arc::new(memory::empty().await.unwrap());
  30. let wallet = WalletBuilder::new()
  31. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  32. .unit(CurrencyUnit::Sat)
  33. .localstore(db.clone())
  34. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  35. .build()
  36. .expect("Wallet");
  37. let mint_info = wallet
  38. .fetch_mint_info()
  39. .await
  40. .expect("mint info")
  41. .expect("could not get mint info");
  42. // Try to get a token with invalid credentials
  43. let token_result =
  44. get_custom_access_token(&mint_info, "invalid_user", "invalid_password").await;
  45. // Should fail with an error
  46. assert!(
  47. token_result.is_err(),
  48. "Expected authentication to fail with invalid credentials"
  49. );
  50. }
  51. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  52. async fn test_quote_status_without_auth() {
  53. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  54. // Test mint quote status
  55. {
  56. let quote_res = client
  57. .get_mint_quote_status("123e4567-e89b-12d3-a456-426614174000")
  58. .await;
  59. assert!(
  60. matches!(quote_res, Err(Error::BlindAuthRequired)),
  61. "Expected AuthRequired error, got {:?}",
  62. quote_res
  63. );
  64. }
  65. // Test melt quote status
  66. {
  67. let quote_res = client
  68. .get_melt_quote_status("123e4567-e89b-12d3-a456-426614174000")
  69. .await;
  70. assert!(
  71. matches!(quote_res, Err(Error::BlindAuthRequired)),
  72. "Expected AuthRequired error, got {:?}",
  73. quote_res
  74. );
  75. }
  76. }
  77. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  78. async fn test_mint_without_auth() {
  79. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  80. {
  81. let request = MintQuoteBolt11Request {
  82. unit: CurrencyUnit::Sat,
  83. amount: 10.into(),
  84. description: None,
  85. pubkey: None,
  86. };
  87. let quote_res = client.post_mint_quote(request).await;
  88. assert!(
  89. matches!(quote_res, Err(Error::BlindAuthRequired)),
  90. "Expected AuthRequired error, got {:?}",
  91. quote_res
  92. );
  93. }
  94. {
  95. let request = MintRequest {
  96. quote: "123e4567-e89b-12d3-a456-426614174000".to_string(),
  97. outputs: vec![],
  98. signature: None,
  99. };
  100. let mint_res = client
  101. .post_mint(&PaymentMethod::Known(KnownMethod::Bolt11), request)
  102. .await;
  103. assert!(
  104. matches!(mint_res, Err(Error::BlindAuthRequired)),
  105. "Expected AuthRequired error, got {:?}",
  106. mint_res
  107. );
  108. }
  109. {
  110. let mint_res = client
  111. .get_mint_quote_status("123e4567-e89b-12d3-a456-426614174000")
  112. .await;
  113. assert!(
  114. matches!(mint_res, Err(Error::BlindAuthRequired)),
  115. "Expected AuthRequired error, got {:?}",
  116. mint_res
  117. );
  118. }
  119. }
  120. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  121. async fn test_mint_bat_without_cat() {
  122. let client = AuthHttpClient::new(MintUrl::from_str(MINT_URL).expect("valid mint url"), None);
  123. let res = client
  124. .post_mint_blind_auth(MintAuthRequest { outputs: vec![] })
  125. .await;
  126. assert!(
  127. matches!(res, Err(Error::ClearAuthRequired)),
  128. "Expected AuthRequired error, got {:?}",
  129. res
  130. );
  131. }
  132. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  133. async fn test_swap_without_auth() {
  134. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  135. let request = SwapRequest::new(vec![], vec![]);
  136. let quote_res = client.post_swap(request).await;
  137. assert!(
  138. matches!(quote_res, Err(Error::BlindAuthRequired)),
  139. "Expected AuthRequired error, got {:?}",
  140. quote_res
  141. );
  142. }
  143. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  144. async fn test_melt_without_auth() {
  145. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  146. // Test melt quote request
  147. {
  148. let request = MeltQuoteBolt11Request {
  149. request: create_fake_invoice(100, "".to_string()),
  150. unit: CurrencyUnit::Sat,
  151. options: None,
  152. };
  153. let quote_res = client.post_melt_quote(request).await;
  154. assert!(
  155. matches!(quote_res, Err(Error::BlindAuthRequired)),
  156. "Expected AuthRequired error, got {:?}",
  157. quote_res
  158. );
  159. }
  160. // Test melt quote
  161. {
  162. let request = MeltQuoteBolt11Request {
  163. request: create_fake_invoice(100, "".to_string()),
  164. unit: CurrencyUnit::Sat,
  165. options: None,
  166. };
  167. let quote_res = client.post_melt_quote(request).await;
  168. assert!(
  169. matches!(quote_res, Err(Error::BlindAuthRequired)),
  170. "Expected AuthRequired error, got {:?}",
  171. quote_res
  172. );
  173. }
  174. // Test melt
  175. {
  176. let request = MeltRequest::new(
  177. "123e4567-e89b-12d3-a456-426614174000".to_string(),
  178. vec![],
  179. None,
  180. );
  181. let melt_res = client
  182. .post_melt(&PaymentMethod::Known(KnownMethod::Bolt11), request)
  183. .await;
  184. assert!(
  185. matches!(melt_res, Err(Error::BlindAuthRequired)),
  186. "Expected AuthRequired error, got {:?}",
  187. melt_res
  188. );
  189. }
  190. // Check melt quote state
  191. {
  192. let melt_res = client
  193. .get_melt_quote_status("123e4567-e89b-12d3-a456-426614174000")
  194. .await;
  195. assert!(
  196. matches!(melt_res, Err(Error::BlindAuthRequired)),
  197. "Expected AuthRequired error, got {:?}",
  198. melt_res
  199. );
  200. }
  201. }
  202. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  203. async fn test_check_without_auth() {
  204. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  205. let request = CheckStateRequest { ys: vec![] };
  206. let quote_res = client.post_check_state(request).await;
  207. assert!(
  208. matches!(quote_res, Err(Error::BlindAuthRequired)),
  209. "Expected AuthRequired error, got {:?}",
  210. quote_res
  211. );
  212. }
  213. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  214. async fn test_restore_without_auth() {
  215. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  216. let request = RestoreRequest { outputs: vec![] };
  217. let restore_res = client.post_restore(request).await;
  218. assert!(
  219. matches!(restore_res, Err(Error::BlindAuthRequired)),
  220. "Expected AuthRequired error, got {:?}",
  221. restore_res
  222. );
  223. }
  224. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  225. async fn test_mint_blind_auth() {
  226. let db = Arc::new(memory::empty().await.unwrap());
  227. let wallet = WalletBuilder::new()
  228. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  229. .unit(CurrencyUnit::Sat)
  230. .localstore(db.clone())
  231. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  232. .build()
  233. .expect("Wallet");
  234. let mint_info = wallet.fetch_mint_info().await.unwrap().unwrap();
  235. let (access_token, _) = get_access_token(&mint_info).await;
  236. wallet.set_cat(access_token).await.unwrap();
  237. wallet
  238. .mint_blind_auth(10.into())
  239. .await
  240. .expect("Could not mint blind auth");
  241. let proofs = wallet
  242. .get_unspent_auth_proofs()
  243. .await
  244. .expect("Could not get auth proofs");
  245. assert!(proofs.len() == 10)
  246. }
  247. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  248. async fn test_mint_with_auth() {
  249. let db = Arc::new(memory::empty().await.unwrap());
  250. let wallet = WalletBuilder::new()
  251. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  252. .unit(CurrencyUnit::Sat)
  253. .localstore(db.clone())
  254. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  255. .build()
  256. .expect("Wallet");
  257. let mint_info = wallet
  258. .fetch_mint_info()
  259. .await
  260. .expect("mint info")
  261. .expect("could not get mint info");
  262. let (access_token, _) = get_access_token(&mint_info).await;
  263. println!("st{}", access_token);
  264. wallet.set_cat(access_token).await.unwrap();
  265. wallet
  266. .mint_blind_auth(10.into())
  267. .await
  268. .expect("Could not mint blind auth");
  269. let wallet = Arc::new(wallet);
  270. let mint_amount: Amount = 100.into();
  271. let quote = wallet.mint_bolt11_quote(mint_amount, None).await.unwrap();
  272. let proofs = wallet
  273. .wait_and_mint_quote(
  274. quote.clone(),
  275. SplitTarget::default(),
  276. None,
  277. tokio::time::Duration::from_secs(60),
  278. )
  279. .await
  280. .expect("payment");
  281. assert!(proofs.total_amount().expect("Could not get proofs amount") == mint_amount);
  282. }
  283. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  284. async fn test_swap_with_auth() {
  285. let db = Arc::new(memory::empty().await.unwrap());
  286. let wallet = WalletBuilder::new()
  287. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  288. .unit(CurrencyUnit::Sat)
  289. .localstore(db.clone())
  290. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  291. .build()
  292. .expect("Wallet");
  293. let mint_info = wallet.fetch_mint_info().await.unwrap().unwrap();
  294. let (access_token, _) = get_access_token(&mint_info).await;
  295. wallet.set_cat(access_token).await.unwrap();
  296. let wallet = Arc::new(wallet);
  297. wallet.mint_blind_auth(10.into()).await.unwrap();
  298. fund_wallet(wallet.clone(), 100.into()).await;
  299. let proofs = wallet
  300. .get_unspent_proofs()
  301. .await
  302. .expect("Could not get proofs");
  303. let swapped_proofs = wallet
  304. .swap(
  305. Some(proofs.total_amount().unwrap()),
  306. SplitTarget::default(),
  307. proofs.clone(),
  308. None,
  309. false,
  310. )
  311. .await
  312. .expect("Could not swap")
  313. .expect("Could not swap");
  314. let check_spent = wallet
  315. .check_proofs_spent(proofs.clone())
  316. .await
  317. .expect("Could not check proofs");
  318. for state in check_spent {
  319. if state.state != State::Spent {
  320. panic!("Input proofs should be spent");
  321. }
  322. }
  323. assert!(swapped_proofs.total_amount().unwrap() == proofs.total_amount().unwrap())
  324. }
  325. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  326. async fn test_melt_with_auth() {
  327. let db = Arc::new(memory::empty().await.unwrap());
  328. let wallet = WalletBuilder::new()
  329. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  330. .unit(CurrencyUnit::Sat)
  331. .localstore(db.clone())
  332. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  333. .build()
  334. .expect("Wallet");
  335. let mint_info = wallet
  336. .fetch_mint_info()
  337. .await
  338. .expect("Mint info not found")
  339. .expect("Mint info not found");
  340. let (access_token, _) = get_access_token(&mint_info).await;
  341. wallet.set_cat(access_token).await.unwrap();
  342. let wallet = Arc::new(wallet);
  343. wallet.mint_blind_auth(10.into()).await.unwrap();
  344. fund_wallet(wallet.clone(), 100.into()).await;
  345. let bolt11 = create_fake_invoice(2_000, "".to_string());
  346. let melt_quote = wallet
  347. .melt_quote(PaymentMethod::BOLT11, bolt11.to_string(), None, None)
  348. .await
  349. .expect("Could not get melt quote");
  350. let prepared = wallet
  351. .prepare_melt(&melt_quote.id, std::collections::HashMap::new())
  352. .await
  353. .expect("Could not prepare melt");
  354. let after_melt = prepared.confirm().await.expect("Could not melt");
  355. assert!(after_melt.state() == MeltQuoteState::Paid);
  356. }
  357. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  358. async fn test_mint_auth_over_max() {
  359. let db = Arc::new(memory::empty().await.unwrap());
  360. let wallet = WalletBuilder::new()
  361. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  362. .unit(CurrencyUnit::Sat)
  363. .localstore(db.clone())
  364. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  365. .build()
  366. .expect("Wallet");
  367. let wallet = Arc::new(wallet);
  368. let mint_info = wallet
  369. .fetch_mint_info()
  370. .await
  371. .expect("Mint info not found")
  372. .expect("Mint info not found");
  373. let (access_token, _) = get_access_token(&mint_info).await;
  374. wallet.set_cat(access_token).await.unwrap();
  375. let auth_proofs = wallet
  376. .mint_blind_auth((mint_info.nuts.nut22.expect("Auth enabled").bat_max_mint + 1).into())
  377. .await;
  378. assert!(
  379. matches!(
  380. auth_proofs,
  381. Err(Error::AmountOutofLimitRange(
  382. Amount::ZERO,
  383. Amount::ZERO,
  384. Amount::ZERO,
  385. ))
  386. ),
  387. "Expected amount out of limit error, got {:?}",
  388. auth_proofs
  389. );
  390. }
  391. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  392. async fn test_reuse_auth_proof() {
  393. let db = Arc::new(memory::empty().await.unwrap());
  394. let wallet = WalletBuilder::new()
  395. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  396. .unit(CurrencyUnit::Sat)
  397. .localstore(db.clone())
  398. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  399. .build()
  400. .expect("Wallet");
  401. let mint_info = wallet.fetch_mint_info().await.unwrap().unwrap();
  402. let (access_token, _) = get_access_token(&mint_info).await;
  403. wallet.set_cat(access_token).await.unwrap();
  404. wallet.mint_blind_auth(1.into()).await.unwrap();
  405. let proofs = wallet
  406. .localstore
  407. .get_proofs(None, Some(CurrencyUnit::Auth), None, None)
  408. .await
  409. .unwrap();
  410. assert!(proofs.len() == 1);
  411. {
  412. let quote = wallet
  413. .mint_bolt11_quote(10.into(), None)
  414. .await
  415. .expect("Quote should be allowed");
  416. assert!(quote.amount == Some(10.into()));
  417. }
  418. wallet
  419. .localstore
  420. .update_proofs(proofs, vec![])
  421. .await
  422. .unwrap();
  423. {
  424. let quote_res = wallet.mint_bolt11_quote(10.into(), None).await;
  425. assert!(
  426. matches!(quote_res, Err(Error::TokenAlreadySpent)),
  427. "Expected AuthRequired error, got {:?}",
  428. quote_res
  429. );
  430. }
  431. }
  432. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  433. async fn test_melt_with_invalid_auth() {
  434. let db = Arc::new(memory::empty().await.unwrap());
  435. let wallet = WalletBuilder::new()
  436. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  437. .unit(CurrencyUnit::Sat)
  438. .localstore(db.clone())
  439. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  440. .build()
  441. .expect("Wallet");
  442. let mint_info = wallet.fetch_mint_info().await.unwrap().unwrap();
  443. let (access_token, _) = get_access_token(&mint_info).await;
  444. wallet.set_cat(access_token).await.unwrap();
  445. wallet.mint_blind_auth(10.into()).await.unwrap();
  446. fund_wallet(Arc::new(wallet.clone()), 1.into()).await;
  447. let proofs = wallet
  448. .get_unspent_proofs()
  449. .await
  450. .expect("wallet has proofs");
  451. println!("{:#?}", proofs);
  452. let proof = proofs.first().expect("wallet has one proof");
  453. let client = HttpClient::new(MintUrl::from_str(MINT_URL).expect("Valid mint url"), None);
  454. {
  455. let invalid_auth_proof = AuthProof {
  456. keyset_id: proof.keyset_id,
  457. secret: proof.secret.clone(),
  458. c: proof.c,
  459. dleq: proof.dleq.clone(),
  460. };
  461. let _auth_token = AuthToken::BlindAuth(BlindAuthToken::new(invalid_auth_proof));
  462. let request = MintQuoteBolt11Request {
  463. unit: CurrencyUnit::Sat,
  464. amount: 10.into(),
  465. description: None,
  466. pubkey: None,
  467. };
  468. let quote_res = client.post_mint_quote(request).await;
  469. assert!(
  470. matches!(quote_res, Err(Error::BlindAuthRequired)),
  471. "Expected AuthRequired error, got {:?}",
  472. quote_res
  473. );
  474. }
  475. {
  476. let (access_token, _) = get_access_token(&mint_info).await;
  477. wallet.set_cat(access_token).await.unwrap();
  478. }
  479. }
  480. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  481. async fn test_refresh_access_token() {
  482. let db = Arc::new(memory::empty().await.unwrap());
  483. let wallet = WalletBuilder::new()
  484. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  485. .unit(CurrencyUnit::Sat)
  486. .localstore(db.clone())
  487. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  488. .build()
  489. .expect("Wallet");
  490. let mint_info = wallet
  491. .fetch_mint_info()
  492. .await
  493. .expect("mint info")
  494. .expect("could not get mint info");
  495. let (access_token, refresh_token) = get_access_token(&mint_info).await;
  496. // Set the initial access token and refresh token
  497. wallet.set_cat(access_token.clone()).await.unwrap();
  498. wallet
  499. .set_refresh_token(refresh_token.clone())
  500. .await
  501. .unwrap();
  502. // Mint some blind auth tokens with the initial access token
  503. wallet.mint_blind_auth(5.into()).await.unwrap();
  504. // Refresh the access token
  505. wallet.refresh_access_token().await.unwrap();
  506. // Verify we can still perform operations with the refreshed token
  507. let mint_amount: Amount = 10.into();
  508. // Try to mint more blind auth tokens with the refreshed token
  509. let auth_proofs = wallet.mint_blind_auth(5.into()).await.unwrap();
  510. assert_eq!(auth_proofs.len(), 5);
  511. let total_auth_proofs = wallet.get_unspent_auth_proofs().await.unwrap();
  512. assert_eq!(total_auth_proofs.len(), 10); // 5 from before refresh + 5 after refresh
  513. // Try to get a mint quote with the refreshed token
  514. let mint_quote = wallet
  515. .mint_bolt11_quote(mint_amount, None)
  516. .await
  517. .expect("failed to get mint quote with refreshed token");
  518. assert_eq!(mint_quote.amount, Some(mint_amount));
  519. // Verify the total number of auth tokens
  520. let total_auth_proofs = wallet.get_unspent_auth_proofs().await.unwrap();
  521. assert_eq!(total_auth_proofs.len(), 9); // 5 from before refresh + 5 after refresh - 1 for the quote
  522. }
  523. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  524. async fn test_invalid_refresh_token() {
  525. let db = Arc::new(memory::empty().await.unwrap());
  526. let wallet = WalletBuilder::new()
  527. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  528. .unit(CurrencyUnit::Sat)
  529. .localstore(db.clone())
  530. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  531. .build()
  532. .expect("Wallet");
  533. let mint_info = wallet
  534. .fetch_mint_info()
  535. .await
  536. .expect("mint info")
  537. .expect("could not get mint info");
  538. let (access_token, _) = get_access_token(&mint_info).await;
  539. // Set the initial access token
  540. wallet.set_cat(access_token.clone()).await.unwrap();
  541. // Set an invalid refresh token
  542. wallet
  543. .set_refresh_token("invalid_refresh_token".to_string())
  544. .await
  545. .unwrap();
  546. // Attempt to refresh the access token with an invalid refresh token
  547. let refresh_result = wallet.refresh_access_token().await;
  548. // Should fail with an error
  549. assert!(refresh_result.is_err(), "Expected refresh token error");
  550. }
  551. #[tokio::test(flavor = "multi_thread", worker_threads = 1)]
  552. async fn test_auth_token_spending_order() {
  553. let db = Arc::new(memory::empty().await.unwrap());
  554. let wallet = WalletBuilder::new()
  555. .mint_url(MintUrl::from_str(MINT_URL).expect("Valid mint url"))
  556. .unit(CurrencyUnit::Sat)
  557. .localstore(db.clone())
  558. .seed(Mnemonic::generate(12).unwrap().to_seed_normalized(""))
  559. .build()
  560. .expect("Wallet");
  561. let mint_info = wallet
  562. .fetch_mint_info()
  563. .await
  564. .expect("mint info")
  565. .expect("could not get mint info");
  566. let (access_token, _) = get_access_token(&mint_info).await;
  567. wallet.set_cat(access_token).await.unwrap();
  568. // Mint auth tokens in two batches to test ordering
  569. wallet.mint_blind_auth(2.into()).await.unwrap();
  570. // Get the first batch of auth proofs
  571. let first_batch = wallet.get_unspent_auth_proofs().await.unwrap();
  572. assert_eq!(first_batch.len(), 2);
  573. // Mint a second batch
  574. wallet.mint_blind_auth(3.into()).await.unwrap();
  575. // Get all auth proofs
  576. let all_proofs = wallet.get_unspent_auth_proofs().await.unwrap();
  577. assert_eq!(all_proofs.len(), 5);
  578. // Use tokens and verify they're used in the expected order (FIFO)
  579. for i in 0..3 {
  580. let mint_quote = wallet
  581. .mint_bolt11_quote(10.into(), None)
  582. .await
  583. .expect("failed to get mint quote");
  584. assert_eq!(mint_quote.amount, Some(10.into()));
  585. // Check remaining tokens after each operation
  586. let remaining = wallet.get_unspent_auth_proofs().await.unwrap();
  587. assert_eq!(
  588. remaining.len(),
  589. 5 - (i + 1),
  590. "Expected {} remaining auth tokens after {} operations",
  591. 5 - (i + 1),
  592. i + 1
  593. );
  594. }
  595. }
  596. async fn get_access_token(mint_info: &MintInfo) -> (String, String) {
  597. let openid_discovery = mint_info
  598. .nuts
  599. .nut21
  600. .clone()
  601. .expect("Nut21 defined")
  602. .openid_discovery;
  603. let oidc_client = OidcClient::new(openid_discovery, None);
  604. // Get the token endpoint from the OIDC configuration
  605. let token_url = oidc_client
  606. .get_oidc_config()
  607. .await
  608. .expect("Failed to get OIDC config")
  609. .token_endpoint;
  610. // Create the request parameters
  611. let (user, password) = get_oidc_credentials();
  612. let params = [
  613. ("grant_type", "password"),
  614. ("client_id", "cashu-client"),
  615. ("username", &user),
  616. ("password", &password),
  617. ];
  618. // Make the token request directly
  619. let client = reqwest::Client::new();
  620. let response = client
  621. .post(token_url)
  622. .form(&params)
  623. .send()
  624. .await
  625. .expect("Failed to send token request");
  626. let token_response: serde_json::Value = response
  627. .json()
  628. .await
  629. .expect("Failed to parse token response");
  630. let access_token = token_response["access_token"]
  631. .as_str()
  632. .expect("No access token in response")
  633. .to_string();
  634. let refresh_token = token_response["refresh_token"]
  635. .as_str()
  636. .expect("No access token in response")
  637. .to_string();
  638. (access_token, refresh_token)
  639. }
  640. /// Get a new access token with custom credentials
  641. async fn get_custom_access_token(
  642. mint_info: &MintInfo,
  643. username: &str,
  644. password: &str,
  645. ) -> Result<(String, String), Error> {
  646. let openid_discovery = mint_info
  647. .nuts
  648. .nut21
  649. .clone()
  650. .expect("Nut21 defined")
  651. .openid_discovery;
  652. let oidc_client = OidcClient::new(openid_discovery, None);
  653. // Get the token endpoint from the OIDC configuration
  654. let token_url = oidc_client
  655. .get_oidc_config()
  656. .await
  657. .map_err(|_| Error::Custom("Failed to get OIDC config".to_string()))?
  658. .token_endpoint;
  659. // Create the request parameters
  660. let params = [
  661. ("grant_type", "password"),
  662. ("client_id", "cashu-client"),
  663. ("username", username),
  664. ("password", password),
  665. ];
  666. // Make the token request directly
  667. let client = reqwest::Client::new();
  668. let response = client
  669. .post(token_url)
  670. .form(&params)
  671. .send()
  672. .await
  673. .map_err(|_| Error::Custom("Failed to send token request".to_string()))?;
  674. if !response.status().is_success() {
  675. return Err(Error::Custom(format!(
  676. "Token request failed with status: {}",
  677. response.status()
  678. )));
  679. }
  680. let token_response: serde_json::Value = response
  681. .json()
  682. .await
  683. .map_err(|_| Error::Custom("Failed to parse token response".to_string()))?;
  684. let access_token = token_response["access_token"]
  685. .as_str()
  686. .ok_or_else(|| Error::Custom("No access token in response".to_string()))?
  687. .to_string();
  688. let refresh_token = token_response["refresh_token"]
  689. .as_str()
  690. .ok_or_else(|| Error::Custom("No refresh token in response".to_string()))?
  691. .to_string();
  692. Ok((access_token, refresh_token))
  693. }