mirror/mCaptcha
1
0
Fork 1
mirror of https://github.com/mCaptcha/mCaptcha.git synced 2026-02-12 02:25:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1868b14b6b5074f7f0cf842b20de2d22f3611fa9
mCaptcha/src/mcaptcha/api/v1/pow/get_config.rs.html
realaravinth 1868b14b6b Deploying to gh-pages from @ c1f6ce3ae2 🚀
2022-12-16 09:48:47 +00:00

608 lines
27 KiB
HTML
Raw Blame History

This file contains ambiguous Unicode characters
This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.
<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/api/v1/pow/get_config.rs`."><meta name="keywords" content="rust, rustlang, rust-lang"><title>get_config.rs - source</title><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../../../SourceSerif4-Regular.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../../../FiraSans-Regular.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../../../FiraSans-Medium.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../../../SourceCodePro-Regular.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../../../SourceSerif4-Bold.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../../../../SourceCodePro-Semibold.ttf.woff2"><link rel="stylesheet" href="../../../../../normalize.css"><link rel="stylesheet" href="../../../../../rustdoc.css" id="mainThemeStyle"><link rel="stylesheet" href="../../../../../ayu.css" disabled><link rel="stylesheet" href="../../../../../dark.css" disabled><link rel="stylesheet" href="../../../../../light.css" id="themeStyle"><script id="default-settings" ></script><script src="../../../../../storage.js"></script><script defer src="../../../../../source-script.js"></script><script defer src="../../../../../source-files.js"></script><script defer src="../../../../../main.js"></script><noscript><link rel="stylesheet" href="../../../../../noscript.css"></noscript><link rel="alternate icon" type="image/png" href="../../../../../favicon-16x16.png"><link rel="alternate icon" type="image/png" href="../../../../../favicon-32x32.png"><link rel="icon" type="image/svg+xml" href="../../../../../favicon.svg"></head><body class="rustdoc source"><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="sidebar"></nav><main><div class="width-limiter"><nav class="sub"><a class="sub-logo-container" href="../../../../../mcaptcha/index.html"><img class="rust-logo" src="../../../../../rust-logo.svg" alt="logo"></a><form class="search-form"><span></span><input class="search-input" name="search" autocomplete="off" spellcheck="false" placeholder="Click or press ‘S’ to search, ‘?’ for more options…" type="search"><div id="help-button" title="help" tabindex="-1"><a href="../../../../../help.html">?</a></div><div id="settings-menu" tabindex="-1"><a href="../../../../../settings.html" title="settings"><img width="22" height="22" alt="Change settings" src="../../../../../wheel.svg"></a></div></form></nav><section id="main-content" class="content"><div class="example-wrap"><pre class="src-line-numbers"><span id="1">1</span>
<span id="2">2</span>
<span id="3">3</span>
<span id="4">4</span>
<span id="5">5</span>
<span id="6">6</span>
<span id="7">7</span>
<span id="8">8</span>
<span id="9">9</span>
<span id="10">10</span>
<span id="11">11</span>
<span id="12">12</span>
<span id="13">13</span>
<span id="14">14</span>
<span id="15">15</span>
<span id="16">16</span>
<span id="17">17</span>
<span id="18">18</span>
<span id="19">19</span>
<span id="20">20</span>
<span id="21">21</span>
<span id="22">22</span>
<span id="23">23</span>
<span id="24">24</span>
<span id="25">25</span>
<span id="26">26</span>
<span id="27">27</span>
<span id="28">28</span>
<span id="29">29</span>
<span id="30">30</span>
<span id="31">31</span>
<span id="32">32</span>
<span id="33">33</span>
<span id="34">34</span>
<span id="35">35</span>
<span id="36">36</span>
<span id="37">37</span>
<span id="38">38</span>
<span id="39">39</span>
<span id="40">40</span>
<span id="41">41</span>
<span id="42">42</span>
<span id="43">43</span>
<span id="44">44</span>
<span id="45">45</span>
<span id="46">46</span>
<span id="47">47</span>
<span id="48">48</span>
<span id="49">49</span>
<span id="50">50</span>
<span id="51">51</span>
<span id="52">52</span>
<span id="53">53</span>
<span id="54">54</span>
<span id="55">55</span>
<span id="56">56</span>
<span id="57">57</span>
<span id="58">58</span>
<span id="59">59</span>
<span id="60">60</span>
<span id="61">61</span>
<span id="62">62</span>
<span id="63">63</span>
<span id="64">64</span>
<span id="65">65</span>
<span id="66">66</span>
<span id="67">67</span>
<span id="68">68</span>
<span id="69">69</span>
<span id="70">70</span>
<span id="71">71</span>
<span id="72">72</span>
<span id="73">73</span>
<span id="74">74</span>
<span id="75">75</span>
<span id="76">76</span>
<span id="77">77</span>
<span id="78">78</span>
<span id="79">79</span>
<span id="80">80</span>
<span id="81">81</span>
<span id="82">82</span>
<span id="83">83</span>
<span id="84">84</span>
<span id="85">85</span>
<span id="86">86</span>
<span id="87">87</span>
<span id="88">88</span>
<span id="89">89</span>
<span id="90">90</span>
<span id="91">91</span>
<span id="92">92</span>
<span id="93">93</span>
<span id="94">94</span>
<span id="95">95</span>
<span id="96">96</span>
<span id="97">97</span>
<span id="98">98</span>
<span id="99">99</span>
<span id="100">100</span>
<span id="101">101</span>
<span id="102">102</span>
<span id="103">103</span>
<span id="104">104</span>
<span id="105">105</span>
<span id="106">106</span>
<span id="107">107</span>
<span id="108">108</span>
<span id="109">109</span>
<span id="110">110</span>
<span id="111">111</span>
<span id="112">112</span>
<span id="113">113</span>
<span id="114">114</span>
<span id="115">115</span>
<span id="116">116</span>
<span id="117">117</span>
<span id="118">118</span>
<span id="119">119</span>
<span id="120">120</span>
<span id="121">121</span>
<span id="122">122</span>
<span id="123">123</span>
<span id="124">124</span>
<span id="125">125</span>
<span id="126">126</span>
<span id="127">127</span>
<span id="128">128</span>
<span id="129">129</span>
<span id="130">130</span>
<span id="131">131</span>
<span id="132">132</span>
<span id="133">133</span>
<span id="134">134</span>
<span id="135">135</span>
<span id="136">136</span>
<span id="137">137</span>
<span id="138">138</span>
<span id="139">139</span>
<span id="140">140</span>
<span id="141">141</span>
<span id="142">142</span>
<span id="143">143</span>
<span id="144">144</span>
<span id="145">145</span>
<span id="146">146</span>
<span id="147">147</span>
<span id="148">148</span>
<span id="149">149</span>
<span id="150">150</span>
<span id="151">151</span>
<span id="152">152</span>
<span id="153">153</span>
<span id="154">154</span>
<span id="155">155</span>
<span id="156">156</span>
<span id="157">157</span>
<span id="158">158</span>
<span id="159">159</span>
<span id="160">160</span>
<span id="161">161</span>
<span id="162">162</span>
<span id="163">163</span>
<span id="164">164</span>
<span id="165">165</span>
<span id="166">166</span>
<span id="167">167</span>
<span id="168">168</span>
<span id="169">169</span>
<span id="170">170</span>
<span id="171">171</span>
<span id="172">172</span>
<span id="173">173</span>
<span id="174">174</span>
<span id="175">175</span>
<span id="176">176</span>
<span id="177">177</span>
<span id="178">178</span>
<span id="179">179</span>
<span id="180">180</span>
<span id="181">181</span>
<span id="182">182</span>
<span id="183">183</span>
<span id="184">184</span>
<span id="185">185</span>
<span id="186">186</span>
<span id="187">187</span>
<span id="188">188</span>
<span id="189">189</span>
<span id="190">190</span>
<span id="191">191</span>
<span id="192">192</span>
<span id="193">193</span>
<span id="194">194</span>
<span id="195">195</span>
<span id="196">196</span>
<span id="197">197</span>
<span id="198">198</span>
<span id="199">199</span>
<span id="200">200</span>
<span id="201">201</span>
<span id="202">202</span>
<span id="203">203</span>
<span id="204">204</span>
<span id="205">205</span>
<span id="206">206</span>
<span id="207">207</span>
<span id="208">208</span>
<span id="209">209</span>
<span id="210">210</span>
<span id="211">211</span>
<span id="212">212</span>
<span id="213">213</span>
<span id="214">214</span>
<span id="215">215</span>
<span id="216">216</span>
<span id="217">217</span>
<span id="218">218</span>
<span id="219">219</span>
<span id="220">220</span>
<span id="221">221</span>
<span id="222">222</span>
<span id="223">223</span>
<span id="224">224</span>
<span id="225">225</span>
<span id="226">226</span>
<span id="227">227</span>
<span id="228">228</span>
<span id="229">229</span>
<span id="230">230</span>
<span id="231">231</span>
<span id="232">232</span>
<span id="233">233</span>
<span id="234">234</span>
<span id="235">235</span>
<span id="236">236</span>
<span id="237">237</span>
<span id="238">238</span>
<span id="239">239</span>
<span id="240">240</span>
<span id="241">241</span>
<span id="242">242</span>
<span id="243">243</span>
<span id="244">244</span>
<span id="245">245</span>
<span id="246">246</span>
<span id="247">247</span>
<span id="248">248</span>
<span id="249">249</span>
<span id="250">250</span>
<span id="251">251</span>
<span id="252">252</span>
<span id="253">253</span>
<span id="254">254</span>
<span id="255">255</span>
<span id="256">256</span>
<span id="257">257</span>
<span id="258">258</span>
<span id="259">259</span>
<span id="260">260</span>
<span id="261">261</span>
<span id="262">262</span>
<span id="263">263</span>
<span id="264">264</span>
<span id="265">265</span>
<span id="266">266</span>
<span id="267">267</span>
<span id="268">268</span>
<span id="269">269</span>
<span id="270">270</span>
<span id="271">271</span>
<span id="272">272</span>
<span id="273">273</span>
<span id="274">274</span>
<span id="275">275</span>
<span id="276">276</span>
<span id="277">277</span>
<span id="278">278</span>
<span id="279">279</span>
<span id="280">280</span>
<span id="281">281</span>
<span id="282">282</span>
<span id="283">283</span>
<span id="284">284</span>
<span id="285">285</span>
<span id="286">286</span>
<span id="287">287</span>
<span id="288">288</span>
<span id="289">289</span>
<span id="290">290</span>
<span id="291">291</span>
<span id="292">292</span>
<span id="293">293</span>
<span id="294">294</span>
<span id="295">295</span>
<span id="296">296</span>
<span id="297">297</span>
<span id="298">298</span>
<span id="299">299</span>
<span id="300">300</span>
<span id="301">301</span>
<span id="302">302</span>
<span id="303">303</span>
</pre><pre class="rust"><code><span class="comment">/*
* Copyright (C) 2022 Aravinth Manivannan &lt;realaravinth@batsense.net&gt;
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see &lt;https://www.gnu.org/licenses/&gt;.
*/
//use actix::prelude::*;
</span><span class="kw">use </span>actix_web::{web, HttpResponse, Responder};
<span class="kw">use </span>libmcaptcha::{
defense::LevelBuilder, master::messages::AddSiteBuilder, DefenseBuilder,
MCaptchaBuilder,
};
<span class="kw">use </span>serde::{Deserialize, Serialize};
<span class="kw">use </span><span class="kw">crate</span>::errors::<span class="kw-2">*</span>;
<span class="comment">//use crate::stats::record::record_fetch;
</span><span class="kw">use </span><span class="kw">crate</span>::AppData;
<span class="kw">use </span><span class="kw">crate</span>::V1_API_ROUTES;
<span class="attribute">#[derive(Clone, Debug, Deserialize, Serialize)]
</span><span class="kw">pub struct </span>GetConfigPayload {
<span class="kw">pub </span>key: String,
}
<span class="comment">// API keys are mcaptcha actor names
</span><span class="doccomment">/// get PoW configuration for an mcaptcha key
</span><span class="attribute">#[my_codegen::post(path = <span class="string">&quot;V1_API_ROUTES.pow.get_config()&quot;</span>)]
</span><span class="kw">pub async fn </span>get_config(
payload: web::Json&lt;GetConfigPayload&gt;,
data: AppData,
) -&gt; ServiceResult&lt;<span class="kw">impl </span>Responder&gt; {
<span class="comment">//if res.exists.is_none() {
</span><span class="kw">if </span>!data.db.captcha_exists(<span class="prelude-val">None</span>, <span class="kw-2">&amp;</span>payload.key).<span class="kw">await</span><span class="question-mark">? </span>{
<span class="kw">return </span><span class="prelude-val">Err</span>(ServiceError::TokenNotFound);
}
<span class="kw">let </span>payload = payload.into_inner();
<span class="kw">match </span>data.captcha.get_pow(payload.key.clone()).<span class="kw">await </span>{
<span class="prelude-val">Ok</span>(<span class="prelude-val">Some</span>(config)) =&gt; {
data.stats.record_fetch(<span class="kw-2">&amp;</span>data, <span class="kw-2">&amp;</span>payload.key).<span class="kw">await</span><span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(HttpResponse::Ok().json(config))
}
<span class="prelude-val">Ok</span>(<span class="prelude-val">None</span>) =&gt; {
init_mcaptcha(<span class="kw-2">&amp;</span>data, <span class="kw-2">&amp;</span>payload.key).<span class="kw">await</span><span class="question-mark">?</span>;
<span class="kw">let </span>config = data
.captcha
.get_pow(payload.key.clone())
.<span class="kw">await
</span>.expect(<span class="string">&quot;mcaptcha should be initialized and ready to go&quot;</span>);
<span class="comment">// background it. would require data::Data to be static
// to satidfy lifetime
</span>data.stats.record_fetch(<span class="kw-2">&amp;</span>data, <span class="kw-2">&amp;</span>payload.key).<span class="kw">await</span><span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(HttpResponse::Ok().json(config))
}
<span class="prelude-val">Err</span>(e) =&gt; <span class="prelude-val">Err</span>(e.into()),
}
<span class="comment">// match res.exists {
// Some(true) =&gt; {
// match data.captcha.get_pow(payload.key.clone()).await {
// Ok(Some(config)) =&gt; {
// record_fetch(&amp;payload.key, &amp;data.db).await;
// Ok(HttpResponse::Ok().json(config))
// }
// Ok(None) =&gt; {
// init_mcaptcha(&amp;data, &amp;payload.key).await?;
// let config = data
// .captcha
// .get_pow(payload.key.clone())
// .await
// .expect(&quot;mcaptcha should be initialized and ready to go&quot;);
// // background it. would require data::Data to be static
// // to satidfy lifetime
// record_fetch(&amp;payload.key, &amp;data.db).await;
// Ok(HttpResponse::Ok().json(config))
// }
// Err(e) =&gt; Err(e.into()),
// }
// }
//
// Some(false) =&gt; Err(ServiceError::TokenNotFound),
// None =&gt; Err(ServiceError::TokenNotFound),
// }
</span>}
<span class="doccomment">/// Call this when [MCaptcha][libmcaptcha::MCaptcha] is not in master.
///
/// This fn gets mcaptcha config from database, builds [Defense][libmcaptcha::Defense],
/// creates [MCaptcha][libmcaptcha::MCaptcha] and adds it to [Master][libmcaptcha::Defense]
</span><span class="kw">pub async fn </span>init_mcaptcha(data: <span class="kw-2">&amp;</span>AppData, key: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">println!</span>(<span class="string">&quot;Initializing captcha&quot;</span>);
<span class="comment">// get levels
</span><span class="kw">let </span>levels = data.db.get_captcha_levels(<span class="prelude-val">None</span>, key).<span class="kw">await</span><span class="question-mark">?</span>;
<span class="kw">let </span>duration = data.db.get_captcha_cooldown(key).<span class="kw">await</span><span class="question-mark">?</span>;
<span class="comment">// build defense
</span><span class="kw">let </span><span class="kw-2">mut </span>defense = DefenseBuilder::default();
<span class="kw">for </span>level <span class="kw">in </span>levels.iter() {
<span class="kw">let </span>level = LevelBuilder::default()
.visitor_threshold(level.visitor_threshold <span class="kw">as </span>u32)
.difficulty_factor(level.difficulty_factor <span class="kw">as </span>u32)
.unwrap()
.build()
.unwrap();
defense.add_level(level).unwrap();
}
<span class="kw">let </span>defense = defense.build()<span class="question-mark">?</span>;
<span class="macro">println!</span>(<span class="string">&quot;{:?}&quot;</span>, defense);
<span class="comment">// create captcha
</span><span class="kw">let </span>mcaptcha = MCaptchaBuilder::default()
.defense(defense)
<span class="comment">// leaky bucket algorithm&#39;s emission interval
</span>.duration(duration <span class="kw">as </span>u64)
<span class="comment">// .cache(cache)
</span>.build()
.unwrap();
<span class="comment">// add captcha to master
</span><span class="kw">let </span>msg = AddSiteBuilder::default()
.id(key.into())
.mcaptcha(mcaptcha)
.build()
.unwrap();
data.captcha.add_site(msg).<span class="kw">await</span><span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="attribute">#[cfg(test)]
</span><span class="kw">pub mod </span>tests {
<span class="kw">use crate</span>::<span class="kw-2">*</span>;
<span class="kw">use </span>libmcaptcha::pow::PoWConfig;
<span class="attribute">#[actix_rt::test]
</span><span class="kw">async fn </span>get_pow_config_works_pg() {
<span class="kw">let </span>data = <span class="kw">crate</span>::tests::pg::get_data().<span class="kw">await</span>;
get_pow_config_works(data).<span class="kw">await</span>;
}
<span class="attribute">#[actix_rt::test]
</span><span class="kw">async fn </span>get_pow_config_works_maria() {
<span class="kw">let </span>data = <span class="kw">crate</span>::tests::maria::get_data().<span class="kw">await</span>;
get_pow_config_works(data).<span class="kw">await</span>;
}
<span class="kw">pub async fn </span>get_pow_config_works(data: ArcData) {
<span class="kw">use super</span>::<span class="kw-2">*</span>;
<span class="kw">use </span><span class="kw">crate</span>::tests::<span class="kw-2">*</span>;
<span class="kw">use crate</span>::<span class="kw-2">*</span>;
<span class="kw">use </span>actix_web::test;
<span class="kw">const </span>NAME: <span class="kw-2">&amp;</span>str = <span class="string">&quot;powusrworks&quot;</span>;
<span class="kw">const </span>PASSWORD: <span class="kw-2">&amp;</span>str = <span class="string">&quot;testingpas&quot;</span>;
<span class="kw">const </span>EMAIL: <span class="kw-2">&amp;</span>str = <span class="string">&quot;randomuser@a.com&quot;</span>;
<span class="kw">let </span>data = <span class="kw-2">&amp;</span>data;
delete_user(data, NAME).<span class="kw">await</span>;
register_and_signin(data, NAME, EMAIL, PASSWORD).<span class="kw">await</span>;
<span class="kw">let </span>(<span class="kw">_</span>, _signin_resp, token_key) = add_levels_util(data, NAME, PASSWORD).<span class="kw">await</span>;
<span class="kw">let </span>app = <span class="macro">get_app!</span>(data).<span class="kw">await</span>;
<span class="kw">let </span>get_config_payload = GetConfigPayload {
key: token_key.key.clone(),
};
<span class="comment">// update and check changes
</span><span class="kw">let </span>url = V1_API_ROUTES.pow.get_config;
<span class="macro">println!</span>(<span class="string">&quot;{}&quot;</span>, <span class="kw-2">&amp;</span>url);
<span class="kw">let </span>get_config_resp = test::call_service(
<span class="kw-2">&amp;</span>app,
<span class="macro">post_request!</span>(<span class="kw-2">&amp;</span>get_config_payload, V1_API_ROUTES.pow.get_config)
.to_request(),
)
.<span class="kw">await</span>;
<span class="macro">assert_eq!</span>(get_config_resp.status(), StatusCode::OK);
<span class="kw">let </span>config: PoWConfig = test::read_body_json(get_config_resp).<span class="kw">await</span>;
<span class="macro">assert_eq!</span>(config.difficulty_factor, L1.difficulty_factor);
}
<span class="attribute">#[actix_rt::test]
</span><span class="kw">async fn </span>pow_difficulty_factor_increases_on_visitor_count_increase_pg() {
<span class="kw">let </span>data = <span class="kw">crate</span>::tests::pg::get_data().<span class="kw">await</span>;
pow_difficulty_factor_increases_on_visitor_count_increase(data).<span class="kw">await</span>;
}
<span class="attribute">#[actix_rt::test]
</span><span class="kw">async fn </span>pow_difficulty_factor_increases_on_visitor_count_increase_maria() {
<span class="kw">let </span>data = <span class="kw">crate</span>::tests::maria::get_data().<span class="kw">await</span>;
pow_difficulty_factor_increases_on_visitor_count_increase(data).<span class="kw">await</span>;
}
<span class="kw">pub async fn </span>pow_difficulty_factor_increases_on_visitor_count_increase(
data: ArcData,
) {
<span class="kw">use super</span>::<span class="kw-2">*</span>;
<span class="kw">use </span><span class="kw">crate</span>::tests::<span class="kw-2">*</span>;
<span class="kw">use crate</span>::<span class="kw-2">*</span>;
<span class="kw">use </span>actix_web::test;
<span class="kw">use </span>libmcaptcha::defense::Level;
<span class="kw">use </span><span class="kw">crate</span>::api::v1::mcaptcha::create::CreateCaptcha;
<span class="kw">use </span><span class="kw">crate</span>::api::v1::mcaptcha::create::MCaptchaDetails;
<span class="kw">const </span>NAME: <span class="kw-2">&amp;</span>str = <span class="string">&quot;powusrworks2&quot;</span>;
<span class="kw">const </span>PASSWORD: <span class="kw-2">&amp;</span>str = <span class="string">&quot;testingpas&quot;</span>;
<span class="kw">const </span>EMAIL: <span class="kw-2">&amp;</span>str = <span class="string">&quot;randomuser2@a.com&quot;</span>;
<span class="kw">pub const </span>L1: Level = Level {
difficulty_factor: <span class="number">10</span>,
visitor_threshold: <span class="number">10</span>,
};
<span class="kw">pub const </span>L2: Level = Level {
difficulty_factor: <span class="number">20</span>,
visitor_threshold: <span class="number">20</span>,
};
<span class="kw">pub const </span>L3: Level = Level {
difficulty_factor: <span class="number">30</span>,
visitor_threshold: <span class="number">30</span>,
};
<span class="kw">let </span>data = <span class="kw-2">&amp;</span>data;
<span class="kw">let </span>levels = [L1, L2, L3];
delete_user(data, NAME).<span class="kw">await</span>;
<span class="kw">let </span>(<span class="kw">_</span>, signin_resp) = register_and_signin(data, NAME, EMAIL, PASSWORD).<span class="kw">await</span>;
<span class="kw">let </span>cookies = <span class="macro">get_cookie!</span>(signin_resp);
<span class="kw">let </span>app = <span class="macro">get_app!</span>(data).<span class="kw">await</span>;
<span class="kw">let </span>create_captcha = CreateCaptcha {
levels: levels.into(),
duration: <span class="number">30</span>,
description: <span class="string">&quot;dummy&quot;</span>.into(),
};
<span class="comment">// 1. add level
</span><span class="kw">let </span>add_token_resp = test::call_service(
<span class="kw-2">&amp;</span>app,
<span class="macro">post_request!</span>(<span class="kw-2">&amp;</span>create_captcha, V1_API_ROUTES.captcha.create)
.cookie(cookies.clone())
.to_request(),
)
.<span class="kw">await</span>;
<span class="macro">assert_eq!</span>(add_token_resp.status(), StatusCode::OK);
<span class="kw">let </span>token_key: MCaptchaDetails = test::read_body_json(add_token_resp).<span class="kw">await</span>;
<span class="kw">let </span>get_config_payload = GetConfigPayload {
key: token_key.key.clone(),
};
<span class="kw">let </span>url = V1_API_ROUTES.pow.get_config;
<span class="kw">let </span><span class="kw-2">mut </span>prev = <span class="number">0</span>;
<span class="kw">for </span>(count, l) <span class="kw">in </span>levels.iter().enumerate() {
<span class="kw">for </span>l <span class="kw">in </span>prev..l.visitor_threshold * <span class="number">2 </span>{
<span class="kw">let </span>get_config_resp = test::call_service(
<span class="kw-2">&amp;</span>app,
<span class="macro">post_request!</span>(<span class="kw-2">&amp;</span>get_config_payload, V1_API_ROUTES.pow.get_config)
.to_request(),
)
.<span class="kw">await</span>;
}
<span class="kw">let </span>get_config_resp = test::call_service(
<span class="kw-2">&amp;</span>app,
<span class="macro">post_request!</span>(<span class="kw-2">&amp;</span>get_config_payload, V1_API_ROUTES.pow.get_config)
.to_request(),
)
.<span class="kw">await</span>;
<span class="kw">let </span>config: PoWConfig = test::read_body_json(get_config_resp).<span class="kw">await</span>;
<span class="macro">println!</span>(
<span class="string">&quot;[{count}] received difficulty_factor: {} prev difficulty_factor {}&quot;</span>,
config.difficulty_factor, prev
);
<span class="kw">if </span>count == levels.len() - <span class="number">1 </span>{
<span class="macro">assert!</span>(config.difficulty_factor == prev);
} <span class="kw">else </span>{
<span class="macro">assert!</span>(config.difficulty_factor &gt; prev);
}
prev = config.difficulty_factor;
}
<span class="comment">// update and check changes
</span>}
}
</code></pre></div>
</section></div></main><div id="rustdoc-vars" data-root-path="../../../../../" data-current-crate="mcaptcha" data-themes="ayu,dark,light" data-resource-suffix="" data-rustdoc-version="1.66.0 (69f9c33d7 2022-12-12)" ></div></body></html>
Reference in New Issue View Git Blame Copy Permalink