1 files changed, 83 insertions, 10 deletions

diff --git a/spec/lib/gitlab/subscription_portal_spec.rb b/spec/lib/gitlab/subscription_portal_spec.rb
index 628eb380396..a3808b0f0e2 100644
--- a/spec/lib/gitlab/subscription_portal_spec.rb
+++ b/spec/lib/gitlab/subscription_portal_spec.rb
@@ -5,23 +5,96 @@ require 'spec_helper'
 RSpec.describe ::Gitlab::SubscriptionPortal do
   using RSpec::Parameterized::TableSyntax
 
-  where(:method_name, :test, :development, :result) do
-    :default_subscriptions_url | false | false  | 'https://customers.gitlab.com'
-    :default_subscriptions_url | false | true   | 'https://customers.stg.gitlab.com'
-    :default_subscriptions_url | true  | false  | 'https://customers.stg.gitlab.com'
-    :payment_form_url          | false | false  | 'https://customers.gitlab.com/payment_forms/cc_validation'
-    :payment_form_url          | false | true   | 'https://customers.stg.gitlab.com/payment_forms/cc_validation'
-    :payment_form_url          | true  | false  | 'https://customers.stg.gitlab.com/payment_forms/cc_validation'
+  let(:env_value) { nil }
+
+  before do
+    stub_env('CUSTOMER_PORTAL_URL', env_value)
+    stub_feature_flags(new_customersdot_staging_url: false)
   end
 
-  with_them do
-    subject { described_class.method(method_name).call }
+  describe '.default_subscriptions_url' do
+    where(:test, :development, :result) do
+      false | false | 'https://customers.gitlab.com'
+      false | true  | 'https://customers.stg.gitlab.com'
+      true  | false | 'https://customers.stg.gitlab.com'
+    end
 
     before do
       allow(Rails).to receive_message_chain(:env, :test?).and_return(test)
       allow(Rails).to receive_message_chain(:env, :development?).and_return(development)
     end
 
-    it { is_expected.to eq(result) }
+    with_them do
+      subject { described_class.default_subscriptions_url }
+
+      it { is_expected.to eq(result) }
+    end
+  end
+
+  describe '.subscriptions_url' do
+    subject { described_class.subscriptions_url }
+
+    context 'when CUSTOMER_PORTAL_URL ENV is unset' do
+      it { is_expected.to eq('https://customers.stg.gitlab.com') }
+    end
+
+    context 'when CUSTOMER_PORTAL_URL ENV is set' do
+      let(:env_value) { 'https://customers.example.com' }
+
+      it { is_expected.to eq(env_value) }
+    end
+  end
+
+  describe '.subscriptions_comparison_url' do
+    subject { described_class.subscriptions_comparison_url }
+
+    link_match = %r{\Ahttps://about\.gitlab\.((cn/pricing/saas)|(com/pricing/gitlab-com))/feature-comparison\z}
+
+    it { is_expected.to match(link_match) }
+  end
+
+  context 'url methods' do
+    where(:method_name, :result) do
+      :default_subscriptions_url         | 'https://customers.stg.gitlab.com'
+      :payment_form_url                  | 'https://customers.stg.gitlab.com/payment_forms/cc_validation'
+      :subscriptions_graphql_url         | 'https://customers.stg.gitlab.com/graphql'
+      :subscriptions_more_minutes_url    | 'https://customers.stg.gitlab.com/buy_pipeline_minutes'
+      :subscriptions_more_storage_url    | 'https://customers.stg.gitlab.com/buy_storage'
+      :subscriptions_manage_url          | 'https://customers.stg.gitlab.com/subscriptions'
+      :subscriptions_plans_url           | 'https://customers.stg.gitlab.com/plans'
+      :subscriptions_instance_review_url | 'https://customers.stg.gitlab.com/instance_review'
+      :subscriptions_gitlab_plans_url    | 'https://customers.stg.gitlab.com/gitlab_plans'
+    end
+
+    with_them do
+      subject { described_class.send(method_name) }
+
+      it { is_expected.to eq(result) }
+    end
+  end
+
+  describe '.add_extra_seats_url' do
+    subject { described_class.add_extra_seats_url(group_id) }
+
+    let(:group_id) { 153 }
+
+    it { is_expected.to eq("https://customers.stg.gitlab.com/gitlab/namespaces/#{group_id}/extra_seats") }
+  end
+
+  describe '.upgrade_subscription_url' do
+    subject { described_class.upgrade_subscription_url(group_id, plan_id) }
+
+    let(:group_id) { 153 }
+    let(:plan_id) { 5 }
+
+    it { is_expected.to eq("https://customers.stg.gitlab.com/gitlab/namespaces/#{group_id}/upgrade/#{plan_id}") }
+  end
+
+  describe '.renew_subscription_url' do
+    subject { described_class.renew_subscription_url(group_id) }
+
+    let(:group_id) { 153 }
+
+    it { is_expected.to eq("https://customers.stg.gitlab.com/gitlab/namespaces/#{group_id}/renew") }
   end
 end