fix tests

DefGuard · Jan 3, 2025 · 00cad1d · 00cad1d
1 parent 1cf32a6
commit 00cad1d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/enterprise/limits.rs b/src/enterprise/limits.rs
@@ -111,8 +111,6 @@ impl Counts {
 
 #[cfg(test)]
 mod test {
-    use std::u32;
-
     use chrono::{TimeDelta, Utc};
 
     use crate::enterprise::license::{set_cached_license, License, LicenseLimits};

diff --git a/tests/common/mod.rs b/tests/common/mod.rs
@@ -138,11 +138,12 @@ pub async fn make_base_client(
     let failed_logins = FailedLoginMap::new();
     let failed_logins = Arc::new(Mutex::new(failed_logins));
 
-    let license = License::new_with_default_limits(
+    let license = License::new(
         "test_customer".to_string(),
         false,
         // Permanent license
         None,
+        None,
     );
 
     set_cached_license(Some(license));

diff --git a/tests/openid_login.rs b/tests/openid_login.rs
@@ -88,10 +88,11 @@ async fn test_openid_providers() {
     assert!(redirect_uri.is_some());
 
     // Test that the endpoint is forbidden when the license is expired
-    let new_license = License::new_with_default_limits(
+    let new_license = License::new(
         "test".to_string(),
         false,
         Some(Utc::now() - Duration::days(1)),
+        None,
     );
     set_cached_license(Some(new_license));
     let response = client.get("/api/v1/openid/auth_info").send().await;