def test_simple_sync(self):
import stripe
stripe.api_key = settings.STRIPE_SECRET_KEY
# Delete all plans
for plan in stripe.Plan.all()["data"]:
try:
plan.delete()
except InvalidRequestError:
continue
sync_plans()
self.assertTrue("Plan created for test", sys.stdout.getvalue().strip())
sync_plans()
self.assertTrue("ERROR: Plan already exists.", sys.stdout.getvalue().strip())
def test_plan_created(self, plan_create_mock):
sync_plans()
self.assertTrue("Plan created for test", sys.stdout.getvalue().strip())
plan_create_mock.assert_any_call(api_key=settings.STRIPE_SECRET_KEY,
amount=1000,
interval="month",
name="Test Plan 0",
currency="usd",
id="test_id_0")
plan_create_mock.assert_any_call(api_key=settings.STRIPE_SECRET_KEY,
amount=2500,
interval="month",
name="Test Plan 1",
currency="usd",
id="test_id")
plan_create_mock.assert_any_call(api_key=settings.STRIPE_SECRET_KEY,
amount=5000,
interval="month",
name="Test Plan 2",
currency="usd",
id="test_id_2")
plan_create_mock.assert_any_call(api_key=settings.STRIPE_SECRET_KEY,
amount=5000,
interval="month",
name="Test Plan 3",
currency="usd",
id="test_id_3")
plan_create_mock.assert_any_call(api_key=settings.STRIPE_SECRET_KEY,
amount=7000,
interval="month",
name="Test Plan 4",
currency="usd",
id="test_id_4",
trial_period_days=7)
self.assertEqual(5, plan_create_mock.call_count)
def test_plan_created(self, plan_create_mock):
sync_plans(api_key)
self.assertTrue("Plan created for test", sys.stdout.getvalue().strip())
plan_create_mock.assert_any_call(
amount=1000,
interval="month",
name="Test Plan 0",
currency="usd",
id="test_id_0",
interval_count=None,
trial_period_days=None,
statement_descriptor=None,
metadata=None,
)
plan_create_mock.assert_any_call(
amount=2500,
interval="month",
name="Test Plan 1",
currency="usd",
id="test_id",
interval_count=None,
trial_period_days=None,
statement_descriptor=None,
metadata=None,
)
plan_create_mock.assert_any_call(
amount=5000,
interval="month",
name="Test Plan 2",
currency="usd",
id="test_id_2",
interval_count=None,
trial_period_days=None,
statement_descriptor=None,
metadata=None,
)
plan_create_mock.assert_any_call(
amount=5000,
interval="month",
name="Test Plan 3",
currency="usd",
id="test_id_3",
interval_count=None,
trial_period_days=None,
statement_descriptor=None,
metadata=None,
)
plan_create_mock.assert_any_call(
amount=7000,
interval="month",
name="Test Plan 4",
currency="usd",
id="test_id_4",
interval_count=None,
trial_period_days=7,
statement_descriptor=None,
metadata=None,
)
self.assertEqual(5, plan_create_mock.call_count)
def test_plan_exists(self, plan_create_mock):
plan_create_mock.side_effect = stripe.StripeError("Plan already exists.")
sync_plans(api_key)
self.assertTrue("ERROR: Plan already exists.", sys.stdout.getvalue().strip())
def test_plan_exists(self, plan_create_mock):
plan_create_mock.side_effect = StripeError("Plan already exists.")
sync_plans()
self.assertTrue("ERROR: Plan already exists.",
sys.stdout.getvalue().strip())
