1 files changed, 44 insertions, 1 deletions
diff --git a/spec/services/members/create_service_spec.rb b/spec/services/members/create_service_spec.rb
index 916941e1111..ffe63a8a94b 100644
--- a/spec/services/members/create_service_spec.rb
+++ b/spec/services/members/create_service_spec.rb
@@ -8,7 +8,8 @@ RSpec.describe Members::CreateService, :aggregate_failures, :clean_gitlab_redis_
   let_it_be(:member) { create(:user) }
   let_it_be(:user_ids) { member.id.to_s }
   let_it_be(:access_level) { Gitlab::Access::GUEST }
-  let(:params) { { user_ids: user_ids, access_level: access_level } }
+  let(:additional_params) { { invite_source: '_invite_source_' } }
+  let(:params) { { user_ids: user_ids, access_level: access_level }.merge(additional_params) }
 
   subject(:execute_service) { described_class.new(user, params.merge({ source: source })).execute }
 
@@ -82,4 +83,46 @@ RSpec.describe Members::CreateService, :aggregate_failures, :clean_gitlab_redis_
       expect(OnboardingProgress.completed?(source.namespace, :user_added)).to be(false)
     end
   end
+
+  context 'when tracking the invite source', :snowplow do
+    context 'when invite_source is not passed' do
+      let(:additional_params) { {} }
+
+      it 'tracks the invite source as unknown' do
+        expect { execute_service }.to raise_error(ArgumentError, 'No invite source provided.')
+
+        expect_no_snowplow_event
+      end
+    end
+
+    context 'when invite_source is passed' do
+      it 'tracks the invite source from params' do
+        execute_service
+
+        expect_snowplow_event(
+          category: described_class.name,
+          action: 'create_member',
+          label: '_invite_source_',
+          property: 'existing_user',
+          user: user
+        )
+      end
+    end
+
+    context 'when it is a net_new_user' do
+      let(:additional_params) { { invite_source: '_invite_source_', user_ids: 'email@example.org' } }
+
+      it 'tracks the invite source from params' do
+        execute_service
+
+        expect_snowplow_event(
+          category: described_class.name,
+          action: 'create_member',
+          label: '_invite_source_',
+          property: 'net_new_user',
+          user: user
+        )
+      end
+    end
+  end
 end