ts-mls/dist/test/scenario/remove.test.js

[](https://github.com/LukaJCB/ts-mls/actions/workflows/ci.yml) [](https://badge.fury.io/js/ts-mls) [![Coverage Status](https://co

import { createGroup, joinGroup, makePskIndex } from "../../src/clientState";
import { createCommit } from "../../src/createCommit";
import { processPrivateMessage } from "../../src/processMessages";
import { emptyPskIndex } from "../../src/pskIndex";
import { ciphersuites, getCiphersuiteFromName, getCiphersuiteImpl } from "../../src/crypto/ciphersuite";
import { generateKeyPackage } from "../../src/keyPackage";
import { checkHpkeKeysMatch } from "../crypto/keyMatch";
import { cannotMessageAnymore, testEveryoneCanMessageEveryone } from "./common";
import { defaultLifetime } from "../../src/lifetime";
import { defaultCapabilities } from "../../src/defaultCapabilities";
import { UsageError } from "../../src/mlsError";
for (const cs of Object.keys(ciphersuites)) {
    test(`Remove ${cs}`, async () => {
        await remove(cs);
    });
}
async function remove(cipherSuite) {
    const impl = await getCiphersuiteImpl(getCiphersuiteFromName(cipherSuite));
    const aliceCredential = { credentialType: "basic", identity: new TextEncoder().encode("alice") };
    const alice = await generateKeyPackage(aliceCredential, defaultCapabilities(), defaultLifetime, [], impl);
    const groupId = new TextEncoder().encode("group1");
    let aliceGroup = await createGroup(groupId, alice.publicPackage, alice.privatePackage, [], impl);
    const bobCredential = { credentialType: "basic", identity: new TextEncoder().encode("bob") };
    const bob = await generateKeyPackage(bobCredential, defaultCapabilities(), defaultLifetime, [], impl);
    const charlieCredential = { credentialType: "basic", identity: new TextEncoder().encode("charlie") };
    const charlie = await generateKeyPackage(charlieCredential, defaultCapabilities(), defaultLifetime, [], impl);
    const addBobProposal = {
        proposalType: "add",
        add: {
            keyPackage: bob.publicPackage,
        },
    };
    const addCharlieProposal = {
        proposalType: "add",
        add: {
            keyPackage: charlie.publicPackage,
        },
    };
    const addBobAndCharlieCommitResult = await createCommit(aliceGroup, emptyPskIndex, false, [addBobProposal, addCharlieProposal], impl);
    aliceGroup = addBobAndCharlieCommitResult.newState;
    let bobGroup = await joinGroup(addBobAndCharlieCommitResult.welcome, bob.publicPackage, bob.privatePackage, emptyPskIndex, impl, aliceGroup.ratchetTree);
    expect(bobGroup.keySchedule.epochAuthenticator).toStrictEqual(aliceGroup.keySchedule.epochAuthenticator);
    let charlieGroup = await joinGroup(addBobAndCharlieCommitResult.welcome, charlie.publicPackage, charlie.privatePackage, emptyPskIndex, impl, aliceGroup.ratchetTree);
    expect(charlieGroup.keySchedule.epochAuthenticator).toStrictEqual(aliceGroup.keySchedule.epochAuthenticator);
    const removeBobProposal = {
        proposalType: "remove",
        remove: {
            removed: bobGroup.privatePath.leafIndex,
        },
    };
    const removeBobCommitResult = await createCommit(aliceGroup, emptyPskIndex, false, [removeBobProposal], impl);
    aliceGroup = removeBobCommitResult.newState;
    if (removeBobCommitResult.commit.wireformat !== "mls_private_message")
        throw new Error("Expected private message");
    const bobProcessCommitResult = await processPrivateMessage(bobGroup, removeBobCommitResult.commit.privateMessage, makePskIndex(bobGroup, {}), impl);
    // bob is removed here
    bobGroup = bobProcessCommitResult.newState;
    const charlieProcessCommitResult = await processPrivateMessage(charlieGroup, removeBobCommitResult.commit.privateMessage, makePskIndex(charlieGroup, {}), impl);
    charlieGroup = charlieProcessCommitResult.newState;
    expect(bobGroup.groupActiveState).toStrictEqual({ kind: "removedFromGroup" });
    //creating a message will fail now
    expect(createCommit(bobGroup, emptyPskIndex, false, [], impl)).rejects.toThrow(UsageError);
    await cannotMessageAnymore(bobGroup, impl);
    await checkHpkeKeysMatch(aliceGroup, impl);
    await checkHpkeKeysMatch(charlieGroup, impl);
    await testEveryoneCanMessageEveryone([aliceGroup, charlieGroup], impl);
}
//# sourceMappingURL=remove.test.js.map