65 lines
3.1 KiB
TypeScript
65 lines
3.1 KiB
TypeScript
import Sinon = require("sinon");
|
|
import BluebirdPromise = require("bluebird");
|
|
|
|
import { TOTPSecretDocument } from "../../../../../src/server/lib/storage/TOTPSecretDocument";
|
|
import { U2FRegistrationDocument } from "../../../../../src/server/lib/storage/U2FRegistrationDocument";
|
|
import { U2FRegistration } from "../../../../../src/types/U2FRegistration";
|
|
import { TOTPSecret } from "../../../../../src/types/TOTPSecret";
|
|
import { AuthenticationTraceDocument } from "../../../../../src/server/lib/storage/AuthenticationTraceDocument";
|
|
import { IdentityValidationDocument } from "../../../../../src/server/lib/storage/IdentityValidationDocument";
|
|
|
|
import { IUserDataStore } from "../../../../../src/server/lib/storage/IUserDataStore";
|
|
|
|
export class UserDataStoreStub implements IUserDataStore {
|
|
saveU2FRegistrationStub: Sinon.SinonStub;
|
|
retrieveU2FRegistrationStub: Sinon.SinonStub;
|
|
saveAuthenticationTraceStub: Sinon.SinonStub;
|
|
retrieveLatestAuthenticationTracesStub: Sinon.SinonStub;
|
|
produceIdentityValidationTokenStub: Sinon.SinonStub;
|
|
consumeIdentityValidationTokenStub: Sinon.SinonStub;
|
|
saveTOTPSecretStub: Sinon.SinonStub;
|
|
retrieveTOTPSecretStub: Sinon.SinonStub;
|
|
|
|
constructor() {
|
|
this.saveU2FRegistrationStub = Sinon.stub();
|
|
this.retrieveU2FRegistrationStub = Sinon.stub();
|
|
this.saveAuthenticationTraceStub = Sinon.stub();
|
|
this.retrieveLatestAuthenticationTracesStub = Sinon.stub();
|
|
this.produceIdentityValidationTokenStub = Sinon.stub();
|
|
this.consumeIdentityValidationTokenStub = Sinon.stub();
|
|
this.saveTOTPSecretStub = Sinon.stub();
|
|
this.retrieveTOTPSecretStub = Sinon.stub();
|
|
}
|
|
|
|
saveU2FRegistration(userId: string, appId: string, registration: U2FRegistration): BluebirdPromise<void> {
|
|
return this.saveU2FRegistrationStub(userId, appId, registration);
|
|
}
|
|
|
|
retrieveU2FRegistration(userId: string, appId: string): BluebirdPromise<U2FRegistrationDocument> {
|
|
return this.retrieveU2FRegistrationStub(userId, appId);
|
|
}
|
|
|
|
saveAuthenticationTrace(userId: string, isAuthenticationSuccessful: boolean): BluebirdPromise<void> {
|
|
return this.saveAuthenticationTraceStub(userId, isAuthenticationSuccessful);
|
|
}
|
|
|
|
retrieveLatestAuthenticationTraces(userId: string, isAuthenticationSuccessful: boolean, count: number): BluebirdPromise<AuthenticationTraceDocument[]> {
|
|
return this.retrieveLatestAuthenticationTracesStub(userId, isAuthenticationSuccessful, count);
|
|
}
|
|
|
|
produceIdentityValidationToken(userId: string, token: string, challenge: string, maxAge: number): BluebirdPromise<any> {
|
|
return this.produceIdentityValidationTokenStub(userId, token, challenge, maxAge);
|
|
}
|
|
|
|
consumeIdentityValidationToken(token: string, challenge: string): BluebirdPromise<IdentityValidationDocument> {
|
|
return this.consumeIdentityValidationTokenStub(token, challenge);
|
|
}
|
|
|
|
saveTOTPSecret(userId: string, secret: TOTPSecret): BluebirdPromise<void> {
|
|
return this.saveTOTPSecretStub(userId, secret);
|
|
}
|
|
|
|
retrieveTOTPSecret(userId: string): BluebirdPromise<TOTPSecretDocument> {
|
|
return this.retrieveTOTPSecretStub(userId);
|
|
}
|
|
} |