1 files changed, 31 insertions, 0 deletions
diff --git a/spec/support/shared_examples/models/member_shared_examples.rb b/spec/support/shared_examples/models/member_shared_examples.rb
index 56c202cb228..a2909c66e22 100644
--- a/spec/support/shared_examples/models/member_shared_examples.rb
+++ b/spec/support/shared_examples/models/member_shared_examples.rb
@@ -299,6 +299,22 @@ RSpec.shared_examples_for "member creation" do
         end
       end
     end
+
+    context 'when `tasks_to_be_done` and `tasks_project_id` are passed' do
+      before do
+        stub_experiments(invite_members_for_task: true)
+      end
+
+      it 'creates a member_task with the correct attributes', :aggregate_failures do
+        task_project = source.is_a?(Group) ? create(:project, group: source) : source
+        described_class.new(source, user, :developer, tasks_to_be_done: %w(ci code), tasks_project_id: task_project.id).execute
+
+        member = source.members.last
+
+        expect(member.tasks_to_be_done).to match_array([:ci, :code])
+        expect(member.member_task.project).to eq(task_project)
+      end
+    end
   end
 end
 
@@ -379,5 +395,20 @@ RSpec.shared_examples_for "bulk member creation" do
         expect(members).to all(be_persisted)
       end
     end
+
+    context 'when `tasks_to_be_done` and `tasks_project_id` are passed' do
+      before do
+        stub_experiments(invite_members_for_task: true)
+      end
+
+      it 'creates a member_task with the correct attributes', :aggregate_failures do
+        task_project = source.is_a?(Group) ? create(:project, group: source) : source
+        members = described_class.add_users(source, [user1], :developer, tasks_to_be_done: %w(ci code), tasks_project_id: task_project.id)
+        member = members.last
+
+        expect(member.tasks_to_be_done).to match_array([:ci, :code])
+        expect(member.member_task.project).to eq(task_project)
+      end
+    end
   end
 end