From 42da983319467541f886d75d92f4c368210a8101 Mon Sep 17 00:00:00 2001
From: Jean-Paul Calderone <exarkun@twistedmatrix.com>
Date: Thu, 28 May 2020 10:15:29 -0400
Subject: [PATCH] use the paired_tokens helper
---
src/_zkapauthorizer/tests/test_model.py | 29 ++++++++-----------------
1 file changed, 9 insertions(+), 20 deletions(-)
diff --git a/src/_zkapauthorizer/tests/test_model.py b/src/_zkapauthorizer/tests/test_model.py
index 80aa1f2..46a794e 100644
--- a/src/_zkapauthorizer/tests/test_model.py
+++ b/src/_zkapauthorizer/tests/test_model.py
@@ -884,35 +884,24 @@ class UnblindedTokenStoreTests(TestCase):
vouchers(),
dummy_ristretto_keys(),
booleans(),
- integers(min_value=1, max_value=100),
integers(min_value=1),
data(),
)
- def test_not_enough_unblinded_tokens(self, get_config, now, voucher_value, public_key, completed, num_tokens, extra, data):
+ def test_not_enough_unblinded_tokens(self, get_config, now, voucher_value, public_key, completed, extra, data):
"""
``get_unblinded_tokens`` raises ``NotEnoughTokens`` if ``count`` is
greater than the number of unblinded tokens in the store.
"""
- random = data.draw(
- lists(
- random_tokens(),
- min_size=num_tokens,
- max_size=num_tokens,
- unique=True,
- ),
- )
- unblinded = data.draw(
- lists(
- unblinded_tokens(),
- min_size=num_tokens,
- max_size=num_tokens,
- unique=True,
- ),
- )
+ random, unblinded = paired_tokens(data)
+ num_tokens = len(random)
store = self.useFixture(TemporaryVoucherStore(get_config, lambda: now)).store
store.add(voucher_value, len(random), 0, lambda: random)
- store.insert_unblinded_tokens_for_voucher(voucher_value, public_key, unblinded, completed)
-
+ store.insert_unblinded_tokens_for_voucher(
+ voucher_value,
+ public_key,
+ unblinded,
+ completed,
+ )
self.assertThat(
lambda: store.get_unblinded_tokens(num_tokens + extra),
raises(NotEnoughTokens),
--
GitLab