From 9eb53840380b26a11a48ac71418a8adf74b3b780 Mon Sep 17 00:00:00 2001
From: Jean-Paul Calderone <exarkun@twistedmatrix.com>
Date: Thu, 26 Mar 2020 10:08:35 -0400
Subject: [PATCH] Make the strategy able to produce more voucher states
---
src/_zkapauthorizer/tests/strategies.py | 15 +++++++++++++--
1 file changed, 13 insertions(+), 2 deletions(-)
diff --git a/src/_zkapauthorizer/tests/strategies.py b/src/_zkapauthorizer/tests/strategies.py
index 130fd30..9157bde 100644
--- a/src/_zkapauthorizer/tests/strategies.py
+++ b/src/_zkapauthorizer/tests/strategies.py
@@ -75,6 +75,8 @@ from ..model import (
Voucher,
Pending,
DoubleSpend,
+ Unpaid,
+ Error,
Redeemed,
)
@@ -302,14 +304,23 @@ def voucher_states():
"""
return one_of(
just(Pending()),
+ builds(
+ Redeemed,
+ finished=datetimes(),
+ token_count=one_of(integers(min_value=1)),
+ ),
builds(
DoubleSpend,
finished=datetimes(),
),
builds(
- Redeemed,
+ Unpaid,
finished=datetimes(),
- token_count=one_of(integers(min_value=1)),
+ ),
+ builds(
+ Error,
+ finished=datetimes(),
+ details=text(),
),
)
--
GitLab