Skip to content

Experiment: Add test to upload and consume 1000 key packages per client #5376

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
supersven wants to merge 1 commit into
base: develop
Choose a base branch
from
Draft

Experiment: Add test to upload and consume 1000 key packages per client #5376

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
35 changes: 35 additions & 0 deletions integration/test/Test/MLS/KeyPackage.hs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ import API.BrigInternal
import MLS.Util
import SetupHelpers
import Testlib.Prelude
import UnliftIO (pooledForConcurrentlyN)

testDeleteKeyPackages :: App ()
testDeleteKeyPackages = do
Expand Down Expand Up @@ -452,3 +453,37 @@ testReplaceKeyPackagesV7 = do
testErrorCases (Just [suite]) altSuiteKeyPackages
testErrorCases (Just [suite]) (oldSuiteKeyPackages <> altSuiteKeyPackages <> suiteKeyPackages)
testErrorCases (Just [suite]) []

testUploadClaim1000KeyPackages :: App ()
testUploadClaim1000KeyPackages = do
let suite = def
keyPackageCount = 1000

alice <- randomUser OwnDomain def
alice1 <- createMLSClient def {ciphersuites = [suite]} alice

bob <- randomUser OwnDomain def
bob1 <- createMLSClient def {ciphersuites = [suite]} bob

-- Generate 1000 key packages in parallel (16 concurrent workers)
kps <- map fst <$> pooledForConcurrentlyN 16 [1 .. keyPackageCount] \_ -> do
generateKeyPackage alice1 suite

-- Upload all 1000 key packages in one batched call
void $ uploadKeyPackages alice1 kps >>= getBody 201

-- Verify count is 1000
bindResponse (countKeyPackages suite alice1) $ \resp -> do
resp.status `shouldMatchInt` 200
resp.json %. "count" `shouldMatchInt` keyPackageCount

-- Bob claims one key package from alice
bindResponse (claimKeyPackages suite bob1 alice) $ \resp -> do
resp.status `shouldMatchInt` 200
kpBundles <- resp.json %. "key_packages" & asList
length kpBundles `shouldMatchInt` 1

-- Verify count is now 999 (one claimed)
bindResponse (countKeyPackages suite alice1) $ \resp -> do
resp.status `shouldMatchInt` 200
resp.json %. "count" `shouldMatchInt` (keyPackageCount - 1)