mirrors/signal-desktop
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages 2 Wiki Activity Actions

Test rate-limiting, stories in mock server

Browse source
This commit is contained in:
Fedor Indutny 2022-09-19 15:08:55 -07:00 • committed by GitHub
parent 450051e541
commit f9453c64dd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 212 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

27
ts/test-mock/bootstrap.ts
View file

@ -62,6 +62,7 @@ export type BootstrapOptions = Readonly<{
linkedDevices?: number;
contactCount?: number;
contactsWithoutProfileKey?: number;
contactNames?: ReadonlyArray<string>;
contactPreKeyCount?: number;
}>;
@ -71,6 +72,7 @@ type BootstrapInternalOptions = Pick<BootstrapOptions, 'extraConfig'> &
benchmark: boolean;
linkedDevices: number;
contactCount: number;
contactsWithoutProfileKey: number;
contactNames: ReadonlyArray<string>;
}>;
@ -104,6 +106,7 @@ export class Bootstrap {
private readonly options: BootstrapInternalOptions;
private privContacts?: ReadonlyArray<PrimaryDevice>;
private privContactsWithoutProfileKey?: ReadonlyArray<PrimaryDevice>;
private privPhone?: PrimaryDevice;
private privDesktop?: Device;
private storagePath?: string;
@ -118,13 +121,17 @@ export class Bootstrap {
this.options = {
linkedDevices: 5,
contactCount: MAX_CONTACTS,
contactsWithoutProfileKey: 0,
contactNames: CONTACT_NAMES,
benchmark: false,
...options,
};
assert(this.options.contactCount <= this.options.contactNames.length);
assert(
this.options.contactCount + this.options.contactsWithoutProfileKey <=
this.options.contactNames.length
);
}
public async init(): Promise<void> {
@ -137,10 +144,10 @@ export class Bootstrap {
const contactNames = this.options.contactNames.slice(
0,
this.options.contactCount
this.options.contactCount + this.options.contactsWithoutProfileKey
);
this.privContacts = await Promise.all(
const allContacts = await Promise.all(
contactNames.map(async profileName => {
const primary = await this.server.createPrimaryDevice({
profileName,
@ -155,9 +162,15 @@ export class Bootstrap {
})
);
this.privContacts = allContacts.slice(0, this.options.contactCount);
this.privContactsWithoutProfileKey = allContacts.slice(
this.contacts.length
);
this.privPhone = await this.server.createPrimaryDevice({
profileName: 'Myself',
contacts: this.contacts,
contactsWithoutProfileKey: this.contactsWithoutProfileKey,
});
this.storagePath = await fs.mkdtemp(path.join(os.tmpdir(), 'mock-signal-'));
@ -309,6 +322,14 @@ export class Bootstrap {
return this.privContacts;
}
public get contactsWithoutProfileKey(): ReadonlyArray<PrimaryDevice> {
assert(
this.privContactsWithoutProfileKey,
'Bootstrap has to be initialized first, see: bootstrap.init()'
);
return this.privContactsWithoutProfileKey;
}
//
// Private
//