Skip to content

Commit 3f28eb9

Browse files
committed
tests/util: Remove CreatedApiToken dependency
1 parent a52cea2 commit 3f28eb9

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

src/tests/util.rs

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,6 @@ mod mock_request;
5050
mod response;
5151
mod test_app;
5252

53-
use crate::controllers::token::CreatedApiToken;
5453
use mock_request::MockRequest;
5554
pub use mock_request::MockRequestExt;
5655
pub use response::Response;
@@ -332,28 +331,27 @@ impl MockCookieUser {
332331
.maybe_expired_at(expired_at)
333332
.build();
334333

335-
let token = CreatedApiToken {
336-
plaintext,
337-
model: new_token.insert(&mut conn).await.unwrap(),
338-
};
334+
let token = new_token.insert(&mut conn).await.unwrap();
339335

340336
MockTokenUser {
341337
app: self.app.clone(),
342338
token,
339+
plaintext,
343340
}
344341
}
345342
}
346343

347344
/// A type that can generate token authenticated requests
348345
pub struct MockTokenUser {
349346
app: TestApp,
350-
token: CreatedApiToken,
347+
token: ApiToken,
348+
plaintext: PlainToken,
351349
}
352350

353351
impl RequestHelper for MockTokenUser {
354352
fn request_builder(&self, method: Method, path: &str) -> MockRequest {
355353
let mut request = req(method, path);
356-
request.header(header::AUTHORIZATION, self.token.plaintext.expose_secret());
354+
request.header(header::AUTHORIZATION, self.plaintext.expose_secret());
357355
request
358356
}
359357

@@ -365,10 +363,10 @@ impl RequestHelper for MockTokenUser {
365363
impl MockTokenUser {
366364
/// Returns a reference to the database `ApiToken` model
367365
pub fn as_model(&self) -> &ApiToken {
368-
&self.token.model
366+
&self.token
369367
}
370368

371369
pub fn plaintext(&self) -> &PlainToken {
372-
&self.token.plaintext
370+
&self.plaintext
373371
}
374372
}

0 commit comments

Comments
 (0)