mirror of
https://github.com/mCaptcha/mCaptcha.git
synced 2026-02-11 18:15:39 +00:00
115 lines
3.5 KiB
Rust
115 lines
3.5 KiB
Rust
/*
|
|
* Copyright (C) 2022 Aravinth Manivannan <realaravinth@batsense.net>
|
|
*
|
|
* 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
use actix_auth_middleware::Authentication;
|
|
use actix_web::web::ServiceConfig;
|
|
|
|
mod auth;
|
|
pub mod errors;
|
|
mod panel;
|
|
pub mod routes;
|
|
mod sitemap;
|
|
|
|
pub const NAME: &str = "mCaptcha";
|
|
|
|
pub fn services(cfg: &mut ServiceConfig) {
|
|
auth::services(cfg);
|
|
panel::services(cfg);
|
|
errors::services(cfg);
|
|
cfg.service(sitemap::sitemap);
|
|
}
|
|
|
|
pub fn get_middleware() -> Authentication<routes::Routes> {
|
|
Authentication::with_identity(routes::ROUTES)
|
|
}
|
|
|
|
#[cfg(not(tarpaulin_include))]
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use actix_web::http::StatusCode;
|
|
use actix_web::test;
|
|
|
|
use super::*;
|
|
use crate::tests::*;
|
|
use crate::*;
|
|
|
|
#[actix_rt::test]
|
|
async fn protected_pages_templates_work() {
|
|
const NAME: &str = "templateuser";
|
|
const PASSWORD: &str = "longpassword";
|
|
const EMAIL: &str = "templateuser@a.com";
|
|
|
|
{
|
|
let data = Data::new().await;
|
|
delete_user(NAME, &data).await;
|
|
}
|
|
|
|
register_and_signin(NAME, EMAIL, PASSWORD).await;
|
|
let (data, _, signin_resp, token_key) = add_levels_util(NAME, PASSWORD).await;
|
|
let cookies = get_cookie!(signin_resp);
|
|
|
|
let app = get_app!(data).await;
|
|
|
|
let edit_sitekey_url = PAGES.panel.sitekey.get_edit_advance(&token_key.key);
|
|
let delete_sitekey_url = PAGES.panel.sitekey.get_delete(&token_key.key);
|
|
let urls = vec![
|
|
PAGES.home,
|
|
PAGES.panel.sitekey.add_advance,
|
|
PAGES.panel.sitekey.add_easy,
|
|
PAGES.panel.sitekey.list,
|
|
PAGES.panel.notifications,
|
|
PAGES.panel.settings.home,
|
|
PAGES.panel.settings.delete_account,
|
|
PAGES.panel.settings.update_secret,
|
|
&delete_sitekey_url,
|
|
&edit_sitekey_url,
|
|
];
|
|
|
|
for url in urls.iter() {
|
|
let resp =
|
|
test::call_service(&app, test::TestRequest::get().uri(url).to_request())
|
|
.await;
|
|
assert_eq!(resp.status(), StatusCode::FOUND);
|
|
|
|
let authenticated_resp = test::call_service(
|
|
&app,
|
|
test::TestRequest::get()
|
|
.uri(url)
|
|
.cookie(cookies.clone())
|
|
.to_request(),
|
|
)
|
|
.await;
|
|
|
|
assert_eq!(authenticated_resp.status(), StatusCode::OK);
|
|
}
|
|
|
|
delete_user(NAME, &data).await;
|
|
}
|
|
|
|
#[actix_rt::test]
|
|
async fn public_pages_tempaltes_work() {
|
|
let app = test::init_service(App::new().configure(services)).await;
|
|
let urls = vec![PAGES.auth.login, PAGES.auth.join, PAGES.sitemap];
|
|
|
|
for url in urls.iter() {
|
|
let resp =
|
|
test::call_service(&app, test::TestRequest::get().uri(url).to_request())
|
|
.await;
|
|
|
|
assert_eq!(resp.status(), StatusCode::OK);
|
|
}
|
|
}
|
|
}
|