mirror of
https://github.com/plausible/analytics.git
synced 2024-12-23 17:44:43 +03:00
7d6f10f0cb
* Ensure team is present before use in sync logic * Ensure teams backfill works against partially assigned sites * Associate site with team on creation * Associate site with team on sync * Reuse alias * Add tests for invitation creation sync * Move team assertions to a helper module * Format * Test team creation on site creation via Sites context module * Add tests for teams sync on subscription changes * Tag tests * Test grace period start syncing up with teams * Test grace period manual lock sycning w/ teams * Test grace period end sycing up w/ teams * Test clearing grace period sync with teams * Update moduledoc * Fix missing preloads and wrong result pattern matching in sync logic * Test sync on accepting invites and site transfers * Test sync on membership role update and member removal * transfer async fix WIP * Stop privisioning team in site factory * Remove unused relationship from Site schema * Ensure consistent parsing of `passthrough` from Paddle webhook * Update team passthrough notification tests & logic --------- Co-authored-by: Adam Rutkowski <hq@mtod.org>
615 lines
19 KiB
Elixir
615 lines
19 KiB
Elixir
defmodule Plausible.Workers.CheckUsageTest do
|
|
use Plausible.DataCase, async: true
|
|
use Bamboo.Test
|
|
use Plausible.Teams.Test
|
|
import Double
|
|
|
|
alias Plausible.Workers.CheckUsage
|
|
|
|
require Plausible.Billing.Subscription.Status
|
|
|
|
setup [:create_user, :create_site]
|
|
@paddle_id_10k "558018"
|
|
@date_range Date.range(Timex.today(), Timex.today())
|
|
|
|
@accepted_status_values [
|
|
Plausible.Billing.Subscription.Status.active(),
|
|
Plausible.Billing.Subscription.Status.past_due(),
|
|
Plausible.Billing.Subscription.Status.deleted()
|
|
]
|
|
|
|
test "ignores user without subscription" do
|
|
CheckUsage.perform(nil)
|
|
|
|
assert_no_emails_delivered()
|
|
end
|
|
|
|
test "operates on the current subscription",
|
|
%{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: :active
|
|
)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: "wont-exist-should-crash",
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
inserted_at: DateTime.shift(DateTime.utc_now(), day: -2),
|
|
status: :deleted
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_email_delivered_with(
|
|
to: [user],
|
|
subject: "[Action required] You have outgrown your Plausible subscription tier"
|
|
)
|
|
end
|
|
|
|
test "sends more than one email", %{user: user} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
user2 = insert(:user)
|
|
insert(:site, members: [user2])
|
|
|
|
for u <- [user, user2] do
|
|
insert(:subscription,
|
|
user: u,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
next_bill_date: Timex.shift(Timex.today(), days: +5),
|
|
status: :active
|
|
)
|
|
end
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_email_delivered_with(
|
|
to: [user],
|
|
subject: "[Action required] You have outgrown your Plausible subscription tier"
|
|
)
|
|
|
|
assert_email_delivered_with(
|
|
to: [user2],
|
|
subject: "[Action required] You have outgrown your Plausible subscription tier"
|
|
)
|
|
end
|
|
|
|
test "ignores user with paused subscription", %{user: user} do
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: Plausible.Billing.Subscription.Status.paused()
|
|
)
|
|
|
|
CheckUsage.perform(nil)
|
|
|
|
assert_no_emails_delivered()
|
|
end
|
|
|
|
for status <- @accepted_status_values do
|
|
describe "#{status} subscription, regular customers" do
|
|
test "ignores user with subscription but no usage", %{user: user} do
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil)
|
|
|
|
assert_no_emails_delivered()
|
|
assert Repo.reload(user).grace_period == nil
|
|
end
|
|
|
|
test "does not send an email if account has been over the limit for one billing month", %{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 9_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_no_emails_delivered()
|
|
assert Repo.reload(user).grace_period == nil
|
|
end
|
|
|
|
test "does not send an email if account is over the limit by less than 10%", %{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 10_999},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_no_emails_delivered()
|
|
assert Repo.reload(user).grace_period == nil
|
|
end
|
|
|
|
test "sends an email when an account is over their limit for two consecutive billing months",
|
|
%{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_email_delivered_with(
|
|
to: [user],
|
|
subject: "[Action required] You have outgrown your Plausible subscription tier"
|
|
)
|
|
|
|
assert Repo.reload(user).grace_period.end_date == Timex.shift(Timex.today(), days: 7)
|
|
end
|
|
|
|
@tag :teams
|
|
test "syncs grace period start with teams",
|
|
%{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert user = Repo.reload!(user)
|
|
assert user.grace_period.end_date == Timex.shift(Timex.today(), days: 7)
|
|
team = assert_team_exists(user)
|
|
assert team.grace_period.end_date == user.grace_period.end_date
|
|
|
|
assert team.grace_period.id == user.grace_period.id
|
|
end
|
|
|
|
test "sends an email suggesting enterprise plan when usage is greater than 10M ", %{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_delivered_email_matches(%{html_body: html_body})
|
|
|
|
assert html_body =~
|
|
"Your usage exceeds our standard plans, so please reply back to this email for a tailored quote"
|
|
end
|
|
|
|
test "skips checking users who already have a grace period", %{user: user} do
|
|
%{grace_period: existing_grace_period} = Plausible.Users.start_grace_period(user)
|
|
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_no_emails_delivered()
|
|
assert Repo.reload(user).grace_period.id == existing_grace_period.id
|
|
end
|
|
|
|
test "recommends a plan to upgrade to", %{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_delivered_email_matches(%{
|
|
html_body: html_body
|
|
})
|
|
|
|
assert html_body =~ "We recommend you upgrade to the 100k/mo plan"
|
|
end
|
|
|
|
test "clears grace period when plan is applicable again", %{user: user} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
assert user |> Repo.reload() |> Plausible.Auth.GracePeriod.active?()
|
|
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 9_000}
|
|
}
|
|
end)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
refute user |> Repo.reload() |> Plausible.Auth.GracePeriod.active?()
|
|
end
|
|
|
|
@tag :teams
|
|
test "syncs clearing grace period with teams", %{user: user} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
assert user |> Repo.reload() |> Plausible.Auth.GracePeriod.active?()
|
|
team = assert_team_exists(user)
|
|
assert Plausible.Auth.GracePeriod.active?(team)
|
|
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 9_000}
|
|
}
|
|
end)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
refute team |> Repo.reload!() |> Plausible.Auth.GracePeriod.active?()
|
|
end
|
|
end
|
|
end
|
|
|
|
for status <- @accepted_status_values do
|
|
describe "#{status} subscription, enterprise customers" do
|
|
test "skips checking enterprise users who already have a grace period", %{user: user} do
|
|
%{grace_period: existing_grace_period} =
|
|
Plausible.Users.start_manual_lock_grace_period(user)
|
|
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 1_100_000},
|
|
last_cycle: %{date_range: @date_range, total: 1_100_000}
|
|
}
|
|
end)
|
|
|
|
enterprise_plan = insert(:enterprise_plan, user: user, monthly_pageview_limit: 1_000_000)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: enterprise_plan.paddle_plan_id,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_no_emails_delivered()
|
|
assert Repo.reload(user).grace_period.id == existing_grace_period.id
|
|
end
|
|
|
|
test "checks billable pageview usage for enterprise customer, sends usage information to enterprise@plausible.io",
|
|
%{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 1_100_000},
|
|
last_cycle: %{date_range: @date_range, total: 1_100_000}
|
|
}
|
|
end)
|
|
|
|
enterprise_plan = insert(:enterprise_plan, user: user, monthly_pageview_limit: 1_000_000)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: enterprise_plan.paddle_plan_id,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_email_delivered_with(
|
|
to: [{nil, "enterprise@plausible.io"}],
|
|
subject: "#{user.email} has outgrown their enterprise plan"
|
|
)
|
|
end
|
|
|
|
test "checks site limit for enterprise customer, sends usage information to enterprise@plausible.io",
|
|
%{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 1},
|
|
last_cycle: %{date_range: @date_range, total: 1}
|
|
}
|
|
end)
|
|
|
|
enterprise_plan = insert(:enterprise_plan, user: user, site_limit: 2)
|
|
|
|
insert(:site, members: [user])
|
|
insert(:site, members: [user])
|
|
insert(:site, members: [user])
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: enterprise_plan.paddle_plan_id,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_email_delivered_with(
|
|
to: [{nil, "enterprise@plausible.io"}],
|
|
subject: "#{user.email} has outgrown their enterprise plan"
|
|
)
|
|
end
|
|
|
|
test "starts grace period when plan is outgrown", %{user: user} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 1_100_000},
|
|
last_cycle: %{date_range: @date_range, total: 1_100_000}
|
|
}
|
|
end)
|
|
|
|
enterprise_plan = insert(:enterprise_plan, user: user, monthly_pageview_limit: 1_000_000)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: enterprise_plan.paddle_plan_id,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
assert user |> Repo.reload() |> Plausible.Auth.GracePeriod.active?()
|
|
end
|
|
|
|
@tag :teams
|
|
test "manual lock grace period is synced with teams", %{user: user} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 1_100_000},
|
|
last_cycle: %{date_range: @date_range, total: 1_100_000}
|
|
}
|
|
end)
|
|
|
|
enterprise_plan = insert(:enterprise_plan, user: user, monthly_pageview_limit: 1_000_000)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: enterprise_plan.paddle_plan_id,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1),
|
|
status: unquote(status)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert user = Repo.reload!(user)
|
|
team = assert_team_exists(user)
|
|
assert team.grace_period.manual_lock == user.grace_period.manual_lock
|
|
assert team.grace_period.manual_lock == true
|
|
end
|
|
end
|
|
end
|
|
|
|
describe "timing" do
|
|
test "checks usage one day after the last_bill_date", %{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: Timex.shift(Timex.today(), days: -1)
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub)
|
|
|
|
assert_email_delivered_with(
|
|
to: [user],
|
|
subject: "[Action required] You have outgrown your Plausible subscription tier"
|
|
)
|
|
end
|
|
|
|
test "does not check exactly one month after last_bill_date", %{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: ~D[2021-03-28]
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub, ~D[2021-03-28])
|
|
|
|
assert_no_emails_delivered()
|
|
end
|
|
|
|
test "for yearly subscriptions, checks usage multiple months + one day after the last_bill_date",
|
|
%{
|
|
user: user
|
|
} do
|
|
usage_stub =
|
|
Plausible.Billing.Quota.Usage
|
|
|> stub(:monthly_pageview_usage, fn _user ->
|
|
%{
|
|
penultimate_cycle: %{date_range: @date_range, total: 11_000},
|
|
last_cycle: %{date_range: @date_range, total: 11_000}
|
|
}
|
|
end)
|
|
|
|
insert(:subscription,
|
|
user: user,
|
|
paddle_plan_id: @paddle_id_10k,
|
|
last_bill_date: ~D[2021-06-29]
|
|
)
|
|
|
|
CheckUsage.perform(nil, usage_stub, ~D[2021-08-30])
|
|
|
|
assert_email_delivered_with(
|
|
to: [user],
|
|
subject: "[Action required] You have outgrown your Plausible subscription tier"
|
|
)
|
|
end
|
|
end
|
|
end
|