jaculabilis/danbooru
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

user upgrades: add UserUpgrade model.

Browse Source 
Add a model to store the status of user upgrades.

* Store the upgrade purchaser and the upgrade receiver (these are
  different for a gifted upgrade, the same for a self upgrade).
* Store the upgrade type: gold, platinum, or gold-to-platinum upgrades.
* Store the upgrade status:
** pending: User is still on the Stripe checkout page, no payment
   received yet.
** processing: User has completed checkout, but the checkout status in
   Stripe is still 'unpaid'.
** complete: We've received notification from Stripe that the payment
   has gone through and the user has been upgraded.
* Store the Stripe checkout ID, to cross-reference the upgrade record on
  Danbooru with the checkout record on Stripe.

This is the upgrade flow:

* When the user clicks the upgrade button on the upgrade page, we call
  POST /user_upgrades and create a pending UserUpgrade.
* We redirect the user to the checkout page on Stripe.
* When the user completes checkout on Stripe, Stripe sends us a webhook
  notification at POST /webhooks/receive.
* When we receive the webhook, we check the payment status, and if it's
  paid we mark the UserUpgrade as complete and upgrade the user.
* After Stripe sees that we have successfully processed the webhook,
  they redirect the user to the /user_upgrades/:id page, where we show
  the user their upgrade receipt.
This commit is contained in:
evazion
2020-12-24 06:40:20 -06:00
parent 7762489d7d
commit 74ed2a8b96
13 changed files with 502 additions and 164 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
test/functional/webhooks_controller_test.rb
View File

@@ -3,8 +3,8 @@ require 'test_helper'
class WebhooksControllerTest < ActionDispatch::IntegrationTest
mock_stripe!
def post_webhook(*args, **metadata)
event = StripeMock.mock_webhook_event(*args, metadata: metadata)
def post_webhook(*args, payment_status: "paid", **metadata)
event = StripeMock.mock_webhook_event(*args, payment_status: payment_status, metadata: metadata)
signature = generate_stripe_signature(event)
headers = { "Stripe-Signature": signature }
@@ -58,105 +58,83 @@ class WebhooksControllerTest < ActionDispatch::IntegrationTest
end
context "for a checkout.session.completed event" do
context "for completed event with an unpaid payment status" do
should "not upgrade the user" do
@user_upgrade = create(:self_gold_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id, payment_status: "unpaid" })
assert_response 200
assert_equal("processing", @user_upgrade.reload.status)
assert_equal(User::Levels::MEMBER, @user_upgrade.recipient.reload.level)
end
end
context "for a self upgrade" do
context "of a Member to Gold" do
context "to Gold" do
should "upgrade the user" do
@user = create(:member_user)
post_webhook("checkout.session.completed", {
recipient_id: @user.id,
purchaser_id: @user.id,
upgrade_type: "gold_upgrade",
level: User::Levels::GOLD,
})
@user_upgrade = create(:self_gold_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id })
assert_response 200
assert_equal(User::Levels::GOLD, @user.reload.level)
assert_equal("complete", @user_upgrade.reload.status)
assert_equal(User::Levels::GOLD, @user_upgrade.recipient.reload.level)
end
end
context "of a Member to Platinum" do
context "to Platinum" do
should "upgrade the user" do
@user = create(:member_user)
post_webhook("checkout.session.completed", {
recipient_id: @user.id,
purchaser_id: @user.id,
upgrade_type: "platinum_upgrade",
level: User::Levels::PLATINUM,
})
@user_upgrade = create(:self_platinum_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id })
assert_response 200
assert_equal(User::Levels::PLATINUM, @user.reload.level)
assert_equal("complete", @user_upgrade.reload.status)
assert_equal(User::Levels::PLATINUM, @user_upgrade.recipient.reload.level)
end
end
context "of a Gold user to Platinum" do
context "from Gold to Platinum" do
should "upgrade the user" do
@user = create(:gold_user)
post_webhook("checkout.session.completed", {
recipient_id: @user.id,
purchaser_id: @user.id,
upgrade_type: "gold_to_platinum_upgrade",
level: User::Levels::PLATINUM,
})
@user_upgrade = create(:self_gold_to_platinum_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id })
assert_response 200
assert_equal(User::Levels::PLATINUM, @user.reload.level)
assert_equal("complete", @user_upgrade.reload.status)
assert_equal(User::Levels::PLATINUM, @user_upgrade.recipient.reload.level)
end
end
end
context "for a gifted upgrade" do
context "of a Member to Gold" do
context "to Gold" do
should "upgrade the user" do
@recipient = create(:member_user)
@purchaser = create(:member_user)
post_webhook("checkout.session.completed", {
recipient_id: @recipient.id,
purchaser_id: @purchaser.id,
upgrade_type: "gold_upgrade",
level: User::Levels::GOLD,
})
@user_upgrade = create(:gift_gold_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id })
assert_response 200
assert_equal(User::Levels::GOLD, @recipient.reload.level)
assert_equal("complete", @user_upgrade.reload.status)
assert_equal(User::Levels::GOLD, @user_upgrade.recipient.reload.level)
end
end
context "of a Member to Platinum" do
context "to Platinum" do
should "upgrade the user" do
@recipient = create(:member_user)
@purchaser = create(:member_user)
post_webhook("checkout.session.completed", {
recipient_id: @recipient.id,
purchaser_id: @purchaser.id,
upgrade_type: "platinum_upgrade",
level: User::Levels::PLATINUM,
})
@user_upgrade = create(:gift_platinum_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id })
assert_response 200
assert_equal(User::Levels::PLATINUM, @recipient.reload.level)
assert_equal("complete", @user_upgrade.reload.status)
assert_equal(User::Levels::PLATINUM, @user_upgrade.recipient.reload.level)
end
end
context "of a Gold user to Platinum" do
context "from Gold to Platinum" do
should "upgrade the user" do
@recipient = create(:gold_user)
@purchaser = create(:member_user)
post_webhook("checkout.session.completed", {
recipient_id: @recipient.id,
purchaser_id: @purchaser.id,
upgrade_type: "gold_to_platinum_upgrade",
level: User::Levels::PLATINUM,
})
@user_upgrade = create(:gift_gold_to_platinum_upgrade)
post_webhook("checkout.session.completed", { user_upgrade_id: @user_upgrade.id })
assert_response 200
assert_equal(User::Levels::PLATINUM, @recipient.reload.level)
assert_equal("complete", @user_upgrade.reload.status)
assert_equal(User::Levels::PLATINUM, @user_upgrade.recipient.reload.level)
end
end
end