1 files changed, 76 insertions, 0 deletions
diff --git a/test_utils/managers/cashout_methods.py b/test_utils/managers/cashout_methods.py
new file mode 100644
index 0000000..c338676
--- /dev/null
+++ b/test_utils/managers/cashout_methods.py
@@ -0,0 +1,76 @@
+from generalresearch.models.thl.wallet import PayoutType, Currency
+from generalresearch.models.thl.wallet.cashout_method import (
+    CashoutMethod,
+    TangoCashoutMethodData,
+    AmtCashoutMethodData,
+)
+import random
+
+from uuid import uuid4
+
+
+def random_ext_id(base: str = "U02"):
+    suffix = random.randint(0, 99999)
+    return f"{base}{suffix:05d}"
+
+
+EXAMPLE_TANGO_CASHOUT_METHODS = [
+    CashoutMethod(
+        id=uuid4().hex,
+        last_updated="2021-06-23T20:45:38.239182Z",
+        is_live=True,
+        type=PayoutType.TANGO,
+        ext_id=random_ext_id(),
+        name="Safeway eGift Card $25",
+        data=TangoCashoutMethodData(
+            value_type="fixed", countries=["US"], utid=random_ext_id()
+        ),
+        user=None,
+        image_url="https://d30s7yzk2az89n.cloudfront.net/images/brands/b694446-1200w-326ppi.png",
+        original_currency=Currency.USD,
+        min_value=2500,
+        max_value=2500,
+    ),
+    CashoutMethod(
+        id=uuid4().hex,
+        last_updated="2021-06-23T20:45:38.239182Z",
+        is_live=True,
+        type=PayoutType.TANGO,
+        ext_id=random_ext_id(),
+        name="Amazon.it Gift Certificate",
+        data=TangoCashoutMethodData(
+            value_type="variable", countries=["IT"], utid="U006961"
+        ),
+        user=None,
+        image_url="https://d30s7yzk2az89n.cloudfront.net/images/brands/b405753-1200w-326ppi.png",
+        original_currency=Currency.EUR,
+        min_value=1,
+        max_value=10000,
+    ),
+]
+
+AMT_ASSIGNMENT_CASHOUT_METHOD = CashoutMethod(
+    id=uuid4().hex,
+    last_updated="2021-06-23T20:45:38.239182Z",
+    is_live=True,
+    type=PayoutType.AMT,
+    ext_id=None,
+    name="AMT Assignment",
+    data=AmtCashoutMethodData(),
+    user=None,
+    min_value=1,
+    max_value=5,
+)
+
+AMT_BONUS_CASHOUT_METHOD = CashoutMethod(
+    id=uuid4().hex,
+    last_updated="2021-06-23T20:45:38.239182Z",
+    is_live=True,
+    type=PayoutType.AMT,
+    ext_id=None,
+    name="AMT Bonus",
+    data=AmtCashoutMethodData(),
+    user=None,
+    min_value=7,
+    max_value=4000,
+)